Merge pull request #3 from skozawa/feature/aws-endpoint

AWS_ENDPOINT指定できるように
koid · Mar 26, 2021 · 7441616 · 7441616
2 parents 714fd22 + 724060b
commit 7441616
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/main.go b/main.go
@@ -137,7 +137,15 @@ func main() {
 	}
 
 	// kinsumer
-	sess := session.Must(session.NewSession(aws.NewConfig()))
+	var sess *session.Session
+	if len(os.Getenv("AWS_ENDPOINT")) != 0 {
+		sess = session.Must(session.NewSession(&aws.Config{
+			Endpoint:         aws.String(os.Getenv("AWS_ENDPOINT")),
+			S3ForcePathStyle: aws.Bool(true),
+		}))
+	} else {
+		sess = session.Must(session.NewSession(aws.NewConfig()))
+	}
 	k, err := kinsumer.NewWithSession(
 		sess,
 		kinesisStreamName,