From cc432012af44079537a4e380d85f278fff4ae08d Mon Sep 17 00:00:00 2001
From: whycxzp <perryhsu@163.com>
Date: 星期三, 30 八月 2023 15:23:55 +0800
Subject: [PATCH] 更新流程统计

---
 src/main/java/com/whyc/webSocket/MyFlowWebSocket.java |   21 ++++++++++++++++++++-
 1 files changed, 20 insertions(+), 1 deletions(-)

diff --git a/src/main/java/com/whyc/webSocket/MyFlowWebSocket.java b/src/main/java/com/whyc/webSocket/MyFlowWebSocket.java
index 0628e13..87ee56b 100644
--- a/src/main/java/com/whyc/webSocket/MyFlowWebSocket.java
+++ b/src/main/java/com/whyc/webSocket/MyFlowWebSocket.java
@@ -4,7 +4,9 @@
 import com.whyc.dto.MyFlowMainDTO;
 import com.whyc.dto.Response;
 import com.whyc.pojo.UserInf;
+import com.whyc.pojo.WorkflowMain;
 import com.whyc.service.WorkflowLinkService;
+import com.whyc.service.WorkflowMainService;
 import com.whyc.util.ActionUtil;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Component;
@@ -30,6 +32,8 @@
 
     private static WorkflowLinkService service;
 
+    private static WorkflowMainService mainService;
+
     private Session session;
 
     private HttpSession httpSession;
@@ -37,6 +41,11 @@
     @Autowired
     public void setService(WorkflowLinkService service) {
         MyFlowWebSocket.service = service;
+    }
+
+    @Autowired
+    public void setMainService(WorkflowMainService mainService) {
+        MyFlowWebSocket.mainService = mainService;
     }
 
     @OnOpen
@@ -59,7 +68,17 @@
                     threadFlagMap.put(thread.getId(), true);
                     try {
                         if (session.isOpen()) {
-                            Response response = service.getFlowMainByUid(uId, myFlowMain.getType(), myFlowMain.getStatus(), myFlowMain.getPageNum(), myFlowMain.getPageSize());
+                            Response response = new Response();
+                            try {
+                                Map<String, Response> resMap = new HashMap<>();
+                                Response resInfo = service.getFlowMainByUid(uId, myFlowMain.getType(), myFlowMain.getStatus(), myFlowMain.getPageNum(), myFlowMain.getPageSize());
+                                resMap.put("info", resInfo);
+                                Response<Map<Integer, Integer>> ownStatistics = mainService.getOwnStatistics2(uId, 3);
+                                resMap.put("statistics", ownStatistics);
+                                response.set(1,resMap);
+                            }catch (Exception e){
+                                response.set(0,"鍙戠敓寮傚父:"+e.toString());
+                            }
                             //鎺ㄩ�佷俊鎭�
                             synchronized (session) {
                                 session.getBasicRemote().sendObject(response);

--
Gitblit v1.9.1