update Executor input mapping to allow optional parameters

alexcasalboni · Jun 10, 2020 · 28c5d0e · 28c5d0e
1 parent de40597
commit 28c5d0e
Show file tree

Hide file tree

Showing 3 changed files with 164 additions and 109 deletions.
diff --git a/lambda/executor.js b/lambda/executor.js
@@ -61,15 +61,16 @@ const validateInput = (lambdaARN, value, num) => {
 };
 
 const extractDataFromInput = (event) => {
+    const input = event.input; // original state machine input
     return {
-        lambdaARN: event.lambdaARN,
         value: parseInt(event.value, 10),
-        num: parseInt(event.num, 10),
-        enableParallel: !!event.parallelInvocation,
-        payload: event.payload,
-        dryRun: event.dryRun === true,
-        preProcessorARN: event.preProcessorARN,
-        postProcessorARN: event.postProcessorARN,
+        lambdaARN: input.lambdaARN,
+        num: parseInt(input.num, 10),
+        enableParallel: !!input.parallelInvocation,
+        payload: input.payload,
+        dryRun: input.dryRun === true,
+        preProcessorARN: input.preProcessorARN,
+        postProcessorARN: input.postProcessorARN,
     };
 };
 

diff --git a/template.yml b/template.yml
@@ -162,13 +162,8 @@ Resources:
                       "ItemsPath": "$.powerValues",
                       "ResultPath": "$.stats",
                       "Parameters": {
-                        "value.$": "$$.Map.Item.Value",
-                        "lambdaARN.$": "$.lambdaARN",
-                        "num.$": "$.num",
-                        "payload.$": "$.payload",
-                        "parallelInvocation.$": "$.parallelInvocation",
-                        "preProcessorARN.$": "$.preProcessorARN",
-                        "postProcessorARN.$": "$.postProcessorARN"
+                        "input.$": "$",
+                        "value.$": "$$.Map.Item.Value"
                       },
                       "MaxConcurrency": 0,
                       "Catch": [{