feat: lambda worker - support maxConcurrency (SqsEventSource)

techfromsage · Apr 2, 2024 · 457459e · 457459e
1 parent bc3a840
commit 457459e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/lib/lambda-worker/lambda-worker-props.ts b/lib/lambda-worker/lambda-worker-props.ts
@@ -45,6 +45,7 @@ export interface LambdaWorkerProps {
     timeout: cdk.Duration;
     vpc?: ec2.IVpc;
     vpcSubnets?: ec2.SubnetSelection;
+    queueMaxConcurrency: number;
   } & Partial<FunctionLambdaProps> &
     Partial<ContainerFromEcrLambdaProps> &
     Partial<ContainerFromImageAssetLambdaProps>;

diff --git a/lib/lambda-worker/lambda-worker.ts b/lib/lambda-worker/lambda-worker.ts
@@ -141,6 +141,7 @@ export class LambdaWorker extends Construct {
       new SqsEventSource(lambdaQueue, {
         enabled: props.lambdaProps.enableQueue ?? true,
         batchSize: 1,
+        maxConcurrency: props.lambdaProps.queueMaxConcurrency,
       }),
     );
     lambdaWorker.addEventSource(