pass ctx to newTimeStreamAdapter

Author: rt-joe

main.go

@@ -133,7 +133,7 @@ func main() {
 
 	sugar.Debugf("Sending logs to timeseries database %s in AWS Region %s", cfg.databaseName, cfg.awsRegion)
 
-	timeStreamAdapter := newTimeStreamAdapter(sugar, cfg, newTimestreamQueryPaginator, nil, nil)
+	timeStreamAdapter := newTimeStreamAdapter(ctx, sugar, cfg, newTimestreamQueryPaginator, nil, nil)
 	if err := serve(ctx, sugar, cfg.listenAddr, timeStreamAdapter); err != nil {
 		sugar.Errorw("Failed to listen", "addr", cfg.listenAddr, "err", err)
 		os.Exit(1)

timestream.go

@@ -81,7 +81,7 @@ func newTimestreamQueryPaginator(client TimestreamQueryApi, params *timestreamqu
 	return timestreamquery.NewQueryPaginator(client, params, optFns...)
 }
 
-func newTimeStreamAdapter(logger *zap.SugaredLogger, cfg *adapterCfg, newQueryPaginator NewQueryPaginator, writeSvc *timestreamwrite.Client, querySvc *timestreamquery.Client) TimeStreamAdapter {
+func newTimeStreamAdapter(ctx context.Context, logger *zap.SugaredLogger, cfg *adapterCfg, newQueryPaginator NewQueryPaginator, writeSvc *timestreamwrite.Client, querySvc *timestreamquery.Client) TimeStreamAdapter {
 	tr := &http.Transport{
 		ResponseHeaderTimeout: 20 * time.Second,
 		// Using DefaultTransport values for other parameters: https://golang.org/pkg/net/http/#RoundTripper
@@ -100,7 +100,7 @@ func newTimeStreamAdapter(logger *zap.SugaredLogger, cfg *adapterCfg, newQueryPa
 	_ = http2.ConfigureTransport(tr)
 
 	if writeSvc == nil || querySvc == nil {
-		awsCfg, err := config.LoadDefaultConfig(context.TODO(), config.WithRegion(cfg.awsRegion))
+		awsCfg, err := config.LoadDefaultConfig(ctx, config.WithRegion(cfg.awsRegion))
 		if err != nil {
 			logger.Fatalw("Unable to load AWS SDK config", "error", err)
 		}