From e30e28712f7ec5a3c5514484b34bdde11d0954c3 Mon Sep 17 00:00:00 2001
From: Lawrence <1934378145@qq.com>
Date: 星期六, 24 十月 2020 15:26:11 +0800
Subject: [PATCH] 播放等待时间增加到30秒,避免部分设备(NVR)播放超时

---
 src/main/java/com/genersoft/iot/vmp/gb28181/transmit/cmd/impl/SIPCommander.java |  177 ++++++++++++++++++++++++++++++++++++++--------------------
 1 files changed, 116 insertions(+), 61 deletions(-)

diff --git a/src/main/java/com/genersoft/iot/vmp/gb28181/transmit/cmd/impl/SIPCommander.java b/src/main/java/com/genersoft/iot/vmp/gb28181/transmit/cmd/impl/SIPCommander.java
index aaf347d..e2c11cb 100644
--- a/src/main/java/com/genersoft/iot/vmp/gb28181/transmit/cmd/impl/SIPCommander.java
+++ b/src/main/java/com/genersoft/iot/vmp/gb28181/transmit/cmd/impl/SIPCommander.java
@@ -15,8 +15,15 @@
 import javax.sip.header.ViaHeader;
 import javax.sip.message.Request;
 
+import com.alibaba.fastjson.JSONObject;
+import com.genersoft.iot.vmp.common.StreamInfo;
+import com.genersoft.iot.vmp.conf.MediaServerConfig;
+import com.genersoft.iot.vmp.gb28181.bean.DeviceChannel;
+import com.genersoft.iot.vmp.media.zlm.ZLMUtils;
+import com.genersoft.iot.vmp.storager.IVideoManagerStorager;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.beans.factory.annotation.Qualifier;
+import org.springframework.beans.factory.annotation.Value;
 import org.springframework.stereotype.Component;
 
 import com.genersoft.iot.vmp.conf.SipConfig;
@@ -42,6 +49,9 @@
 	
 	@Autowired
 	private VideoStreamSessionManager streamSession;
+
+	@Autowired
+	private IVideoManagerStorager storager;
 	
 	@Autowired
 	@Qualifier(value="tcpSipProvider")
@@ -50,7 +60,15 @@
 	@Autowired
 	@Qualifier(value="udpSipProvider")
 	private SipProvider udpSipProvider;
