admin/lx-admin-backend
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update 优化 ws模块替换session的时候关闭session连接

Browse Source
This commit is contained in:
疯狂的狮子Li 2025-01-16 11:50:19 +08:00
parent 875c9fa77c
commit 9d8b9fabbe
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
ruoyi-common/ruoyi-common-websocket/src/main/java/org/dromara/common/websocket/holder/WebSocketSessionHolder.java
View File

@ -2,6 +2,7 @@ package org.dromara.common.websocket.holder;
import lombok.AccessLevel; import lombok.AccessLevel;
import lombok.NoArgsConstructor; import lombok.NoArgsConstructor;
import org.springframework.web.socket.CloseStatus;
import org.springframework.web.socket.WebSocketSession; import org.springframework.web.socket.WebSocketSession;
import java.util.Map; import java.util.Map;
@ -25,6 +26,7 @@ public class WebSocketSessionHolder {
* @param session 要添加的WebSocket会话 * @param session 要添加的WebSocket会话
*/ */
public static void addSession(Long sessionKey, WebSocketSession session) { public static void addSession(Long sessionKey, WebSocketSession session) {
removeSession(sessionKey);
USER_SESSION_MAP.put(sessionKey, session); USER_SESSION_MAP.put(sessionKey, session);
} }
@ -34,8 +36,10 @@ public class WebSocketSessionHolder {
* @param sessionKey 要移除的会话键 * @param sessionKey 要移除的会话键
*/ */
public static void removeSession(Long sessionKey) { public static void removeSession(Long sessionKey) {
if (USER_SESSION_MAP.containsKey(sessionKey)) { WebSocketSession session = USER_SESSION_MAP.remove(sessionKey);
USER_SESSION_MAP.remove(sessionKey); try {
session.close(CloseStatus.BAD_DATA);
} catch (Exception ignored) {
} }
} }