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);