fix: await cleanup in timer

dherault · Oct 10, 2022 · f5b8c68 · f5b8c68
1 parent b03cf79
commit f5b8c68
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/src/lambda/LambdaFunctionPool.js b/src/lambda/LambdaFunctionPool.js
@@ -25,7 +25,9 @@ export default class LambdaFunctionPool {
 
     // NOTE: don't use setInterval, as it would schedule always a new run,
     // regardless of function processing time and e.g. user action (debugging)
-    this.#timerRef = setTimeout(() => {
+    this.#timerRef = setTimeout(async () => {
+      const cleanupWait = []
+
       // console.log('run cleanup')
       this.#pool.forEach((lambdaFunctions, functionKey) => {
         lambdaFunctions.forEach((lambdaFunction) => {
@@ -35,8 +37,8 @@ export default class LambdaFunctionPool {
             status === 'IDLE' &&
             idleTimeInMillis >= functionCleanupIdleTimeInMillis
           ) {
-            // console.log(`removed Lambda Function ${lambdaFunction.functionName}`)
-            lambdaFunction.cleanup()
+            cleanupWait.push(lambdaFunction.cleanup())
+
             lambdaFunctions.delete(lambdaFunction)
           }
         })
@@ -46,6 +48,8 @@ export default class LambdaFunctionPool {
         }
       })
 
+      await Promise.all(cleanupWait)
+
       // schedule new timer
       this.#startCleanTimer()
     }, functionCleanupIdleTimeInMillis)