许家凯 5 سال پیش
والد
کامیت
e9f4896f7b
1فایلهای تغییر یافته به همراه6 افزوده شده و 3 حذف شده
  1. 6 3
      src/main/java/com/winhc/dataworks/flow/touch/Main.java

+ 6 - 3
src/main/java/com/winhc/dataworks/flow/touch/Main.java

@@ -11,6 +11,7 @@ import com.winhc.dataworks.flow.touch.service.TouchService;
 import lombok.SneakyThrows;
 import lombok.extern.slf4j.Slf4j;
 
+import java.util.ArrayList;
 import java.util.List;
 import java.util.Map;
 import java.util.stream.Collectors;
@@ -74,6 +75,7 @@ public class Main {
             int failedNode = 0;
 
             boolean flag = true;
+            List<String> empty = new ArrayList<>();
 
             for (TaskInfo taskInfo : collect) {
                 Map<String, TaskFlowEnum> query = touchService.query(taskInfo.getProject(), taskInfo.getDagId());
@@ -87,9 +89,10 @@ public class Main {
                             }
                             return entry;
                         }).collect(Collectors.groupingBy(Entry::getKey, Collectors.mapping(Entry::getValue, Collectors.toList())));
-                List<String> success = status.get(TaskFlowEnum.SUCCESS.getMsg());
-                List<String> failure = status.get(TaskFlowEnum.FAILURE.getMsg());
-                List<String> await = status.get("运行中");
+                List<String> success = status.getOrDefault(TaskFlowEnum.SUCCESS.getMsg(), empty);
+                List<String> failure = status.getOrDefault(TaskFlowEnum.FAILURE.getMsg(), empty);
+                List<String> await = status.getOrDefault("运行中", empty);
+
 
                 if (await.size() != 0) {
                     awaitTask++;