From d4a8855af34e11874ae2a2d06d1b02e955df723d Mon Sep 17 00:00:00 2001
From: wangzhijiang <12345678>
Date: Wed, 1 Jun 2022 10:44:51 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B5=81=E7=A8=8B=20-=20=E6=8F=90=E4=BA=A4?=
 =?UTF-8?q?=E4=BB=BB=E5=8A=A1=20=20=E5=8F=91=E9=80=81webSocket=E6=B6=88?=
 =?UTF-8?q?=E6=81=AF=E3=80=82?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../wkflow/controller/workFlowController.java      | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/jero-boot/jero-boot-modules/src/main/java/com/jero/modules/wkflow/controller/workFlowController.java b/jero-boot/jero-boot-modules/src/main/java/com/jero/modules/wkflow/controller/workFlowController.java
index c1b9d218e..704ac4a14 100644
--- a/jero-boot/jero-boot-modules/src/main/java/com/jero/modules/wkflow/controller/workFlowController.java
+++ b/jero-boot/jero-boot-modules/src/main/java/com/jero/modules/wkflow/controller/workFlowController.java
@@ -2,7 +2,9 @@ package com.jero.modules.wkflow.controller;
 
 import com.jero.common.api.vo.Result;
 import com.jero.common.aspect.annotation.AutoLog;
+import com.jero.common.constant.WebsocketConst;
 import com.jero.common.system.vo.LoginUser;
+import com.jero.modules.message.websocket.WebSocket;
 import com.jero.modules.project.entity.ConditionAssessmentEO;
 import com.jero.modules.project.enums.CurrentProjectStatusEnum;
 import com.jero.modules.project.enums.ProjectRoleEnum;
@@ -36,6 +38,9 @@ public class workFlowController {
     @Autowired
     private IConditionAssessmentEOService conditionAssessmentEOService;
 
+    @Autowired
+    private WebSocket webSocket;
+
     public Result activiti_define_start(@RequestBody JSONObject jsonObject){
         return  workFlowFeignClient.activiti_define_start(jsonObject);
     }
@@ -80,9 +85,18 @@ public class workFlowController {
         if(str.getCode() == 500){
             str.setSuccess(false);
         }
+        sendWebsocket(busMes.getTaskId(),busMes.getTaskId());
         return str;
     }
 
+    public void sendWebsocket(String msgId, String msgTet) {
+        com.alibaba.fastjson.JSONObject obj = new com.alibaba.fastjson.JSONObject();
+        obj.put(WebsocketConst.MSG_CMD, WebsocketConst.CMD_TOPIC);
+        obj.put(WebsocketConst.MSG_ID, msgId);
+        obj.put(WebsocketConst.MSG_TXT, msgTet);
+        webSocket.sendMessage(obj.toJSONString());
+    }
+
     @AutoLog(value = "通过流程实例id查询当前待办的流程实例是否结束")
     @ApiOperation(value="通过流程实例id查询当前待办的流程实例是否结束", notes="通过流程实例id查询当前待办的流程实例是否结束")
     @GetMapping("/queryTaskWhetherToEnd")