From 4cdeeb4a3656a35c4b91cbcdd99d8b7c679f5e84 Mon Sep 17 00:00:00 2001
From: 648540858 <648540858@qq.com>
Date: 星期二, 17 十一月 2020 15:27:13 +0800
Subject: [PATCH] 修复一些日志设置

---
 src/main/java/com/genersoft/iot/vmp/web/ApiStreamController.java |   98 ++++++++++++++++++++++++++++++++++++++++++++++++-
 1 files changed, 96 insertions(+), 2 deletions(-)

diff --git a/src/main/java/com/genersoft/iot/vmp/web/ApiStreamController.java b/src/main/java/com/genersoft/iot/vmp/web/ApiStreamController.java
index 8d845b1..76df0d2 100644
--- a/src/main/java/com/genersoft/iot/vmp/web/ApiStreamController.java
+++ b/src/main/java/com/genersoft/iot/vmp/web/ApiStreamController.java
@@ -1,16 +1,19 @@
 package com.genersoft.iot.vmp.web;
 
 import com.alibaba.fastjson.JSON;
+import com.alibaba.fastjson.JSONArray;
 import com.alibaba.fastjson.JSONObject;
 import com.genersoft.iot.vmp.common.StreamInfo;
 import com.genersoft.iot.vmp.gb28181.bean.Device;
 import com.genersoft.iot.vmp.gb28181.bean.DeviceChannel;
 import com.genersoft.iot.vmp.gb28181.transmit.cmd.impl.SIPCommander;
+import com.genersoft.iot.vmp.media.zlm.ZLMRESTfulUtils;
 import com.genersoft.iot.vmp.storager.IVideoManagerStorager;
 import com.genersoft.iot.vmp.vmanager.play.PlayController;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.beans.factory.annotation.Value;
 import org.springframework.http.HttpStatus;
 import org.springframework.http.ResponseEntity;
 import org.springframework.web.bind.annotation.*;
@@ -30,6 +33,13 @@
 
     @Autowired
     private IVideoManagerStorager storager;
