fix(core): task restore interrupt problem on restart server (apache#2401

) * Update StandardTaskScheduler.java
VGalaxies · Jan 12, 2024 · 6349466 · 6349466
1 parent 9dc38f7
commit 6349466
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/...-server/hugegraph-core/src/main/java/org/apache/hugegraph/task/StandardTaskScheduler.java b/...-server/hugegraph-core/src/main/java/org/apache/hugegraph/task/StandardTaskScheduler.java
@@ -138,6 +138,7 @@ private TaskTransaction tx() {
     @Override
     public <V> void restoreTasks() {
         Id selfServer = this.serverManager().selfNodeId();
+        List<HugeTask<V>> taskList = new ArrayList<>();
         // Restore 'RESTORING', 'RUNNING' and 'QUEUED' tasks in order.
         for (TaskStatus status : TaskStatus.PENDING_STATUSES) {
             String page = this.supportsPaging() ? PageInfo.PAGE_NONE : null;
@@ -147,14 +148,18 @@ public <V> void restoreTasks() {
                      iter.hasNext();) {
                     HugeTask<V> task = iter.next();
                     if (selfServer.equals(task.server())) {
-                        this.restore(task);
+                        taskList.add(task);
                     }
                 }
                 if (page != null) {
                     page = PageInfo.pageInfo(iter);
                 }
             } while (page != null);
         }
+        for (HugeTask<V> task : taskList){
+            LOG.info("restore task {}", task);
+            this.restore(task);
+        }
     }
 
     private <V> Future<?> restore(HugeTask<V> task) {