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/src/main/java/com/ard/work/component/websocket/handler/TyMessageHandler.java | 27 ++++++++++++++++++++++-----
1 files changed, 22 insertions(+), 5 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 ca41681..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) {
@@ -192,8 +196,8 @@
}
/**
- * 处理无人机控制
- * 根据接收到的消息内容，决定无人机控制信息的订阅或退订
+ * 处理无人机控制者
+ * 根据接收到的消息内容，决定无人机控制者信息的订阅或退订
*
* @param payload 消息负载，包含控制信息
* @param userId 用户ID，用于标识用户
@@ -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