From 937d3cfacde345b355cf30b1d0a08ea91ed0bc67 Mon Sep 17 00:00:00 2001
From: liusuyi <1951119284@qq.com>
Date: Sat, 18 Oct 2025 10:36:00 +0800
Subject: [PATCH] 优化websocket断开时用户ID为null

---
 ard-work/src/main/java/com/ard/work/component/websocket/handler/TyMessageHandler.java |    3 +++
 1 files changed, 3 insertions(+), 0 deletions(-)

diff --git a/ard-work/src/main/java/com/ard/work/component/websocket/handler/TyMessageHandler.java b/ard-work/src/main/java/com/ard/work/component/websocket/handler/TyMessageHandler.java
index f6ce332..36a3023 100644
--- a/ard-work/src/main/java/com/ard/work/component/websocket/handler/TyMessageHandler.java
+++ b/ard-work/src/main/java/com/ard/work/component/websocket/handler/TyMessageHandler.java
@@ -259,6 +259,9 @@
     public void afterConnectionClosed(WebSocketSession session, CloseStatus status) {
         log.info("【cmd】Disconnected: {}", session.getId());
         String userIdStr = (String) session.getAttributes().get("userId");
+        if (userIdStr == null || userIdStr.isBlank()) {
+            return;
+        }
         Long userId = Long.parseLong(userIdStr);
         WebSocketSessionManager.removeSession(userId);
         log.info("【cmd】size: {}", WebSocketSessionManager.getOnlineUserCount());

--
Gitblit v1.9.3