Change aws credentials env loading

oneee-playground · Jul 14, 2024 · 7498bcb · 7498bcb
1 parent 66de331
commit 7498bcb
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 4 deletions.
diff --git a/cmd/app/main.go b/cmd/app/main.go
@@ -6,7 +6,8 @@ import (
 	"os"
 	"time"
 
-	"github.com/aws/aws-sdk-go-v2/config"
+	"github.com/aws/aws-sdk-go-v2/aws"
+	"github.com/aws/aws-sdk-go-v2/credentials"
 	"github.com/aws/aws-sdk-go-v2/service/sqs"
 	"github.com/docker/docker/client"
 	influxdb2 "github.com/influxdata/influxdb-client-go"
@@ -36,9 +37,9 @@ func main() {
 	httpClient := &http.Client{}
 	storage := storage.NewFSStorage(conf.WorkStoragePath)
 
-	awsConfig, err := config.LoadDefaultConfig(context.Background(), config.WithRegion(conf.AWSRegion))
-	if err != nil {
-		logger.Fatal("failed to load aws config", zap.Error(err))
+	awsConfig := aws.Config{
+		Region:      "ap-northeast-2",
+		Credentials: credentials.NewStaticCredentialsProvider(conf.AccessKeyID, conf.SecretAccessKey, ""),
 	}
 
 	sqsClient := sqs.NewFromConfig(awsConfig)

diff --git a/internal/config/load.go b/internal/config/load.go
@@ -12,4 +12,7 @@ func LoadFromEnv() {
 
 	JobQueueURL = os.Getenv("AWS_SQS_JOB_QUEUE_URL")
 	EventQueueURL = os.Getenv("AWS_SQS_TEST_EVENT_QUEUE_URL")
+
+	AccessKeyID = os.Getenv("AWS_ACCESS_KEY_ID")
+	SecretAccessKey = os.Getenv("AWS_SECRET_ACCESS_KEY")
 }
diff --git a/internal/config/var.go b/internal/config/var.go
@@ -12,4 +12,7 @@ var (
 var (
 	JobQueueURL   string
 	EventQueueURL string
+
+	AccessKeyID     string
+	SecretAccessKey string
 )