Allow up to 5 warming waves for lambda with 3 warm lambdas and more memory

Author: devksingh4

cloudformation/warmer.yml

@@ -12,7 +12,7 @@ Parameters:
     AllowedPattern: ^[a-zA-Z0-9]+[a-zA-Z0-9-]+[a-zA-Z0-9]+$
   NumInstances:
     Type: Number
-    Default: 2
+    Default: 3
 
 Resources:
   WarmerLambdaLogGroup:
@@ -67,7 +67,7 @@ Resources:
       Description: !Sub "Lambda warmer for function ${FunctionToWarm}"
       FunctionName: !Sub "${LambdaFunctionNameBase}-${FunctionToWarm}"
       Handler: lambda.handler
-      MemorySize: 128
+      MemorySize: 512
       Role: !GetAtt WarmerLambdaIAMRole.Arn
       Timeout: 15
       Environment:

src/api/warmer/lambda.ts

@@ -1,14 +1,9 @@
-import { LambdaClient, InvokeCommand, LogType } from "@aws-sdk/client-lambda";
+import { LambdaClient, InvokeCommand } from "@aws-sdk/client-lambda";
 import { TextDecoder } from "util";
 
 const lambdaClient = new LambdaClient({});
 const textDecoder = new TextDecoder();
 
-interface WarmerEvent {
-  lambdaName: string;
-  numInstances: number;
-}
-
 /**
  * Invokes a batch of lambdas concurrently and returns the unique instance IDs found.
  */
@@ -64,8 +59,8 @@ export const handler = async (event: {}) => {
   }
 
   const numInstances = parseInt(numInstancesStr, 10);
-  // Default to 2 waves if MAX_WAVES is not set
-  const maxWaves = parseInt(maxWavesStr || "2", 10);
+  // Default to 5 waves if MAX_WAVES is not set
+  const maxWaves = parseInt(maxWavesStr || "5", 10);
 
   let totalInvocations = 0;
   let wavesCompleted = 0;