From a3686cfa49bf53fb91a2ceb960cf15b3ebdac641 Mon Sep 17 00:00:00 2001
From: yangys <y_ys79@sina.com>
Date: 星期四, 30 五月 2024 16:38:00 +0800
Subject: [PATCH] 测试代码

---
 smart-man-boot/src/main/java/com/qianwen/smartman/common/websocket/realtime/RealTimeDataJsonMessageHandler.java |   15 +++++++--------
 1 files changed, 7 insertions(+), 8 deletions(-)

diff --git a/smart-man-boot/src/main/java/com/qianwen/smartman/common/websocket/realtime/RealTimeDataJsonMessageHandler.java b/smart-man-boot/src/main/java/com/qianwen/smartman/common/websocket/realtime/RealTimeDataJsonMessageHandler.java
index 1dfcf47..b233ed2 100644
--- a/smart-man-boot/src/main/java/com/qianwen/smartman/common/websocket/realtime/RealTimeDataJsonMessageHandler.java
+++ b/smart-man-boot/src/main/java/com/qianwen/smartman/common/websocket/realtime/RealTimeDataJsonMessageHandler.java
@@ -2,7 +2,6 @@
 
 import java.util.List;
 import java.util.Map;
-import java.util.Set;
 import java.util.stream.Collectors;
 import com.qianwen.smartman.common.cache.cps.WorkstationCache;
 import com.qianwen.core.tool.utils.Func;
@@ -13,6 +12,9 @@
 import org.springframework.stereotype.Component;
 import org.springframework.web.socket.WebSocketSession;
 
+/**
+ * 瀹炵幇浜唚ebsocket妯″潡涓殑JsonMessageHandler锛屽疄鏃剁湅鏉夸娇鐢ㄨ澶勭悊鍣紝payload涓簲璇ユ惡甯ype:"realTimeData"灞炴��
+ */
 @Component
 public class RealTimeDataJsonMessageHandler implements JsonMessageHandler<RealTimeDataRequestJsonWebSocketMessage> {
     private final IOrderWorkstationService orderWorkstationService;
@@ -23,12 +25,8 @@
 
     public void handle(WebSocketSession session, RealTimeDataRequestJsonWebSocketMessage message) {
         List<String> workstationIdList = message.getWorkstationIdList();
-        List<OrderWorkstation> workstationInProcess = this.orderWorkstationService.getWorkstationInProcess((Set) workstationIdList.stream().map(Long::valueOf).collect(Collectors.toSet()));
-        Map<Long, String> orderWorkstationMap = (Map) workstationInProcess.stream().collect(Collectors.toMap((v0) -> {
-            return v0.getWorkstationId();
-        }, (v0) -> {
-            return v0.getOrderCode();
-        }));
+        List<OrderWorkstation> workstationInProcess = this.orderWorkstationService.getWorkstationInProcess(workstationIdList.stream().map(Long::valueOf).collect(Collectors.toSet()));
+        Map<Long, String> orderWorkstationMap = workstationInProcess.stream().collect(Collectors.toMap(OrderWorkstation::getWorkstationId, OrderWorkstation::getOrderCode));
         if (Func.isNotEmpty(workstationIdList)) {
             for (String workstationId : workstationIdList) {
                 RealTimeDaraResponseJsonWebSocketMessage jsonWebSocketMessage = new RealTimeDaraResponseJsonWebSocketMessage();
@@ -41,10 +39,11 @@
         }
     }
 
+    @Override
     public String type() {
         return "realTimeData";
     }
-
+    @Override
     public Class<RealTimeDataRequestJsonWebSocketMessage> getMessageClass() {
         return RealTimeDataRequestJsonWebSocketMessage.class;
     }

--
Gitblit v1.9.3