From 7c3f5b61d3d05ba860fad3c19b3289aefeee9d1f Mon Sep 17 00:00:00 2001
From: qiaozhanwei <qiaozhanwei@analysys.com.cn>
Date: Tue, 23 Apr 2019 18:23:56 +0800
Subject: [PATCH] command count update

---
 .../api/service/DataAnalysisService.java        | 17 ++++++++++++-----
 1 file changed, 12 insertions(+), 5 deletions(-)

diff --git a/escheduler-api/src/main/java/cn/escheduler/api/service/DataAnalysisService.java b/escheduler-api/src/main/java/cn/escheduler/api/service/DataAnalysisService.java
index 57c05d0689..7462519ca1 100644
--- a/escheduler-api/src/main/java/cn/escheduler/api/service/DataAnalysisService.java
+++ b/escheduler-api/src/main/java/cn/escheduler/api/service/DataAnalysisService.java
@@ -17,6 +17,7 @@
 package cn.escheduler.api.service;
 
 
+import cn.escheduler.api.dto.CommandStateCount;
 import cn.escheduler.api.dto.DefineUserDto;
 import cn.escheduler.api.dto.TaskCountDto;
 import cn.escheduler.api.enums.Status;
@@ -38,10 +39,7 @@ import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 
 import java.text.MessageFormat;
-import java.util.Date;
-import java.util.HashMap;
-import java.util.List;
-import java.util.Map;
+import java.util.*;
 
 /**
  * data analysis service
@@ -298,7 +296,16 @@ public class DataAnalysisService {
             dataMap.put(errorExecutionStatus.getExecutionStatus(),errorCommandStateCountsMap);
         }
 
-        result.put(Constants.DATA_LIST, dataMap);
+        List<CommandStateCount> list = new ArrayList<>();
+        Iterator<Map.Entry<ExecutionStatus, Map<String, Integer>>> iterator = dataMap.entrySet().iterator();
+        while (iterator.hasNext()){
+            Map.Entry<ExecutionStatus, Map<String, Integer>> next = iterator.next();
+            CommandStateCount commandStateCount = new CommandStateCount(next.getValue().get("errorCommandState"),
+                    next.getValue().get("commandState"),next.getKey());
+            list.add(commandStateCount);
+        }
+
+        result.put(Constants.DATA_LIST, list);
         putMsg(result, Status.SUCCESS);
         return  result;
     }