feat(任务中心): 统计显示执行中的任务消息

backend/src/main/java/io/metersphere/websocket/TaskCenterWebSocket.java

@@ -0,0 +1,105 @@
+package io.metersphere.websocket;
+
+import io.metersphere.commons.utils.LogUtil;
+import io.metersphere.task.dto.TaskCenterRequest;
+import io.metersphere.task.service.TaskService;
+import org.springframework.stereotype.Component;
+
+import javax.annotation.Resource;
+import javax.websocket.*;
+import javax.websocket.server.PathParam;
+import javax.websocket.server.ServerEndpoint;
+import java.util.Timer;
+import java.util.TimerTask;
+import java.util.concurrent.ConcurrentHashMap;
+
+@ServerEndpoint("/task/center/count/running/{projectId}")
+@Component
+public class TaskCenterWebSocket {
+    private static TaskService taskService;
+    private static ConcurrentHashMap<Session, Timer> refreshTasks = new ConcurrentHashMap<>();
+
+    @Resource
+    public void setTaskService(TaskService taskService) {
+        TaskCenterWebSocket.taskService = taskService;
+    }
+
+    /**
+     * 开启连接的操作
+     */
+    @OnOpen
+    public void onOpen(@PathParam("projectId") String projectId, Session session) {
+        Timer timer = new Timer(true);
+        TaskCenterWebSocket.TaskCenter task = new TaskCenterWebSocket.TaskCenter(session, projectId);
+        timer.schedule(task, 0, 10 * 1000);
+        refreshTasks.putIfAbsent(session, timer);
+    }
+
+    /**
+     * 连接关闭的操作
+     */
+    @OnClose
+    public void onClose(Session session) {
+        Timer timer = refreshTasks.get(session);
+        if (timer != null) {
+            timer.cancel();
+            refreshTasks.remove(session);
+        }
+    }
+
+    /**
+     * 推送消息
+     */
+    @OnMessage
+    public void onMessage(@PathParam("projectId") String projectId, Session session, String message) {
+        int refreshTime = 10;
+        try {
+            refreshTime = Integer.parseInt(message);
+        } catch (Exception e) {
+        }
+        try {
+            Timer timer = refreshTasks.get(session);
+            timer.cancel();
+
+            Timer newTimer = new Timer(true);
+            newTimer.schedule(new TaskCenterWebSocket.TaskCenter(session, projectId), 0, refreshTime * 1000L);
+            refreshTasks.put(session, newTimer);
+        } catch (Exception e) {
+            LogUtil.error(e.getMessage(), e);
+        }
+    }
+
+    /**
+     * 出错的操作
+     */
+    @OnError
+    public void onError(Throwable error) {
+        System.out.println(error);
+        error.printStackTrace();
+    }
+
+    public static class TaskCenter extends TimerTask {
+        private Session session;
+        private TaskCenterRequest request;
+
+        TaskCenter(Session session, String projectId) {
+            this.session = session;
+            TaskCenterRequest request = new TaskCenterRequest();
+            request.setProjectId(projectId);
+            this.request = request;
+        }
+
+        @Override
+        public void run() {
+            try {
+                int taskTotal = taskService.getRunningTasks(request).size();
+                if (!session.isOpen()) {
+                    return;
+                }
+                session.getBasicRemote().sendText(taskTotal + "");
+            } catch (Exception e) {
+                LogUtil.error(e.getMessage(), e);
+            }
+        }
+    }
+}

frontend/src/business/components/task/TaskCenter.vue

@@ -126,6 +126,32 @@ export default {
     format(item) {
       return '';
     },
+    initWebSocket() {
+      let protocol = "ws://";
+      if (window.location.protocol === 'https:') {
+        protocol = "wss://";
+      }
+      const uri = protocol + window.location.host + "/task/center/count/running/" + getCurrentProjectID();
+      this.websocket = new WebSocket(uri);
+      this.websocket.onmessage = this.onMessage;
+      this.websocket.onopen = this.onOpen;
+      this.websocket.onerror = this.onError;
+      this.websocket.onclose = this.onClose;
+    },
+    onOpen() {
+    },
+    onError(e) {
+    },
+    onMessage(e) {
+      let taskTotal = e.data;
+      this.runningTotal = taskTotal;
+    },
+    onClose(e) {
+      if (e.code === 1005) {
+        // 强制删除之后关闭socket，不用刷新report
+        return;
+      }
+    },
     showTaskCenter() {
       this.getTaskRunning();
       this.init();
@@ -207,11 +233,7 @@ export default {
       return mode;
     },
     getTaskRunning() {
-      this.condition.projectId = getCurrentProjectID();
-      this.$post('/task/center/count/running', this.condition, response => {
-        this.runningTotal = response.data;
-        setTimeout(this.getTaskRunning, 5000);
-      });
+      this.initWebSocket();
     },
     init() {
       this.result.loading = true;