2025-11-24 zwy 在连接断开时取消远程调试模式

zhaowenyu

2025-11-24 9742da1ce6dc62aba444348348e71b77e3fb3295

 ard-sample/src/main/java/com/ard/sample/component/websocket/handler/DockWebSocketHandler.java

@@ -1,11 +1,12 @@
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.checkerframework.checker.units.qual.A;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import org.springframework.web.socket.CloseStatus;
@@ -23,6 +24,9 @@

    @Autowired
    private LiveStreamHandler _liveStreamService;

    @Autowired
    private IControlService _controlService;

    // 存储用户会话：userId -> WebSocketSession
    private final Map<String, WebSocketSession> userSessions = new ConcurrentHashMap<>();
@@ -101,11 +105,12 @@
                _liveStreamService.StartLiveStreaming(sn);
            }
        } else {
            // 停止直播
            if(MessageType.LIVESTREAM.getType().equals(type)){
                System.out.println("停止直播");
                _liveStreamService.StopLiveStreaming(sn);
            }
            // 停止直播--前端不主动关注直播，则在断开连接时停止
//            if(MessageType.LIVESTREAM.getType().equals(type)){
//                System.out.println("停止直播");
//
//                _liveStreamService.StopLiveStreaming(sn);
//            }
            // 取消订阅操作
            if (subscriptions.containsKey(subscriptionKey)) {
                subscriptions.get(subscriptionKey).remove(userId);
@@ -127,6 +132,12 @@
        if (userId != null) {
            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());
@@ -139,12 +150,13 @@
     */
    private void removeUserAllSubscriptions(String userId) {
        for (Map.Entry<String, Set<String>> entry : subscriptions.entrySet()) {
            if (entry.getKey().contains(MessageType.LIVESTREAM.getType())) {
                String sn = entry.getKey().split(":")[0];
                _liveStreamService.StopLiveStreaming(sn);
            }
            // 停止直播逻辑
            entry.getValue().remove(userId);
            if (entry.getValue().isEmpty()) {
                if (entry.getKey().contains(MessageType.LIVESTREAM.getType())) {
                    String sn = entry.getKey().split(":")[0];
                    _liveStreamService.StopLiveStreaming(sn);
                }
                subscriptions.remove(entry.getKey());
            }
        }