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 ++++++++++++++++++++++++++++++++
 src/main/java/com/genersoft/iot/vmp/gb28181/transmit/request/impl/RegisterRequestProcessor.java |   24 ++++---
 pom.xml                                                                                         |   14 ----
 src/main/resources/application.yml                                                              |   12 +++
 4 files changed, 120 insertions(+), 28 deletions(-)

diff --git a/pom.xml b/pom.xml
index dba80cf..86b4c17 100644
--- a/pom.xml
+++ b/pom.xml
@@ -169,8 +169,8 @@
     </dependencies>
 	
 	<build>
+		<finalName>wvp-2.5.8</finalName>
 		<plugins>
-
 			<plugin>
 				<groupId>org.springframework.boot</groupId>
 				<artifactId>spring-boot-maven-plugin</artifactId>
@@ -184,18 +184,6 @@
 				<configuration>
 					<source>1.8</source>
 					<target>1.8</target>
-					<!-- 
-					<webResources>
-						<resource>
-							<directory>${project.basedir}/libs</directory>
-							<targetPath>WEB-INF/lib</targetPath>
-							<filtering>true</filtering>
-							<includes>
-								<include>**/*.jar</include>
-							</includes>
-						</resource>
-					</webResources>
-					 -->
 				</configuration>
 			</plugin>
 
diff --git a/src/main/java/com/genersoft/iot/vmp/gb28181/transmit/request/impl/RegisterRequestProcessor.java b/src/main/java/com/genersoft/iot/vmp/gb28181/transmit/request/impl/RegisterRequestProcessor.java
index 276127f..be076bd 100644
--- a/src/main/java/com/genersoft/iot/vmp/gb28181/transmit/request/impl/RegisterRequestProcessor.java
+++ b/src/main/java/com/genersoft/iot/vmp/gb28181/transmit/request/impl/RegisterRequestProcessor.java
@@ -16,6 +16,8 @@
 import javax.sip.message.Request;
 import javax.sip.message.Response;
 
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
 import org.springframework.util.StringUtils;
 
 import com.genersoft.iot.vmp.common.VideoManagerConstants;
@@ -39,6 +41,8 @@
  */
 public class RegisterRequestProcessor extends SIPRequestAbstractProcessor {
 
+	private Logger logger = LoggerFactory.getLogger(RegisterRequestProcessor.class);
+
 	private SipConfig sipConfig;
 	
 	private RegisterLogicHandler handler;
@@ -46,17 +50,15 @@
 	private IVideoManagerStorager storager;
 	
 	private EventPublisher publisher;
-	
-	/***
+
+	/**
 	 * 鏀跺埌娉ㄥ唽璇锋眰 澶勭悊
-	 * 
-	 * @param request
-	 *            璇锋眰娑堟伅
-	 */ 
+ 	 * @param evt
+	 */
 	@Override
 	public void process(RequestEvent evt) {
 		try {
-			System.out.println("鏀跺埌娉ㄥ唽璇锋眰锛屽紑濮嬪鐞�");
+			logger.info("鏀跺埌娉ㄥ唽璇锋眰锛屽紑濮嬪鐞�");
 			Request request = evt.getRequest();
 
 			Response response = null; 
@@ -75,9 +77,9 @@
 			if (authorhead == null || !passwordCorrect) {
 				
 				if (authorhead == null) {
-					System.out.println("鏈惡甯︽巿鏉冨ご 鍥炲401");
+					logger.info("鏈惡甯︽巿鏉冨ご 鍥炲401");
 				} else if (!passwordCorrect) {
-					System.out.println("瀵嗙爜閿欒 鍥炲401");
+					logger.info("瀵嗙爜閿欒 鍥炲401");
 				}
 				response = getMessageFactory().createResponse(Response.UNAUTHORIZED, request);
 				new DigestServerAuthenticationHelper().generateChallenge(getHeaderFactory(), response, sipConfig.getSipDomain());
@@ -138,12 +140,12 @@
 			// 淇濆瓨鍒皉edis
 			// 涓嬪彂catelog鏌ヨ鐩綍
 			if (registerFlag == 1 && device != null) {
-				System.out.println("娉ㄥ唽鎴愬姛! deviceId:" + device.getDeviceId());
+				logger.info("娉ㄥ唽鎴愬姛! deviceId:" + device.getDeviceId());
 				storager.updateDevice(device);
 				publisher.onlineEventPublish(device.getDeviceId(), VideoManagerConstants.EVENT_ONLINE_REGISTER);
 				handler.onRegister(device);
 			} else if (registerFlag == 2) {
-				System.out.println("娉ㄩ攢鎴愬姛! deviceId:" + device.getDeviceId());
+				logger.info("娉ㄩ攢鎴愬姛! deviceId:" + device.getDeviceId());
 				publisher.outlineEventPublish(device.getDeviceId(), VideoManagerConstants.EVENT_OUTLINE_UNREGISTER);
 			}
 		} catch (SipException | InvalidArgumentException | NoSuchAlgorithmException | ParseException e) {
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());
diff --git a/src/main/resources/application.yml b/src/main/resources/application.yml
index 5a752e5..5e6c1fc 100644
--- a/src/main/resources/application.yml
+++ b/src/main/resources/application.yml
@@ -49,5 +49,13 @@
     rtp: # 鍚敤udp澶氱鍙fā寮�
         enable: true
         udpPortRange: 30000,30500 # 绔彛鑼冨洿
-
-
+logging:
+    file:
+        name: logs/wvp.log
+        max-history: 30
+        max-size: 10MB
+        total-size-cap:300MB
+    level:
+        com:
+            genersoft:
+                iot: debug
\ No newline at end of file

--
Gitblit v1.8.0