lighter 2 年之前
父節點
當前提交
f5ee49b448
共有 1 個文件被更改,包括 10 次插入1 次删除
  1. 10 1
      src/main/java/thyyxxk/webserver/websocket/WebSocketServer.java

+ 10 - 1
src/main/java/thyyxxk/webserver/websocket/WebSocketServer.java

@@ -118,7 +118,16 @@ public class WebSocketServer {
     public static int sendMessage(String sid, String message) {
         Session session = SESSION_MAP.get(sid);
         if (null == session) {
-            return ExceptionEnum.NULL_POINTER.getCode();
+            String userid = sid.split("-")[0];
+            for (Map.Entry<String, Session> entry : SESSION_MAP.entrySet()) {
+                if (entry.getKey().startsWith(userid)) {
+                    session = entry.getValue();
+                    break;
+                }
+            }
+            if (null == session) {
+                return ExceptionEnum.NULL_POINTER.getCode();
+            }
         }
         try {
             session.getBasicRemote().sendText(message);