From e48fa711a3664bece9b3e58840a75fe7c05bc47c Mon Sep 17 00:00:00 2001
From: panlinlin <648540858@qq.com>
Date: 星期六, 08 五月 2021 17:14:05 +0800
Subject: [PATCH] 添加截图(快照)功能

---
 src/main/java/com/genersoft/iot/vmp/gb28181/transmit/request/impl/InviteRequestProcessor.java |  158 +++++++++++++++++++++++++++++++++++++---------------
 1 files changed, 113 insertions(+), 45 deletions(-)

diff --git a/src/main/java/com/genersoft/iot/vmp/gb28181/transmit/request/impl/InviteRequestProcessor.java b/src/main/java/com/genersoft/iot/vmp/gb28181/transmit/request/impl/InviteRequestProcessor.java
index a7b3d94..a6f0ae7 100644
--- a/src/main/java/com/genersoft/iot/vmp/gb28181/transmit/request/impl/InviteRequestProcessor.java
+++ b/src/main/java/com/genersoft/iot/vmp/gb28181/transmit/request/impl/InviteRequestProcessor.java
@@ -11,19 +11,16 @@
 import javax.sip.message.Request;
 import javax.sip.message.Response;
 
-import com.genersoft.iot.vmp.conf.MediaServerConfig;
-import com.genersoft.iot.vmp.gb28181.bean.Device;
-import com.genersoft.iot.vmp.gb28181.bean.DeviceChannel;
-import com.genersoft.iot.vmp.gb28181.bean.ParentPlatform;
-import com.genersoft.iot.vmp.gb28181.bean.SendRtpItem;
+import com.genersoft.iot.vmp.media.zlm.ZLMServerConfig;
+import com.genersoft.iot.vmp.gb28181.bean.*;
 import com.genersoft.iot.vmp.gb28181.transmit.cmd.impl.SIPCommander;
 import com.genersoft.iot.vmp.gb28181.transmit.cmd.impl.SIPCommanderFroPlatform;
 import com.genersoft.iot.vmp.gb28181.transmit.request.SIPRequestAbstractProcessor;
 import com.genersoft.iot.vmp.media.zlm.ZLMRTPServerFactory;
 import com.genersoft.iot.vmp.storager.IRedisCatchStorage;
 import com.genersoft.iot.vmp.storager.IVideoManagerStorager;
-import com.genersoft.iot.vmp.vmanager.play.bean.PlayResult;
-import com.genersoft.iot.vmp.vmanager.service.IPlayService;
+import com.genersoft.iot.vmp.vmanager.gb28181.play.bean.PlayResult;
+import com.genersoft.iot.vmp.service.IPlayService;
 import gov.nist.javax.sip.address.AddressImpl;
 import gov.nist.javax.sip.address.SipUri;
 import org.slf4j.Logger;
@@ -89,16 +86,31 @@
 			}
 
 			// 鏌ヨ璇锋眰鏂规槸鍚︿笂绾у钩鍙�
