From e16302f9d475c7cc4dd18c5abf1a23cb5502e362 Mon Sep 17 00:00:00 2001
From: whyclxw <810412026@qq.com>
Date: 星期三, 28 五月 2025 14:57:56 +0800
Subject: [PATCH] 密码验证加-

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

diff --git a/src/main/java/com/whyc/webSocket/MyFlowWebSocket.java b/src/main/java/com/whyc/webSocket/MyFlowWebSocket.java
index 91ab73f..a78d5e8 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
@@ -50,7 +59,7 @@
         MyFlowMainDTO myFlowMain = ActionUtil.getGson().fromJson(message, MyFlowMainDTO.class);
         UserInf user = (UserInf) this.httpSession.getAttribute("user");
         final int uId = user.getUId().intValue();
-        //final int userId = 1087;
+        //final int uId = 1086;
         thread = new Thread("Thread_myFlow") {
             @Override
             public void run() {
@@ -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, myFlowMain.getType());
+                                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