From 5578c76b1a32f1b1c0b99c50573d8d0246205d0b Mon Sep 17 00:00:00 2001
From: liusuyi <1951119284@qq.com>
Date: Fri, 21 Nov 2025 09:42:14 +0800
Subject: [PATCH] 优化

---
 ard-sample/src/main/java/com/ard/sample/component/websocket/handler/DockWebSocketHandler.java |   20 ++++++++++++--------
 1 files changed, 12 insertions(+), 8 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 89ccdd4..e682685 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
@@ -5,7 +5,6 @@
 import com.ard.sample.component.websocket.model.Payload;
 import com.ard.sample.component.websocket.model.WebSocketMessage;
 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;
@@ -87,7 +86,6 @@
     private void handleSubscription(String userId, String type, Payload payload) {
         String sn = payload.getSn();
         boolean subscribe = payload.isSubscribe();
-        String videoType = payload.getVideotype();
         String subscriptionKey = getSubscriptionKey(sn, type); // 生成唯一订阅标识
 
         if (subscribe) {
@@ -99,14 +97,15 @@
                 // 开启直播
                 // 开启直播逻辑
                 System.out.println("开启直播");
-                _liveStreamService.StartLiveStreaming(sn, videoType);
+                _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);
@@ -140,8 +139,13 @@
      */
     private void removeUserAllSubscriptions(String userId) {
         for (Map.Entry<String, Set<String>> entry : subscriptions.entrySet()) {
+            // 停止直播逻辑
             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());
             }
         }

--
Gitblit v1.9.3