-			ParentPlatform platform = storager.queryParentPlatById(requesterId);
+			ParentPlatform platform = storager.queryParentPlatByServerGBId(requesterId);
 			if (platform != null) {
 				// 鏌ヨ骞冲彴涓嬫槸鍚︽湁璇ラ�氶亾
 				DeviceChannel channel = storager.queryChannelInParentPlatform(requesterId, channelId);
-				if (channel == null) {
+				GbStream gbStream = storager.queryStreamInParentPlatform(requesterId, channelId);
+				// 涓嶆槸閫氶亾鍙兘鏄洿鎾祦
+				if (channel != null && gbStream == null ) {
+					if (channel.getStatus() == 0) {
+						logger.info("閫氶亾绂荤嚎锛岃繑鍥�400");
+						responseAck(evt, Response.BAD_REQUEST, "channel [" + channel.getChannelId() + "] offline");
+						return;
+					}
+					responseAck(evt, Response.CALL_IS_BEING_FORWARDED); // 閫氶亾瀛樺湪锛屽彂181锛屽懠鍙浆鎺ヤ腑
+				}else if(channel == null && gbStream != null){
+					Boolean streamReady = zlmrtpServerFactory.isStreamReady(gbStream.getApp(), gbStream.getStream());
+					if (!streamReady) {
+						logger.info("[ app={}, stream={} ]閫氶亾绂荤嚎锛岃繑鍥�400",gbStream.getApp(), gbStream.getStream());
+						responseAck(evt, Response.BAD_REQUEST, "channel [" + gbStream.getGbId() + "] offline");
+						return;
+					}
+					responseAck(evt, Response.CALL_IS_BEING_FORWARDED); // 閫氶亾瀛樺湪锛屽彂181锛屽懠鍙浆鎺ヤ腑
+				}else {
 					logger.info("閫氶亾涓嶅瓨鍦紝杩斿洖404");
 					responseAck(evt, Response.NOT_FOUND); // 閫氶亾涓嶅瓨鍦紝鍙�404锛岃祫婧愪笉瀛樺湪
 					return;
-				}else {
-					responseAck(evt, Response.CALL_IS_BEING_FORWARDED); // 閫氶亾瀛樺湪锛屽彂181锛屽懠鍙浆鎺ヤ腑
 				}
 				// 瑙f瀽sdp娑堟伅, 浣跨敤jainsip 鑷甫鐨剆dp瑙f瀽鏂瑰紡
 				String contentString = new String(request.getRawContent());
@@ -153,33 +165,93 @@
 				String addressStr = sdp.getOrigin().getAddress();
 				//String sessionName = sdp.getSessionName().getValue();
 				logger.info("[涓婄骇鐐规挱]鐢ㄦ埛锛歿}锛� 鍦板潃锛歿}:{}锛� ssrc锛歿}", username, addressStr, port, ssrc);
+				Device device  = null;
+				// 閫氳繃 channel 鍜� gbStream 鏄惁涓簄ull 鍊煎垽鏂潵婧愭槸鐩存挱娴佸悎閫傚浗鏍�
+				if (channel != null) {
+					device = storager.queryVideoDeviceByPlatformIdAndChannelId(requesterId, channelId);
+					if (device == null) {
+						logger.warn("鐐规挱骞冲彴{}鐨勯�氶亾{}鏃舵湭鎵惧埌璁惧淇℃伅", requesterId, channel);
+						responseAck(evt, Response.SERVER_INTERNAL_ERROR);
+						return;
+					}
+					SendRtpItem sendRtpItem = zlmrtpServerFactory.createSendRtpItem(addressStr, port, ssrc, requesterId,
+							device.getDeviceId(), channelId,
+							mediaTransmissionTCP);
+					if (tcpActive != null) {
+						sendRtpItem.setTcpActive(tcpActive);
+					}
+					if (sendRtpItem == null) {
+						logger.warn("鏈嶅姟鍣ㄧ鍙h祫婧愪笉瓒�");
+						responseAck(evt, Response.BUSY_HERE);
+						return;
+					}
 
-				Device device = storager.queryVideoDeviceByPlatformIdAndChannelId(requesterId, channelId);
-				if (device == null) {
-					logger.warn("鐐规挱骞冲彴{}鐨勯�氶亾{}鏃舵湭鎵惧埌璁惧淇℃伅", requesterId, channel);
-					responseAck(evt, Response.SERVER_INTERNAL_ERROR);
-					return;
-				}
-				SendRtpItem sendRtpItem = zlmrtpServerFactory.createSendRtpItem(addressStr, port, ssrc, requesterId, device.getDeviceId(), channelId,
-						mediaTransmissionTCP);
-				if (tcpActive != null) {
-					sendRtpItem.setTcpActive(tcpActive);
-				}
-				if (sendRtpItem == null) {
-					logger.warn("鏈嶅姟鍣ㄧ鍙h祫婧愪笉瓒�");
-					responseAck(evt, Response.BUSY_HERE);
-					return;
-				}
+					// 鍐欏叆redis锛� 瓒呮椂鏃跺洖澶�
+					redisCatchStorage.updateSendRTPSever(sendRtpItem);
+					// 閫氱煡涓嬬骇鎺ㄦ祦锛�
+					PlayResult playResult = playService.play(device.getDeviceId(), channelId, (responseJSON)->{
+						// 鏀跺埌鎺ㄦ祦锛� 鍥炲200OK, 绛夊緟ack
+						// if (sendRtpItem == null) return;
+						sendRtpItem.setStatus(1);
+						redisCatchStorage.updateSendRTPSever(sendRtpItem);
+						// TODO 娣诲姞瀵箃cp鐨勬敮鎸�
+						ZLMServerConfig mediaInfo = redisCatchStorage.getMediaInfo();
+						StringBuffer content = new StringBuffer(200);
+						content.append("v=0\r\n");
+						content.append("o="+"00000"+" 0 0 IN IP4 "+mediaInfo.getWanIp()+"\r\n");
+						content.append("s=Play\r\n");
+						content.append("c=IN IP4 "+mediaInfo.getWanIp()+"\r\n");
+						content.append("t=0 0\r\n");
+						content.append("m=video "+ sendRtpItem.getLocalPort()+" RTP/AVP 96\r\n");
+						content.append("a=sendonly\r\n");
+						content.append("a=rtpmap:96 PS/90000\r\n");
+						content.append("y="+ ssrc + "\r\n");
+						content.append("f=\r\n");
 
-				// 鍐欏叆redis锛� 瓒呮椂鏃跺洖澶�
-				redisCatchStorage.updateSendRTPSever(sendRtpItem);
-				// 閫氱煡涓嬬骇鎺ㄦ祦锛�
-				PlayResult playResult = playService.play(device.getDeviceId(), channelId, (responseJSON)->{
-					// 鏀跺埌鎺ㄦ祦锛� 鍥炲200OK, 绛夊緟ack
+						try {
+							responseAck(evt, content.toString());
+						} catch (SipException e) {
+							e.printStackTrace();
+						} catch (InvalidArgumentException e) {
+							e.printStackTrace();
+						} catch (ParseException e) {
+							e.printStackTrace();
+						}
+					} ,(event -> {
+						// 鏈煡閿欒銆傜洿鎺ヨ浆鍙戣澶囩偣鎾殑閿欒
+						Response response = null;
+						try {
+							response = getMessageFactory().createResponse(event.getResponse().getStatusCode(), evt.getRequest());
+							getServerTransaction(evt).sendResponse(response);
+						} catch (ParseException | SipException | InvalidArgumentException e) {
+							e.printStackTrace();
+						}
+					}));
+					if (logger.isDebugEnabled()) {
+						logger.debug(playResult.getResult().toString());
+					}
+
+				}else if (gbStream != null) {
+					SendRtpItem sendRtpItem = zlmrtpServerFactory.createSendRtpItem(addressStr, port, ssrc, requesterId,
+							gbStream.getApp(), gbStream.getStream(), channelId,
+							mediaTransmissionTCP);
+
+					if (tcpActive != null) {
+						sendRtpItem.setTcpActive(tcpActive);
+					}
+					if (sendRtpItem == null) {
+						logger.warn("鏈嶅姟鍣ㄧ鍙h祫婧愪笉瓒�");
+						responseAck(evt, Response.BUSY_HERE);
+						return;
+					}
+
+					// 鍐欏叆redis锛� 瓒呮椂鏃跺洖澶�
+					redisCatchStorage.updateSendRTPSever(sendRtpItem);
+
 					sendRtpItem.setStatus(1);
 					redisCatchStorage.updateSendRTPSever(sendRtpItem);
 					// TODO 娣诲姞瀵箃cp鐨勬敮鎸�
-					MediaServerConfig mediaInfo = redisCatchStorage.getMediaInfo();
+					ZLMServerConfig mediaInfo = redisCatchStorage.getMediaInfo();
 					StringBuffer content = new StringBuffer(200);
 					content.append("v=0\r\n");
 					content.append("o="+"00000"+" 0 0 IN IP4 "+mediaInfo.getWanIp()+"\r\n");
@@ -201,19 +273,8 @@
 					} catch (ParseException e) {
 						e.printStackTrace();
 					}
-				},(event -> {
-					// 鏈煡閿欒銆傜洿鎺ヨ浆鍙戣澶囩偣鎾殑閿欒
-					Response response = null;
-					try {
-						response = getMessageFactory().createResponse(event.getResponse().getStatusCode(), evt.getRequest());
-						getServerTransaction(evt).sendResponse(response);
-					} catch (ParseException | SipException | InvalidArgumentException e) {
-						e.printStackTrace();
-					}
-				}));
-				if (logger.isDebugEnabled()) {
-					logger.debug(playResult.getResult().toString());
 				}
+
 			} else {
 				// 闈炰笂绾у钩鍙拌姹傦紝鏌ヨ鏄惁璁惧璇锋眰锛堥�氬父涓烘帴鏀惰闊冲箍鎾殑璁惧锛�
 				Device device = storager.queryVideoDevice(requesterId);
@@ -298,6 +359,7 @@
 		}
 	}
 
+
 	/***
 	 * 鍥炲鐘舵�佺爜
 	 * 100 trying
@@ -314,6 +376,12 @@
 		getServerTransaction(evt).sendResponse(response);
 	}
 
+	private void responseAck(RequestEvent evt, int statusCode, String msg) throws SipException, InvalidArgumentException, ParseException {
+		Response response = getMessageFactory().createResponse(statusCode, evt.getRequest());
+		response.setReasonPhrase(msg);
+		getServerTransaction(evt).sendResponse(response);
+	}
+
 	/**
 	 * 鍥炲甯dp鐨�200
 	 * @param evt

--
Gitblit v1.8.0