From 20a3cb9e73d16bd08d99780b91ca0e8d8722a0f9 Mon Sep 17 00:00:00 2001
From: 648540858 <648540858@qq.com>
Date: 星期五, 10 十二月 2021 11:28:25 +0800
Subject: [PATCH] 启用失败的流转发如果开启无人观看自动移除则不保存在数据库

---
 src/main/java/com/genersoft/iot/vmp/service/impl/StreamPushServiceImpl.java |   88 +++++++++++++++++++++++++------------------
 1 files changed, 51 insertions(+), 37 deletions(-)

diff --git a/src/main/java/com/genersoft/iot/vmp/service/impl/StreamPushServiceImpl.java b/src/main/java/com/genersoft/iot/vmp/service/impl/StreamPushServiceImpl.java
index dcca0e5..63596b2 100644
--- a/src/main/java/com/genersoft/iot/vmp/service/impl/StreamPushServiceImpl.java
+++ b/src/main/java/com/genersoft/iot/vmp/service/impl/StreamPushServiceImpl.java
@@ -154,47 +154,61 @@
         if (mediaServerItem == null) {
             return;
         }
+        // 鏁版嵁搴撹褰�
         List<StreamPushItem> pushList = getPushList(mediaServerId);
+        Map<String, StreamPushItem> pushItemMap = new HashMap<>();
+        // redis璁板綍
+        List<StreamInfo> streamInfoPushList = redisCatchStorage.getStreams(mediaServerId, "PUSH");
+        Map<String, StreamInfo> streamInfoPushItemMap = new HashMap<>();
         if (pushList.size() > 0) {
-            Map<String, StreamPushItem> pushItemMap = new HashMap<>();
             for (StreamPushItem streamPushItem : pushList) {
                 pushItemMap.put(streamPushItem.getApp() + streamPushItem.getStream(), streamPushItem);
             }
-            zlmresTfulUtils.getMediaList(mediaServerItem, (mediaList ->{
-                if (mediaList == null) return;
-                String dataStr = mediaList.getString("data");
-
-                Integer code = mediaList.getInteger("code");
-                List<StreamPushItem> streamPushItems = null;
-                if (code == 0 ) {
-                    if (dataStr != null) {
-                        streamPushItems = handleJSON(dataStr, mediaServerItem);
-                    }
-                }
-
-                if (streamPushItems != null) {
-                    for (StreamPushItem streamPushItem : streamPushItems) {
-                        pushItemMap.remove(streamPushItem.getApp() + streamPushItem.getStream());
-                    }
-                }
-                Collection<StreamPushItem> offlinePushItems = pushItemMap.values();
-                if (offlinePushItems.size() > 0) {
-                    String type = "PUSH";
-                    streamPushMapper.delAll(new ArrayList<>(offlinePushItems));
-                    for (StreamPushItem offlinePushItem : offlinePushItems) {
-                        JSONObject jsonObject = new JSONObject();
-                        jsonObject.put("serverId", userSetup.getServerId());
-                        jsonObject.put("app", offlinePushItem.getApp());
-                        jsonObject.put("stream", offlinePushItem.getStream());
-                        jsonObject.put("register", false);
-                        jsonObject.put("mediaServerId", mediaServerId);
-                        redisCatchStorage.sendStreamChangeMsg(type, jsonObject);
-                        // 绉婚櫎redis鍐呮祦鐨勪俊鎭�
-                        redisCatchStorage.removeStream(mediaServerItem.getId(), "PUSH", offlinePushItem.getApp(), offlinePushItem.getStream());
-                    }
-                }
-            }));
         }
+        if (streamInfoPushList.size() > 0) {
+            for (StreamInfo streamInfo : streamInfoPushList) {
+                streamInfoPushItemMap.put(streamInfo.getApp() + streamInfo.getStreamId(), streamInfo);
+            }
+        }
+        zlmresTfulUtils.getMediaList(mediaServerItem, (mediaList ->{
+            if (mediaList == null) return;
+            String dataStr = mediaList.getString("data");
+
+            Integer code = mediaList.getInteger("code");
+            List<StreamPushItem> streamPushItems = null;
+            if (code == 0 ) {
+                if (dataStr != null) {
+                    streamPushItems = handleJSON(dataStr, mediaServerItem);
+                }
+            }
+
+            if (streamPushItems != null) {
+                for (StreamPushItem streamPushItem : streamPushItems) {
+                    pushItemMap.remove(streamPushItem.getApp() + streamPushItem.getStream());
+                    streamInfoPushItemMap.remove(streamPushItem.getApp() + streamPushItem.getStream());
+                }
+            }
+            Collection<StreamPushItem> offlinePushItems = pushItemMap.values();
+            if (offlinePushItems.size() > 0) {
+                String type = "PUSH";
+                streamPushMapper.delAll(new ArrayList<>(offlinePushItems));
+            }
+            Collection<StreamInfo> offlineStreamInfoItems = streamInfoPushItemMap.values();
+            if (offlineStreamInfoItems.size() > 0) {
+                String type = "PUSH";
+                for (StreamInfo offlineStreamInfoItem : offlineStreamInfoItems) {
+                    JSONObject jsonObject = new JSONObject();
+                    jsonObject.put("serverId", userSetup.getServerId());
+                    jsonObject.put("app", offlineStreamInfoItem.getApp());
+                    jsonObject.put("stream", offlineStreamInfoItem.getStreamId());
+                    jsonObject.put("register", false);
+                    jsonObject.put("mediaServerId", mediaServerId);
+                    redisCatchStorage.sendStreamChangeMsg(type, jsonObject);
+                    // 绉婚櫎redis鍐呮祦鐨勪俊鎭�
+                    redisCatchStorage.removeStream(mediaServerItem.getId(), "PUSH", offlineStreamInfoItem.getApp(), offlineStreamInfoItem.getStreamId());
+                }
+            }
+        }));
     }
 
     @Override
@@ -211,6 +225,8 @@
         List<StreamInfo> streamInfoList = redisCatchStorage.getStreams(mediaServerId, type);
         if (streamInfoList.size() > 0) {
             for (StreamInfo streamInfo : streamInfoList) {
+                // 绉婚櫎redis鍐呮祦鐨勪俊鎭�
+                redisCatchStorage.removeStream(mediaServerId, type, streamInfo.getApp(), streamInfo.getStreamId());
                 JSONObject jsonObject = new JSONObject();
                 jsonObject.put("serverId", userSetup.getServerId());
                 jsonObject.put("app", streamInfo.getApp());
@@ -218,8 +234,6 @@
                 jsonObject.put("register", false);
                 jsonObject.put("mediaServerId", mediaServerId);
                 redisCatchStorage.sendStreamChangeMsg(type, jsonObject);
-                // 绉婚櫎redis鍐呮祦鐨勪俊鎭�
-                redisCatchStorage.removeStream(mediaServerId, type, streamInfo.getApp(), streamInfo.getStreamId());
             }
         }
     }

--
Gitblit v1.8.0