From 8b459796efa4f1e0571f4436953a52d25a063e70 Mon Sep 17 00:00:00 2001
From: zhaowenyu <35589726@qq.com>
Date: Sat, 29 Nov 2025 09:30:11 +0800
Subject: [PATCH] 2025-11-29 zwy 在websocket断开时给5秒延时，给前端刷新缓冲
---
ard-sample/src/main/java/com/ard/sample/component/websocket/handler/DockWebSocketHandler.java | 49 +++++++++++++++++++++++++++++++++++++++++++------
1 files changed, 43 insertions(+), 6 deletions(-)
diff --git a/ard-sample/src/main/java/com/ard/sample/component/websocket/handler/DockWebSocketHandler.java b/ard-sample/src/main/java/com/ard/sample/component/websocket/handler/DockWebSocketHandler.java
index e682685..74fa790 100644
--- a/ard-sample/src/main/java/com/ard/sample/component/websocket/handler/DockWebSocketHandler.java
+++ b/ard-sample/src/main/java/com/ard/sample/component/websocket/handler/DockWebSocketHandler.java
@@ -1,9 +1,11 @@
package com.ard.sample.component.websocket.handler;
import com.alibaba.fastjson2.JSON;
+import com.ard.sample.common.model.ResponseResult;
import com.ard.sample.component.websocket.enums.MessageType;
import com.ard.sample.component.websocket.model.Payload;
import com.ard.sample.component.websocket.model.WebSocketMessage;
+import com.ard.sample.manage.service.IControlService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
@@ -15,6 +17,9 @@
import java.io.IOException;
import java.util.*;
import java.util.concurrent.ConcurrentHashMap;
+import java.util.concurrent.Executors;
+import java.util.concurrent.ScheduledExecutorService;
+import java.util.concurrent.TimeUnit;
@Slf4j
@Component
@@ -22,6 +27,9 @@
@Autowired
private LiveStreamHandler _liveStreamService;
+
+ @Autowired
+ private IControlService _controlService;
// 存储用户会话：userId -> WebSocketSession
private final Map<String, WebSocketSession> userSessions = new ConcurrentHashMap<>();
@@ -31,6 +39,13 @@
// 关键：为每个WebSocketSession分配独立的锁对象，确保写操作同步
private final Map<WebSocketSession, Object> sessionLocks = new ConcurrentHashMap<>();
+
+ // 延时任务线程池（单线程，守护线程），用于处理连接关闭后的延时清理
+ private final ScheduledExecutorService delayExecutor = Executors.newSingleThreadScheduledExecutor(runnable -> {
+ Thread thread = new Thread(runnable, "websocket-关闭-延迟-线程");
+ thread.setDaemon(true); // 设为守护线程，不阻塞JVM退出
+ return thread;
+ });
/**
* 连接建立后回调
@@ -125,12 +140,34 @@
public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception {
String userId = (String) session.getAttributes().get("userId");
if (userId != null) {
- userSessions.remove(userId);
- sessionLocks.remove(session); // 移除会话锁
- // 清除该用户的所有订阅
- removeUserAllSubscriptions(userId);
- //System.out.println("用户[" + userId + "]已断开连接，当前在线人数：" + userSessions.size());
- log.info("用户[{}]已断开连接，当前在线人数：{}", userId, userSessions.size());
+ log.info("用户[{}]连接关闭，将延时5秒执行清理操作（给前端刷新缓冲）", userId);
+
+ // 延时3秒执行清理逻辑
+ delayExecutor.schedule(() -> {
+ try {
+
+ userSessions.remove(userId);
+ sessionLocks.remove(session); // 移除会话锁
+ // 关闭远程调试模式,暂时写死sn，后续改为动态获取sn
+ // ResponseResult responseResult = _controlService.controlDock("7CTDLCE00A1UPE","debug_mode_close",null);
+ // log.info("关闭远程调试模式");
+ // if(responseResult.getCode() == ResponseResult.CODE_SUCCESS){
+ // log.info("关闭远程调试模式成功");
+ // }
+ // 清除该用户的所有订阅
+ removeUserAllSubscriptions(userId);
+ //System.out.println("用户[" + userId + "]已断开连接，当前在线人数：" + userSessions.size());
+// log.info("用户[{}]已断开连接，当前在线人数：{}", userId, userSessions.size());
+ log.info("用户[{}]延时5秒清理完成，当前在线人数：{}", userId, userSessions.size());
+ } catch (Exception e) {
+ log.error("用户[{}]延时清理操作异常", userId, e);
+ }
+ }, 5, TimeUnit.SECONDS); // 延时5秒执行
+ }
+ else {
+ // 无认证用户，直接清理会话锁（无需延时）
+ sessionLocks.remove(session);
+ log.info("无认证用户连接关闭，直接清理会话锁");
}
}
--
Gitblit v1.9.3