From bf8fb0c51f4e7894a62636dede6c69a2a85d8458 Mon Sep 17 00:00:00 2001
From: panlinlin <648540858@qq.com>
Date: 星期二, 27 四月 2021 15:35:22 +0800
Subject: [PATCH] 添加级联平台自动注册时失败自动重新注册

---
 src/main/java/com/genersoft/iot/vmp/media/zlm/ZLMRunner.java |   37 ++++++++++++++++++++++++-------------
 1 files changed, 24 insertions(+), 13 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 15fa957..2bbc1be 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,11 +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.media.zlm.dto.StreamProxyItem;
 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 com.genersoft.iot.vmp.service.IStreamProxyService;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -61,6 +61,9 @@
     @Value("${media.autoConfig}")
     private boolean autoConfig;
 
+    @Value("${server.ssl.enabled}")
+    private boolean sslEnabled;
+
     @Autowired
     private ZLMRESTfulUtils zlmresTfulUtils;
 
@@ -75,9 +78,8 @@
 
     @Override
     public void run(String... strings) throws Exception {
-        JSONObject subscribeKey = new JSONObject();
         // 璁㈤槄 zlm鍚姩浜嬩欢
-        hookSubscribe.addSubscribe(ZLMHttpHookSubscribe.HookType.on_server_started,subscribeKey,(response)->{
+        hookSubscribe.addSubscribe(ZLMHttpHookSubscribe.HookType.on_server_started,null,(response)->{
             MediaServerConfig mediaServerConfig = JSONObject.toJavaObject(response, MediaServerConfig.class);
             zLmRunning(mediaServerConfig);
         });
@@ -117,14 +119,14 @@
         if (StringUtils.isEmpty(mediaHookIp)) {
             mediaHookIp = sipIP;
         }
-
-        String hookPrex = String.format("http://%s:%s/index/hook", mediaHookIp, serverPort);
+        String protocol = sslEnabled ? "https" : "http";
+        String hookPrex = String.format("%s://%s:%s/index/hook", protocol, mediaHookIp, serverPort);
         Map<String, Object> param = new HashMap<>();
         param.put("api.secret",mediaSecret); // -profile:v Baseline
         param.put("ffmpeg.cmd","%s -fflags nobuffer -rtsp_transport tcp -i %s -c:a aac -strict -2 -ar 44100 -ab 48k -c:v libx264  -f flv %s");
         param.put("hook.enable","1");
         param.put("hook.on_flow_report","");
-        param.put("hook.on_play","");
+        param.put("hook.on_play",String.format("%s/on_play", hookPrex));
         param.put("hook.on_http_access","");
         param.put("hook.on_publish",String.format("%s/on_publish", hookPrex));
         param.put("hook.on_record_mp4","");
@@ -152,23 +154,32 @@
      * zlm 杩炴帴鎴愬姛鎴栬�厇lm閲嶅惎鍚�
      */
     private void zLmRunning(MediaServerConfig mediaServerConfig){
-        logger.info("zlm鎺ュ叆鎴愬姛...");
+        logger.info( "[ id: " + mediaServerConfig.getGeneralMediaServerId() + "] zlm鎺ュ叆鎴愬姛...");
         if (autoConfig) saveZLMConfig();
         MediaServerConfig mediaInfo = redisCatchStorage.getMediaInfo();
-        if (System.currentTimeMillis() - mediaInfo.getUpdateTime() < 50){
-            logger.info("zlm鍒氬垰鏇存柊锛屽拷鐣ヨ繖娆℃洿鏂�");
+        if (mediaInfo != null && System.currentTimeMillis() - mediaInfo.getUpdateTime() < 50){
+            logger.info("[ id: " + mediaServerConfig.getGeneralMediaServerId() + "]zlm鍒氬垰鏇存柊锛屽拷鐣ヨ繖娆℃洿鏂�");
             return;
         }
         mediaServerConfig.setLocalIP(mediaIp);
         mediaServerConfig.setWanIp(StringUtils.isEmpty(mediaWanIp)? mediaIp: mediaWanIp);
         redisCatchStorage.updateMediaInfo(mediaServerConfig);
+
+        // 娓呯┖鎵�鏈塻ession
+//        zlmMediaListManager.clearAllSessions();
+
         // 鏇存柊娴佸垪琛�
         zlmMediaListManager.updateMediaList();
         // 鎭㈠娴佷唬鐞�
-        List<StreamProxyDto> streamProxyListForEnable = storager.getStreamProxyListForEnable(true);
-        for (StreamProxyDto streamProxyDto : streamProxyListForEnable) {
+        List<StreamProxyItem> streamProxyListForEnable = storager.getStreamProxyListForEnable(true);
+        for (StreamProxyItem streamProxyDto : streamProxyListForEnable) {
             logger.info("鎭㈠娴佷唬鐞嗭紝" + streamProxyDto.getApp() + "/" + streamProxyDto.getStream());
-            streamProxyService.addStreamProxyToZlm(streamProxyDto);
+            JSONObject jsonObject = streamProxyService.addStreamProxyToZlm(streamProxyDto);
+            if (jsonObject == null) {
+                // 璁剧疆涓烘湭鍚敤
+                logger.info("鎭㈠娴佷唬鐞嗗け璐ワ紝璇锋鏌ユ祦鍦板潃鍚庨噸鏂板惎鐢�" + streamProxyDto.getApp() + "/" + streamProxyDto.getStream());
+                streamProxyService.stop(streamProxyDto.getApp(), streamProxyDto.getStream());
+            }
         }
     }
 }

--
Gitblit v1.8.0