feat: add Lambda Ephemeral storage to LambdaWorker props (#61)

* feat: add Lambda Ephemeral storage to LambdaWorker props
techfromsage · Aug 24, 2022 · f4c1ce8 · f4c1ce8
1 parent de4cd95
commit f4c1ce8
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 0 deletions.
diff --git a/lib/lambda-worker/lambda-worker-props.ts b/lib/lambda-worker/lambda-worker-props.ts
@@ -20,6 +20,7 @@ export interface LambdaWorkerProps {
     entry?: string;
     enableQueue?: boolean;
     environment?: { [key: string]: string };
+    ephemeralStorageSize?: cdk.Size;
     filesystem?: lambda.FileSystem;
     memorySize: number; // LambdaWorker will set a minimum memory size of 1024
     policyStatements?: iam.PolicyStatement[];

diff --git a/lib/lambda-worker/lambda-worker.ts b/lib/lambda-worker/lambda-worker.ts
@@ -252,6 +252,7 @@ export class LambdaWorker extends cdk.Construct {
       handler: props.lambdaProps.handler,
       description: props.lambdaProps.description,
       environment: props.lambdaProps.environment,
+      ephemeralStorageSize: props.lambdaProps.ephemeralStorageSize,
       memorySize: props.lambdaProps.memorySize,
       reservedConcurrentExecutions:
         props.lambdaProps.reservedConcurrentExecutions,
@@ -297,6 +298,7 @@ export class LambdaWorker extends cdk.Construct {
       functionName: props.name,
       description: props.lambdaProps.description,
       environment: props.lambdaProps.environment,
+      ephemeralStorageSize: props.lambdaProps.ephemeralStorageSize,
       memorySize: props.lambdaProps.memorySize,
       reservedConcurrentExecutions:
         props.lambdaProps.reservedConcurrentExecutions,

diff --git a/test/infra/lambda-worker/lambda-worker.test.ts b/test/infra/lambda-worker/lambda-worker.test.ts
@@ -226,6 +226,7 @@ describe("LambdaWorker", () => {
 
             // Optional
             description: "Test Description",
+            ephemeralStorageSize: cdk.Size.mebibytes(1024),
             environment: {
               TALIS_ENV_VAR: "some value",
             },
@@ -254,6 +255,9 @@ describe("LambdaWorker", () => {
                 AWS_NODEJS_CONNECTION_REUSE_ENABLED: "1",
               },
             },
+            EphemeralStorage: {
+              Size: 1024,
+            },
             ReservedConcurrentExecutions: 10,
           })
         );