-	
+
+	@Autowired
+	private ZLMUtils zlmUtils;
+
+	@Value("${media.rtp.enable}")
+	private boolean rtpEnable;
+
+
+
 	/**
 	 * 浜戝彴鏂瑰悜鏀炬帶鍒讹紝浣跨敤閰嶇疆鏂囦欢涓殑榛樿闀滃ご绉诲姩閫熷害
 	 * 
@@ -58,7 +76,6 @@
 	 * @param channelId  棰勮閫氶亾
 	 * @param leftRight  闀滃ご宸︾Щ鍙崇Щ 0:鍋滄 1:宸︾Щ 2:鍙崇Щ
      * @param upDown     闀滃ご涓婄Щ涓嬬Щ 0:鍋滄 1:涓婄Щ 2:涓嬬Щ
-     * @param moveSpeed  闀滃ご绉诲姩閫熷害
 	 */
 	@Override
 	public boolean ptzdirectCmd(Device device, String channelId, int leftRight, int upDown) {
@@ -164,15 +181,15 @@
 		try {
 			String cmdStr= cmdString(leftRight, upDown, inOut, moveSpeed, zoomSpeed);
 			StringBuffer ptzXml = new StringBuffer(200);
-			ptzXml.append("<?xml version=\"1.0\" ?>");
-			ptzXml.append("<Control>");
-			ptzXml.append("<CmdType>DeviceControl</CmdType>");
-			ptzXml.append("<SN>" + (int)((Math.random()*9+1)*100000) + "</SN>");
-			ptzXml.append("<DeviceID>" + channelId + "</DeviceID>");
-			ptzXml.append("<PTZCmd>" + cmdStr + "</PTZCmd>");
-			ptzXml.append("<Info>");
-			ptzXml.append("</Info>");
-			ptzXml.append("</Control>");
+			ptzXml.append("<?xml version=\"1.0\" ?>\r\n");
+			ptzXml.append("<Control>\r\n");
+			ptzXml.append("<CmdType>DeviceControl</CmdType>\r\n");
+			ptzXml.append("<SN>" + (int)((Math.random()*9+1)*100000) + "</SN>\r\n");
+			ptzXml.append("<DeviceID>" + channelId + "</DeviceID>\r\n");
+			ptzXml.append("<PTZCmd>" + cmdStr + "</PTZCmd>\r\n");
+			ptzXml.append("<Info>\r\n");
+			ptzXml.append("</Info>\r\n");
+			ptzXml.append("</Control>\r\n");
 			
 			Request request = headerProvider.createMessageRequest(device, ptzXml.toString(), "ViaPtzBranch", "FromPtzTag", "ToPtzTag");
 			
@@ -191,39 +208,62 @@
 	 * @param channelId  棰勮閫氶亾
 	 */  
 	@Override
-	public String playStreamCmd(Device device, String channelId) {
+	public StreamInfo playStreamCmd(Device device, String channelId) {
 		try {
-			
+
 			String ssrc = streamSession.createPlaySsrc();
-			String transport = device.getTransport();
+			String streamMode = device.getStreamMode().toUpperCase();
+			MediaServerConfig mediaInfo = storager.getMediaInfo();
+			String mediaPort = null;
+			// 浣跨敤鍔ㄦ�乽dp绔彛
+			if (rtpEnable) {
+				mediaPort = zlmUtils.getNewRTPPort(ssrc) + "";
+			}else {
+				mediaPort = mediaInfo.getRtpProxyPort();
+			}
 			//
 			StringBuffer content = new StringBuffer(200);
 	        content.append("v=0\r\n");
-	        content.append("o="+channelId+" 0 0 IN IP4 "+sipConfig.getSipIp()+"\r\n");
+	        content.append("o="+channelId+" 0 0 IN IP4 "+mediaInfo.getLocalIP()+"\r\n");
 	        content.append("s=Play\r\n");
-	        content.append("c=IN IP4 "+sipConfig.getMediaIp()+"\r\n");
+	        content.append("c=IN IP4 "+mediaInfo.getLocalIP()+"\r\n");
 	        content.append("t=0 0\r\n");
-	        if("TCP".equals(transport)) {
-	        	content.append("m=video "+sipConfig.getMediaPort()+" TCP/RTP/AVP 96 98 97\r\n");
-			}
-	        if("UDP".equals(transport)) {
-	        	content.append("m=video "+sipConfig.getMediaPort()+" RTP/AVP 96 98 97\r\n");
+	        if("TCP-PASSIVE".equals(streamMode)) {
+	        	content.append("m=video "+ mediaPort +" TCP/RTP/AVP 96 98 97\r\n");
+			}else if ("TCP-ACTIVE".equals(streamMode)) {
+				content.append("m=video "+ mediaPort +" TCP/RTP/AVP 96 98 97\r\n");
+			}else if("UDP".equals(streamMode)) {
+	        	content.append("m=video "+ mediaPort +" RTP/AVP 96 98 97\r\n");
 			}
 	        content.append("a=recvonly\r\n");
 	        content.append("a=rtpmap:96 PS/90000\r\n");
 	        content.append("a=rtpmap:98 H264/90000\r\n");
 	        content.append("a=rtpmap:97 MPEG4/90000\r\n");
-	        if("TCP".equals(transport)){
-	             content.append("a=setup:passive\r\n");
-	             content.append("a=connection:new\r\n");
-	        }
+	        if("TCP-PASSIVE".equals(streamMode)){ // tcp琚姩妯″紡
+	        	content.append("a=setup:passive\r\n");
+				content.append("a=connection:new\r\n");
+	        }else if ("TCP-ACTIVE".equals(streamMode)) { // tcp涓诲姩妯″紡
+				content.append("a=setup:active\r\n");
+				content.append("a=connection:new\r\n");
+			}
 	        content.append("y="+ssrc+"\r\n");//ssrc
 	        
 	        Request request = headerProvider.createInviteRequest(device, channelId, content.toString(), null, "live", null);
 	
 	        ClientTransaction transaction = transmitRequest(device, request);
 	        streamSession.put(ssrc, transaction);
-			return ssrc;
+			DeviceChannel deviceChannel = storager.queryChannel(device.getDeviceId(), channelId);
+			if (deviceChannel != null) {
+				deviceChannel.setSsrc(ssrc);
+				storager.updateChannel(device.getDeviceId(), deviceChannel);
+			}
+
+			StreamInfo streamInfo = new StreamInfo();
+			streamInfo.setSsrc(ssrc);
+			streamInfo.setCahnnelId(channelId);
+			streamInfo.setDeviceID(device.getDeviceId());
+			storager.startPlay(streamInfo);
+			return streamInfo;
 		} catch ( SipException | ParseException | InvalidArgumentException e) {
 			e.printStackTrace();
 			return null;
@@ -241,7 +281,7 @@
 	@Override
 	public String playbackStreamCmd(Device device, String channelId, String startTime, String endTime) {
 		try {
-			
+			MediaServerConfig mediaInfo = storager.getMediaInfo();
 			String ssrc = streamSession.createPlayBackSsrc();
 			//
 			StringBuffer content = new StringBuffer(200);
@@ -249,22 +289,35 @@
 	        content.append("o="+sipConfig.getSipId()+" 0 0 IN IP4 "+sipConfig.getSipIp()+"\r\n");
 	        content.append("s=Playback\r\n");
 	        content.append("u="+channelId+":0\r\n");
-	        content.append("c=IN IP4 "+sipConfig.getMediaIp()+"\r\n");
-	        content.append("t="+DateUtil.yyyy_MM_dd_HH_mm_ssToTimestamp(startTime)+" "+DateUtil.yyyy_MM_dd_HH_mm_ssToTimestamp(endTime) +"\r\n");
-	        if(device.getTransport().equals("TCP")) {
-	        	content.append("m=video "+sipConfig.getMediaPort()+" TCP/RTP/AVP 96 98 97\r\n");
+	        content.append("c=IN IP4 "+mediaInfo.getLocalIP()+"\r\n");
+	        content.append("t="+DateUtil.yyyy_MM_dd_HH_mm_ssToTimestamp(startTime)+" "
+					+DateUtil.yyyy_MM_dd_HH_mm_ssToTimestamp(endTime) +"\r\n");
+			String mediaPort = null;
+			// 浣跨敤鍔ㄦ�乽dp绔彛
+			if (rtpEnable) {
+				mediaPort = zlmUtils.getNewRTPPort(ssrc) + "";
+			}else {
+				mediaPort = mediaInfo.getRtpProxyPort();
 			}
-	        if(device.getTransport().equals("UDP")) {
-	        	content.append("m=video "+sipConfig.getMediaPort()+" RTP/AVP 96 98 97\r\n");
+			String streamMode = device.getStreamMode().toUpperCase();
+			if("TCP-PASSIVE".equals(streamMode)) {
+				content.append("m=video "+ mediaPort +" TCP/RTP/AVP 96 98 97\r\n");
+			}else if ("TCP-ACTIVE".equals(streamMode)) {
+				content.append("m=video "+ mediaPort +" TCP/RTP/AVP 96 98 97\r\n");
+			}else if("UDP".equals(streamMode)) {
+				content.append("m=video "+ mediaPort +" RTP/AVP 96 98 97\r\n");
 			}
 	        content.append("a=recvonly\r\n");
 	        content.append("a=rtpmap:96 PS/90000\r\n");
 	        content.append("a=rtpmap:98 H264/90000\r\n");
 	        content.append("a=rtpmap:97 MPEG4/90000\r\n");
-	        if(device.getTransport().equals("TCP")){
-	             content.append("a=setup:passive\r\n");
-	             content.append("a=connection:new\r\n");
-	        }
+			if("TCP-PASSIVE".equals(streamMode)){ // tcp琚姩妯″紡
+				content.append("a=setup:passive\r\n");
+				content.append("a=connection:new\r\n");
+			}else if ("TCP-ACTIVE".equals(streamMode)) { // tcp涓诲姩妯″紡
+				content.append("a=setup:active\r\n");
+				content.append("a=connection:new\r\n");
+			}
 	        content.append("y="+ssrc+"\r\n");//ssrc
 	        
 	        Request request = headerProvider.createPlaybackInviteRequest(device, channelId, content.toString(), null, "playback", null);
@@ -272,6 +325,7 @@
 	        ClientTransaction transaction = transmitRequest(device, request);
 	        streamSession.put(ssrc, transaction);
 			return ssrc;
+
 		} catch ( SipException | ParseException | InvalidArgumentException e) {
 			e.printStackTrace();
 			return null;
@@ -281,8 +335,6 @@
 	/**
 	 * 瑙嗛娴佸仠姝�
 	 * 
-	 * @param device  瑙嗛璁惧
-	 * @param channelId  棰勮閫氶亾
 	 */
 	@Override
 	public void streamByeCmd(String ssrc) {
@@ -423,12 +475,12 @@
 	public boolean deviceInfoQuery(Device device) {
 		try {
 			StringBuffer catalogXml = new StringBuffer(200);
-			catalogXml.append("<?xml version=\"1.0\" encoding=\"GB2312\"?>");
-			catalogXml.append("<Query>");
-			catalogXml.append("<CmdType>DeviceInfo</CmdType>");
-			catalogXml.append("<SN>" + (int)((Math.random()*9+1)*100000) + "</SN>");
-			catalogXml.append("<DeviceID>" + device.getDeviceId() + "</DeviceID>");
-			catalogXml.append("</Query>");
+			catalogXml.append("<?xml version=\"1.0\" encoding=\"GB2312\"?>\r\n");
+			catalogXml.append("<Query>\r\n");
+			catalogXml.append("<CmdType>DeviceInfo</CmdType>\r\n");
+			catalogXml.append("<SN>" + (int)((Math.random()*9+1)*100000) + "</SN>\r\n");
+			catalogXml.append("<DeviceID>" + device.getDeviceId() + "</DeviceID>\r\n");
+			catalogXml.append("</Query>\r\n");
 			
 			Request request = headerProvider.createMessageRequest(device, catalogXml.toString(), "ViaDeviceInfoBranch", "FromDeviceInfoTag", "ToDeviceInfoTag");
 			transmitRequest(device, request);
@@ -447,14 +499,16 @@
 	 */ 
 	@Override
 	public boolean catalogQuery(Device device) {
+		// 娓呯┖閫氶亾
+		storager.cleanChannelsForDevice(device.getDeviceId());
 		try {
 			StringBuffer catalogXml = new StringBuffer(200);
-			catalogXml.append("<?xml version=\"1.0\" encoding=\"GB2312\"?>");
-			catalogXml.append("<Query>");
-			catalogXml.append("<CmdType>Catalog</CmdType>");
-			catalogXml.append("<SN>" + (int)((Math.random()*9+1)*100000) + "</SN>");
-			catalogXml.append("<DeviceID>" + device.getDeviceId() + "</DeviceID>");
-			catalogXml.append("</Query>");
+			catalogXml.append("<?xml version=\"1.0\" encoding=\"GB2312\"?>\r\n");
+			catalogXml.append("<Query>\r\n");
+			catalogXml.append("<CmdType>Catalog</CmdType>\r\n");
+			catalogXml.append("<SN>" + (int)((Math.random()*9+1)*100000) + "</SN>\r\n");
+			catalogXml.append("<DeviceID>" + device.getDeviceId() + "</DeviceID>\r\n");
+			catalogXml.append("</Query>\r\n");
 			
 			Request request = headerProvider.createMessageRequest(device, catalogXml.toString(), "ViaCatalogBranch", "FromCatalogTag", "ToCatalogTag");
 			transmitRequest(device, request);
@@ -477,17 +531,17 @@
 		
 		try {
 			StringBuffer recordInfoXml = new StringBuffer(200);
-			recordInfoXml.append("<?xml version=\"1.0\" encoding=\"GB2312\"?>");
-			recordInfoXml.append("<Query>");
-			recordInfoXml.append("<CmdType>RecordInfo</CmdType>");
-			recordInfoXml.append("<SN>" + (int)((Math.random()*9+1)*100000) + "</SN>");
-			recordInfoXml.append("<DeviceID>" + channelId + "</DeviceID>");
-			recordInfoXml.append("<StartTime>" + DateUtil.yyyy_MM_dd_HH_mm_ssToISO8601(startTime) + "</StartTime>");
-			recordInfoXml.append("<EndTime>" + DateUtil.yyyy_MM_dd_HH_mm_ssToISO8601(endTime) + "</EndTime>");
-			recordInfoXml.append("<Secrecy>0</Secrecy>");
+			recordInfoXml.append("<?xml version=\"1.0\" encoding=\"GB2312\"?>\r\n");
+			recordInfoXml.append("<Query>\r\n");
+			recordInfoXml.append("<CmdType>RecordInfo</CmdType>\r\n");
+			recordInfoXml.append("<SN>" + (int)((Math.random()*9+1)*100000) + "</SN>\r\n");
+			recordInfoXml.append("<DeviceID>" + channelId + "</DeviceID>\r\n");
+			recordInfoXml.append("<StartTime>" + DateUtil.yyyy_MM_dd_HH_mm_ssToISO8601(startTime) + "</StartTime>\r\n");
+			recordInfoXml.append("<EndTime>" + DateUtil.yyyy_MM_dd_HH_mm_ssToISO8601(endTime) + "</EndTime>\r\n");
+			recordInfoXml.append("<Secrecy>0</Secrecy>\r\n");
 			// 澶у崕NVR瑕佹眰蹇呴』澧炲姞涓�涓�间负all鐨勬枃鏈厓绱犺妭鐐筎ype
-			recordInfoXml.append("<Type>all</Type>");
-			recordInfoXml.append("</Query>");
+			recordInfoXml.append("<Type>all</Type>\r\n");
+			recordInfoXml.append("</Query>\r\n");
 			
 			Request request = headerProvider.createMessageRequest(device, recordInfoXml.toString(), "ViaRecordInfoBranch", "FromRecordInfoTag", "ToRecordInfoTag");
 			transmitRequest(device, request);
@@ -553,4 +607,5 @@
 		return clientTransaction;
 	}
 
+
 }

--
Gitblit v1.8.0