From 9c4b29fdf79de231ebbe15668ad40f6e229f8109 Mon Sep 17 00:00:00 2001
From: liusuyi <1951119284@qq.com>
Date: Thu, 13 Nov 2025 16:40:00 +0800
Subject: [PATCH] 增加websocket连接关闭延迟10秒删除token
---
ard-work/pom.xml | 6 +++++-
ard-work/src/main/java/com/ard/work/component/websocket/handler/TyMessageHandler.java | 23 ++++++++++++++++++++---
ard-work/src/main/java/com/ard/work/component/websocket/WebSocketSessionManager.java | 5 +++++
ard-framework/src/main/java/com/ard/framework/web/service/TokenService.java | 8 ++++++++
4 files changed, 38 insertions(+), 4 deletions(-)
diff --git a/ard-framework/src/main/java/com/ard/framework/web/service/TokenService.java b/ard-framework/src/main/java/com/ard/framework/web/service/TokenService.java
index d44ee58..a49ca2d 100644
--- a/ard-framework/src/main/java/com/ard/framework/web/service/TokenService.java
+++ b/ard-framework/src/main/java/com/ard/framework/web/service/TokenService.java
@@ -248,4 +248,12 @@
{
return Constants.LOGIN_USERID_KEY + userId;
}
+
+ public LoginUser getLoginUserByUserId(Long userId) {
+ String userIdKey = getUserIdKey(userId);
+ String tokenKey = redisCache.getCacheObject(userIdKey);
+ if (tokenKey == null) return null;
+ return redisCache.getCacheObject(tokenKey);
+ }
+
}
diff --git a/ard-work/pom.xml b/ard-work/pom.xml
index 24d7824..ef3f0a4 100644
--- a/ard-work/pom.xml
+++ b/ard-work/pom.xml
@@ -24,7 +24,11 @@
<groupId>com.ard</groupId>
<artifactId>ard-common</artifactId>
</dependency>
-
+ <!-- framework框架核心-->
+ <dependency>
+ <groupId>com.ard</groupId>
+ <artifactId>ard-framework</artifactId>
+ </dependency>
<!-- Forest -->
<dependency>
<groupId>com.dtflys.forest</groupId>
diff --git a/ard-work/src/main/java/com/ard/work/component/websocket/WebSocketSessionManager.java b/ard-work/src/main/java/com/ard/work/component/websocket/WebSocketSessionManager.java
index 0946889..0cb420e 100644
--- a/ard-work/src/main/java/com/ard/work/component/websocket/WebSocketSessionManager.java
+++ b/ard-work/src/main/java/com/ard/work/component/websocket/WebSocketSessionManager.java
@@ -28,4 +28,9 @@
public static int getOnlineUserCount() {
return ONLINE_USER_SESSIONS.size();
}
+
+ public static boolean containsUser(Long userId) {
+ return ONLINE_USER_SESSIONS.containsKey(userId);
+ }
+
}
\ No newline at end of file
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 b7e380a..9807872 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
@@ -2,6 +2,8 @@
import com.alibaba.fastjson2.JSON;
import com.alibaba.fastjson2.TypeReference;
+import com.ard.common.core.domain.model.LoginUser;
+import com.ard.framework.web.service.TokenService;
import com.ard.work.component.websocket.WebSocketSessionManager;
import com.ard.work.domain.enums.MessageType;
import com.ard.work.domain.message.*;
@@ -12,8 +14,8 @@
import org.springframework.web.socket.TextMessage;
import org.springframework.web.socket.WebSocketSession;
import org.springframework.web.socket.handler.TextWebSocketHandler;
-
import javax.annotation.Resource;
+import java.util.concurrent.CompletableFuture;
@Component
@@ -34,6 +36,8 @@
private UavControlManager uavControlManager;
@Resource
private UavAllManager uavAllManager;
+ @Resource
+ private TokenService tokenService;
@Override
public void handleTextMessage(WebSocketSession session, TextMessage message) {
@@ -276,7 +280,20 @@
uavTelemetryManager.removeSubscribe(userId);
uavSignalManager.removeSubscribe(userId);
uavControlManager.removeSubscribe(userId);
+
+ // 延迟10秒删除token（若期间没重连）
+ CompletableFuture.runAsync(() -> {
+ try {
+ Thread.sleep(10_000); // 延迟10秒
+ } catch (InterruptedException ignored) {}
+
+ if (!WebSocketSessionManager.containsUser(userId)) {
+ LoginUser loginUser = tokenService.getLoginUserByUserId(userId);
+ if (loginUser != null) {
+ tokenService.delLoginUser(loginUser.getToken(), userId);
+ log.info("【cmd】用户 [{}] 超时未重连，已删除 token", userId);
+ }
+ }
+ });
}
-
-
}
--
Gitblit v1.9.3