From 56859d09df8d4226882d43934acf32d60a3b51d7 Mon Sep 17 00:00:00 2001
From: panlinlin <648540858@qq.com>
Date: 星期二, 30 三月 2021 18:46:34 +0800
Subject: [PATCH] 添加推流列表和拉流代理,下一步与国标关联

---
 src/main/java/com/genersoft/iot/vmp/media/zlm/ZLMRunner.java |   55 +++++++++++++++++++++++++++++++++++++++++++++----------
 1 files changed, 45 insertions(+), 10 deletions(-)

diff --git a/src/main/java/com/genersoft/iot/vmp/media/zlm/ZLMRunner.java b/src/main/java/com/genersoft/iot/vmp/media/zlm/ZLMRunner.java
index e65fae0..15fa957 100644
--- a/src/main/java/com/genersoft/iot/vmp/media/zlm/ZLMRunner.java
+++ b/src/main/java/com/genersoft/iot/vmp/media/zlm/ZLMRunner.java
@@ -4,8 +4,11 @@
 import com.alibaba.fastjson.JSONArray;
 import com.alibaba.fastjson.JSONObject;
 import com.genersoft.iot.vmp.conf.MediaServerConfig;
+import com.genersoft.iot.vmp.media.zlm.dto.StreamProxyDto;
 import com.genersoft.iot.vmp.storager.IRedisCatchStorage;
 //import com.genersoft.iot.vmp.storager.IVideoManagerStorager;
+import com.genersoft.iot.vmp.storager.IVideoManagerStorager;
+import com.genersoft.iot.vmp.vmanager.service.IStreamProxyService;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -16,6 +19,7 @@
 import org.springframework.util.StringUtils;
 
 import java.util.HashMap;
+import java.util.List;
 import java.util.Map;
 
 @Component
@@ -24,8 +28,8 @@
 
     private final static Logger logger = LoggerFactory.getLogger(ZLMRunner.class);
 
-    // @Autowired
-    // private IVideoManagerStorager storager;
+     @Autowired
+     private IVideoManagerStorager storager;
 
     @Autowired
     private IRedisCatchStorage redisCatchStorage;
@@ -63,18 +67,27 @@
     @Autowired
     private ZLMMediaListManager zlmMediaListManager;
 
+    @Autowired
+    private ZLMHttpHookSubscribe hookSubscribe;
+
+    @Autowired
+    private IStreamProxyService streamProxyService;
+
     @Override
     public void run(String... strings) throws Exception {
+        JSONObject subscribeKey = new JSONObject();
+        // 璁㈤槄 zlm鍚姩浜嬩欢
+        hookSubscribe.addSubscribe(ZLMHttpHookSubscribe.HookType.on_server_started,subscribeKey,(response)->{
+            MediaServerConfig mediaServerConfig = JSONObject.toJavaObject(response, MediaServerConfig.class);
+            zLmRunning(mediaServerConfig);
+        });
+
         // 鑾峰彇zlm淇℃伅
         logger.info("绛夊緟zlm鎺ュ叆...");
         MediaServerConfig mediaServerConfig = getMediaServerConfig();
+
         if (mediaServerConfig != null) {
-            logger.info("zlm鎺ュ叆鎴愬姛...");
-            if (autoConfig) saveZLMConfig();
-            mediaServerConfig = getMediaServerConfig();
-            redisCatchStorage.updateMediaInfo(mediaServerConfig);
-            // 鏇存柊娴佸垪琛�
-            zlmMediaListManager.updateMediaList();
+            zLmRunning(mediaServerConfig);
         }
     }
 
@@ -85,8 +98,7 @@
             JSONArray data = responseJSON.getJSONArray("data");
             if (data != null && data.size() > 0) {
                 mediaServerConfig = JSON.parseObject(JSON.toJSONString(data.get(0)), MediaServerConfig.class);
-                mediaServerConfig.setLocalIP(mediaIp);
-                mediaServerConfig.setWanIp(StringUtils.isEmpty(mediaWanIp)? mediaIp: mediaWanIp);
+
             }
         } else {
             logger.error("getMediaServerConfig澶辫触, 1s鍚庨噸璇�");
@@ -136,4 +148,27 @@
         }
     }
 
+    /**
+     * zlm 杩炴帴鎴愬姛鎴栬�厇lm閲嶅惎鍚�
+     */
+    private void zLmRunning(MediaServerConfig mediaServerConfig){
+        logger.info("zlm鎺ュ叆鎴愬姛...");
+        if (autoConfig) saveZLMConfig();
+        MediaServerConfig mediaInfo = redisCatchStorage.getMediaInfo();
+        if (System.currentTimeMillis() - mediaInfo.getUpdateTime() < 50){
+            logger.info("zlm鍒氬垰鏇存柊锛屽拷鐣ヨ繖娆℃洿鏂�");
+            return;
+        }
+        mediaServerConfig.setLocalIP(mediaIp);
+        mediaServerConfig.setWanIp(StringUtils.isEmpty(mediaWanIp)? mediaIp: mediaWanIp);
+        redisCatchStorage.updateMediaInfo(mediaServerConfig);
+        // 鏇存柊娴佸垪琛�
+        zlmMediaListManager.updateMediaList();
+        // 鎭㈠娴佷唬鐞�
+        List<StreamProxyDto> streamProxyListForEnable = storager.getStreamProxyListForEnable(true);
+        for (StreamProxyDto streamProxyDto : streamProxyListForEnable) {
+            logger.info("鎭㈠娴佷唬鐞嗭紝" + streamProxyDto.getApp() + "/" + streamProxyDto.getStream());
+            streamProxyService.addStreamProxyToZlm(streamProxyDto);
+        }
+    }
 }

--
Gitblit v1.8.0