resume subtasks

st-tech · Dec 20, 2023 · faf9d38 · faf9d38
1 parent 703d50f
commit faf9d38
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/digdag-core/src/main/java/io/digdag/core/workflow/TaskControl.java b/digdag-core/src/main/java/io/digdag/core/workflow/TaskControl.java
@@ -182,6 +182,20 @@ private static long addTasks(TaskControlStore store,
             firstTask = false;
         }
 
+        List<String> taskNames = tasks.stream().map(t-> t.getFullName()).collect(Collectors.toList());
+
+        resumingTasks
+            .stream()
+            .filter(resumingTask -> !taskNames.contains(resumingTask.getFullName()))
+            .forEach(resumingTask -> {
+                store.addResumedSubtask(attemptId,
+                    resumingTask.getParentId().get(),
+                    resumingTask.getTaskType(),
+                    TaskStateCode.SUCCESS,
+                    (isInitialTask ? TaskStateFlags.empty().withInitialTask() : TaskStateFlags.empty()),
+                    resumingTask);
+            });
+
         return rootTaskId;
     }