From 09bc222ffe1ce569cf7317617769cc6c405e38e9 Mon Sep 17 00:00:00 2001 From: Lauren Leach Date: Wed, 30 Jul 2025 13:30:47 -0700 Subject: [PATCH] aws lambda start with context --- pkg/cli/lambda_server__added.go | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/pkg/cli/lambda_server__added.go b/pkg/cli/lambda_server__added.go index 553011214..d8faa00b4 100644 --- a/pkg/cli/lambda_server__added.go +++ b/pkg/cli/lambda_server__added.go @@ -95,7 +95,7 @@ func OptionallyAddLambdaCommand[T field.Configurable]( } client, webKey, err := c1_lambda_config.GetConnectorConfigServiceClient( - ctx, + runCtx, v.GetString(field.LambdaServerClientIDField.GetName()), v.GetString(field.LambdaServerClientSecretField.GetName()), ) @@ -104,7 +104,7 @@ func OptionallyAddLambdaCommand[T field.Configurable]( } // Get configuration, convert it to viper flag values, then proceed. - config, err := client.GetConnectorConfig(ctx, &pb_connector_api.GetConnectorConfigRequest{}) + config, err := client.GetConnectorConfig(runCtx, &pb_connector_api.GetConnectorConfigRequest{}) if err != nil { return fmt.Errorf("lambda-run: failed to get connector config: %w", err) } @@ -178,9 +178,9 @@ func OptionallyAddLambdaCommand[T field.Configurable]( } s := c1_lambda_grpc.NewServer(authOpt) - connector.Register(ctx, s, c, opts) + connector.Register(runCtx, s, c, opts) - aws_lambda.Start(s.Handler) + aws_lambda.StartWithOptions(s.Handler, aws_lambda.WithContext(runCtx)) return nil } return nil