From 5f3a2e98c54ce815f270d149872b2b3d0d008ff0 Mon Sep 17 00:00:00 2001
From: 懒羊羊 <15939171744@163.com>
Date: 星期六, 13 一月 2024 16:17:54 +0800
Subject: [PATCH] 接口websocket

---
 jcdm-framework/src/main/java/com/jcdm/framework/websocket/WebSocketServer.java |   13 +++++++++----
 1 files changed, 9 insertions(+), 4 deletions(-)

diff --git a/jcdm-framework/src/main/java/com/jcdm/framework/websocket/WebSocketServer.java b/jcdm-framework/src/main/java/com/jcdm/framework/websocket/WebSocketServer.java
index b2a1b6a..57921d7 100644
--- a/jcdm-framework/src/main/java/com/jcdm/framework/websocket/WebSocketServer.java
+++ b/jcdm-framework/src/main/java/com/jcdm/framework/websocket/WebSocketServer.java
@@ -6,6 +6,7 @@
 import javax.websocket.OnMessage;
 import javax.websocket.OnOpen;
 import javax.websocket.Session;
+import javax.websocket.server.PathParam;
 import javax.websocket.server.ServerEndpoint;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
@@ -17,7 +18,7 @@
  * @author ruoyi
  */
 @Component
-@ServerEndpoint("/websocket/message")
+@ServerEndpoint("/websocket/message/{userId}")
 public class WebSocketServer
 {
     /**
@@ -32,11 +33,14 @@
 
     private static Semaphore socketSemaphore = new Semaphore(socketMaxOnlineCount);
 
+    /**鎺ユ敹userId*/
+    private String userId="";
+
     /**
      * 杩炴帴寤虹珛鎴愬姛璋冪敤鐨勬柟娉�
      */
     @OnOpen
-    public void onOpen(Session session) throws Exception
+    public void onOpen(Session session, @PathParam("userId") String userId) throws Exception
     {
         boolean semaphoreFlag = false;
         // 灏濊瘯鑾峰彇淇″彿閲�
@@ -51,7 +55,8 @@
         else
         {
             // 娣诲姞鐢ㄦ埛
-            WebSocketUsers.put(session.getId(), session);
+//            WebSocketUsers.put(session.getId(), session);
+            WebSocketUsers.put(userId, session);
             LOGGER.info("\n 寤虹珛杩炴帴 - {}", session);
             LOGGER.info("\n 褰撳墠浜烘暟 - {}", WebSocketUsers.getUsers().size());
             WebSocketUsers.sendMessageToUserByText(session, "杩炴帴鎴愬姛");
@@ -98,6 +103,6 @@
     public void onMessage(String message, Session session)
     {
         String msg = message.replace("浣�", "鎴�").replace("鍚�", "");
-        WebSocketUsers.sendMessageToUserByText(session, msg);
+//        WebSocketUsers.sendMessageToUserByText(session, msg);
     }
 }

--
Gitblit v1.9.3