From 2ca6c4ecb921a11a1d9d4cbb8a1a3163a4353bf7 Mon Sep 17 00:00:00 2001
From: Shuhei Nagasawa <s.nagasawa.github@gmail.com>
Date: Thu, 15 Feb 2024 15:06:41 +0900
Subject: [PATCH] fix collectResumingTasksForResumeFailedMode

---
 .../java/io/digdag/server/rs/AttemptResource.java    | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/digdag-server/src/main/java/io/digdag/server/rs/AttemptResource.java b/digdag-server/src/main/java/io/digdag/server/rs/AttemptResource.java
index a678e3e690..c739bf52ae 100644
--- a/digdag-server/src/main/java/io/digdag/server/rs/AttemptResource.java
+++ b/digdag-server/src/main/java/io/digdag/server/rs/AttemptResource.java
@@ -2,6 +2,7 @@
 
 import java.util.List;
 import java.util.Set;
+import java.util.Arrays;
 import java.util.HashSet;
 import java.util.stream.Collectors;
 import javax.ws.rs.Consumes;
@@ -298,21 +299,24 @@ private List<Long> collectResumingTasks(RestSessionAttemptRequest.Resume resume)
 
     private List<Long> collectResumingTasksForResumeFailedMode(long attemptId)
     {
+        TaskStateCode[] statusArr = {TaskStateCode.SUCCESS, TaskStateCode.GROUP_ERROR, TaskStateCode.ERROR, TaskStateCode.CANCELED};
+        List<TaskStateCode> statuses = Arrays.asList(statusArr);
+
         List<ArchivedTask> tasks = sm
                 .getSessionStore(getSiteId())
                 .getTasksOfAttempt(attemptId);
 
-        List<Long> successTasks = tasks.stream()
-                .filter(task -> task.getState() == TaskStateCode.SUCCESS)
+        List<Long> ids = tasks.stream()
+                .filter(t-> statuses.contains(t.getState()))
                 .map(task -> {
-                    if (!task.getParentId().isPresent()) {
+                    if (!task.getParentId().isPresent() && task.getState() == TaskStateCode.SUCCESS) {
                         throw new IllegalArgumentException("Resuming successfully completed attempts is not supported");
                     }
                     return task.getId();
                 })
                 .collect(Collectors.toList());
 
-        return ImmutableList.copyOf(successTasks);
+        return ImmutableList.copyOf(ids);
     }
 
     private List<Long> collectResumingTasksForResumeFromMode(long attemptId, String fromTaskPattern)