add initial check to support partially completed executions

explore-flights · Oct 28, 2024 · 051f8b6 · 051f8b6
1 parent e16b0d3
commit 051f8b6
Showing 1 changed file with 24 additions and 19 deletions.
diff --git a/cdk/lib/constructs/sfn-construct.ts b/cdk/lib/constructs/sfn-construct.ts
@@ -34,26 +34,31 @@ export class SfnConstruct extends Construct {
 
     const checkRemainingChoice = new Choice(this, 'CheckRemaining', {});
 
-    const definition = new LambdaInvoke(this, 'PrepareDailyCron', {
-      lambdaFunction: props.cronLambda_1G,
-      payload: TaskInput.fromObject({
-        'action': 'cron',
-        'params': {
-          'prepareDailyCron': {
-            'time': JsonPath.stringAt('$.time'),
-            'offset': -1,
-            'total': 30 * 12,
+    const definition = new Choice(this, 'CheckInitial', {})
+      .when(
+        Condition.isNotPresent('$.loadScheduleRanges'),
+        new LambdaInvoke(this, 'PrepareDailyCron', {
+          lambdaFunction: props.cronLambda_1G,
+          payload: TaskInput.fromObject({
+            'action': 'cron',
+            'params': {
+              'prepareDailyCron': {
+                'time': JsonPath.stringAt('$.time'),
+                'offset': -1,
+                'total': 30 * 12,
+              },
+            },
+          }),
+          payloadResponseOnly: true,
+          resultSelector: {
+            'completed': [],
+            'remaining': JsonPath.objectAt('$.prepareDailyCron.dateRanges'),
           },
-        },
-      }),
-      payloadResponseOnly: true,
-      resultSelector: {
-        'completed': [],
-        'remaining': JsonPath.objectAt('$.prepareDailyCron.dateRanges'),
-      },
-      resultPath: '$.loadScheduleRanges',
-      retryOnServiceExceptions: true,
-    })
+          resultPath: '$.loadScheduleRanges',
+          retryOnServiceExceptions: true,
+        }),
+      )
+      .afterwards()
       .next(
         checkRemainingChoice
           // region loop body -> remaining dates