2025-11-6 zwy 调整判断已经开启直播的情况下，关闭直播重新开启。

zhaowenyu

2025-11-06 fc6ca367242f0893b295907c3eaa08f035ae78ef

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

@@ -1,8 +1,11 @@
package com.ard.sample.component.websocket.handler;

import com.alibaba.fastjson2.JSON;
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 org.checkerframework.checker.units.qual.A;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import org.springframework.web.socket.CloseStatus;
import org.springframework.web.socket.TextMessage;
@@ -15,6 +18,9 @@

@Component
public class DockWebSocketHandler extends TextWebSocketHandler {

    @Autowired
    private LiveStreamHandler _liveStreamService;

    // 存储用户会话：userId -> WebSocketSession
    private final Map<String, WebSocketSession> userSessions = new ConcurrentHashMap<>();
@@ -58,11 +64,11 @@
            Payload payload = webSocketMessage.getPayload();

            // 验证消息类型是否合法
            if (!Arrays.asList("status", "control", "progress").contains(type)) {
            //if (!Arrays.asList("status", "control", "progress", "state").contains(type)) {
            if(!MessageType.fromString(type)){
                session.sendMessage(new TextMessage("不支持的消息类型：" + type));
                return;
            }

            // 处理订阅/取消订阅
            handleSubscription(userId, type, payload);

@@ -84,7 +90,18 @@
            // 订阅操作
            subscriptions.computeIfAbsent(subscriptionKey, k -> ConcurrentHashMap.newKeySet()).add(userId);
            System.out.println("用户[" + userId + "]订阅了 设备[" + sn + "]的[" + type + "]类型消息");
            if(MessageType.LIVESTREAM.getType().equals(type)){
                // 开启直播
                // 开启直播逻辑
                System.out.println("开启直播");
                _liveStreamService.StartLiveStreaming(sn);
            }
        } else {
            // 停止直播
            if(MessageType.LIVESTREAM.getType().equals(type)){
                System.out.println("停止直播");
                _liveStreamService.StopLiveStreaming(sn);
            }
            // 取消订阅操作
            if (subscriptions.containsKey(subscriptionKey)) {
                subscriptions.get(subscriptionKey).remove(userId);