+
+    @Value("${media.closeWaitRTPInfo}")
+    private boolean closeWaitRTPInfo;
+
+
+    @Autowired
+    private ZLMRESTfulUtils zlmresTfulUtils;
 
     /**
      * 瀹炴椂鐩存挱 - 寮�濮嬬洿鎾�
@@ -56,28 +66,112 @@
                              @RequestParam(required = false)String timeout
 
     ){
-
+        int getEncoding = closeWaitRTPInfo?  1: 0;
         Device device = storager.queryVideoDevice(serial);
+
         if (device == null ) {
             JSONObject result = new JSONObject();
             result.put("error","device[ " + serial + " ]鏈壘鍒�");
             return result;
+        }else if (device.getOnline() == 0) {
+            JSONObject result = new JSONObject();
+            result.put("error","device[ " + code + " ]offline");
+            return result;
         }
+
         DeviceChannel deviceChannel = storager.queryChannel(serial, code);
         if (deviceChannel == null) {
             JSONObject result = new JSONObject();
             result.put("error","channel[ " + code + " ]鏈壘鍒�");
             return result;
+        }else if (deviceChannel.getStatus() == 0) {
+            JSONObject result = new JSONObject();
+            result.put("error","channel[ " + code + " ]offline");
+            return result;
         }
+
         // 鏌ヨ鏄惁宸茬粡鍦ㄦ挱鏀�
         StreamInfo streamInfo = storager.queryPlayByDevice(device.getDeviceId(), code);
-        if (streamInfo == null) streamInfo = cmder.playStreamCmd(device, code);
+        if (streamInfo == null) {
+            logger.debug("streamInfo 绛変簬null, 閲嶆柊鐐规挱");
+            streamInfo = cmder.playStreamCmd(device, code);
+        }else {
+            logger.debug("streamInfo 涓嶇瓑浜巒ull, 鍚戞祦濯掍綋鏌ヨ鏄惁姝e湪鎺ㄦ祦");
+            String streamId = String.format("%08x", Integer.parseInt(streamInfo.getSsrc())).toUpperCase();
+            JSONObject rtpInfo = zlmresTfulUtils.getRtpInfo(streamId);
+            if (rtpInfo.getBoolean("exist")) {
+                logger.debug("鍚戞祦濯掍綋鏌ヨ姝e湪鎺ㄦ祦, 鐩存帴杩斿洖: " + streamInfo.getRtsp());
+                JSONObject result = new JSONObject();
+                result.put("StreamID", streamInfo.getSsrc());
+                result.put("DeviceID", device.getDeviceId());
+                result.put("ChannelID", code);
+                result.put("ChannelName", deviceChannel.getName());
+                result.put("ChannelCustomName ", "");
+                result.put("FLV ", streamInfo.getFlv());
+                result.put("WS_FLV ", streamInfo.getWs_flv());
+                result.put("RTMP", streamInfo.getRtmp());
+                result.put("HLS", streamInfo.getHls());
+                result.put("RTSP", streamInfo.getRtsp());
+                result.put("CDN", "");
+                result.put("SnapURL", "");
+                result.put("Transport", device.getTransport());
+                result.put("StartAt", "");
+                result.put("Duration", "");
+                result.put("SourceVideoCodecName", "");
+                result.put("SourceVideoWidth", "");
+                result.put("SourceVideoHeight", "");
+                result.put("SourceVideoFrameRate", "");
+                result.put("SourceAudioCodecName", "");
+                result.put("SourceAudioSampleRate", "");
+                result.put("AudioEnable", "");
+                result.put("Ondemand", "");
+                result.put("InBytes", "");
+                result.put("InBitRate", "");
+                result.put("OutBytes", "");
+                result.put("NumOutputs", "");
+                result.put("CascadeSize", "");
+                result.put("RelaySize", "");
+                result.put("ChannelPTZType", 0);
+                return result;
+            } else {
+                logger.debug("鍚戞祦濯掍綋鏌ヨ娌℃湁鎺ㄦ祦, 閲嶆柊鐐规挱");
+                storager.stopPlay(streamInfo);
+                streamInfo = cmder.playStreamCmd(device, code);
+            }
+        }
 
         if (logger.isDebugEnabled()) {
             logger.debug(String.format("璁惧棰勮 API璋冪敤锛宒eviceId锛�%s 锛宑hannelId锛�%s",serial, code));
             logger.debug("璁惧棰勮 API璋冪敤锛宻src锛�"+streamInfo.getSsrc()+",ZLMedia streamId:"+Integer.toHexString(Integer.parseInt(streamInfo.getSsrc())));
         }
+        boolean lockFlag = true;
+        long startTime = System.currentTimeMillis();
+        while (lockFlag) {
+            try {
+                if (System.currentTimeMillis() - startTime > 10 * 1000) {
+                    storager.stopPlay(streamInfo);
+                    logger.info("鎾斁绛夊緟瓒呮椂");
+                    JSONObject result = new JSONObject();
+                    result.put("error","timeout");
+                    return result;
+                } else {
 
+                    StreamInfo streamInfoNow = storager.queryPlayByDevice(serial, code);
+                    logger.debug("姝e湪鍚戞祦濯掍綋鏌ヨ");
+                    if (streamInfoNow != null && streamInfoNow.getFlv() != null) {
+                        streamInfo = streamInfoNow;
+                        logger.debug("鍚戞祦濯掍綋鏌ヨ鍒�: " + streamInfoNow.getRtsp());
+                        lockFlag = false;
+                        continue;
+                    } else {
+                        Thread.sleep(2000);
+                        continue;
+                    }
+                }
+            } catch (InterruptedException e) {
+                e.printStackTrace();
+            }
+        }
         if(streamInfo!=null) {
             JSONObject result = new JSONObject();
             result.put("StreamID", streamInfo.getSsrc());

--
Gitblit v1.8.0