From 354a39961ad26949f597e4c434b0cd470b7f78ee Mon Sep 17 00:00:00 2001
From: 648540858 <648540858@qq.com>
Date: 星期一, 14 三月 2022 18:24:30 +0800
Subject: [PATCH] 优化点播, 级联点播级联录像。级联列表显示订阅状态

---
 src/main/java/com/genersoft/iot/vmp/gb28181/transmit/event/request/impl/InviteRequestProcessor.java |   90 +++++++++++++++++++++++++++++----------------
 1 files changed, 58 insertions(+), 32 deletions(-)

diff --git a/src/main/java/com/genersoft/iot/vmp/gb28181/transmit/event/request/impl/InviteRequestProcessor.java b/src/main/java/com/genersoft/iot/vmp/gb28181/transmit/event/request/impl/InviteRequestProcessor.java
index f82f781..fef3412 100644
--- a/src/main/java/com/genersoft/iot/vmp/gb28181/transmit/event/request/impl/InviteRequestProcessor.java
+++ b/src/main/java/com/genersoft/iot/vmp/gb28181/transmit/event/request/impl/InviteRequestProcessor.java
@@ -6,6 +6,7 @@
 import com.genersoft.iot.vmp.conf.DynamicTask;
 import com.genersoft.iot.vmp.gb28181.bean.*;
 import com.genersoft.iot.vmp.gb28181.event.SipSubscribe;
+import com.genersoft.iot.vmp.gb28181.session.VideoStreamSessionManager;
 import com.genersoft.iot.vmp.gb28181.transmit.SIPProcessorObserver;
 import com.genersoft.iot.vmp.gb28181.transmit.callback.RequestMessage;
 import com.genersoft.iot.vmp.gb28181.transmit.cmd.ISIPCommander;
@@ -90,6 +91,9 @@
 
 	@Autowired
 	private SIPProcessorObserver sipProcessorObserver;
+
+	@Autowired
+	private VideoStreamSessionManager sessionManager;
 
 
 	@Override
@@ -233,6 +237,7 @@
 				}
 				String username = sdp.getOrigin().getUsername();
 				String addressStr = sdp.getOrigin().getAddress();
+
 				logger.info("[涓婄骇鐐规挱]鐢ㄦ埛锛歿}锛� 鍦板潃锛歿}:{}锛� ssrc锛歿}", username, addressStr, port, ssrc);
 				Device device  = null;
 				// 閫氳繃 channel 鍜� gbStream 鏄惁涓簄ull 鍊煎垽鏂潵婧愭槸鐩存挱娴佸悎閫傚浗鏍�
@@ -266,13 +271,14 @@
 					sendRtpItem.setDialog(dialogByteArray);
 					byte[] transactionByteArray = SerializeUtils.serialize(evt.getServerTransaction());
 					sendRtpItem.setTransaction(transactionByteArray);
-					// 鍐欏叆redis锛� 瓒呮椂鏃跺洖澶�
-					redisCatchStorage.updateSendRTPSever(sendRtpItem);
+
 
 					Long finalStartTime = startTime;
 					Long finalStopTime = stopTime;
 					ZLMHttpHookSubscribe.Event hookEvent = (mediaServerItemInUSe, responseJSON)->{
-						logger.info("[涓婄骇鐐规挱]涓嬬骇宸茬粡寮�濮嬫帹娴併�� 鍥炲200OK(SDP)锛� {}/{}", sendRtpItem.getApp(), sendRtpItem.getStreamId());
+						String app = responseJSON.getString("app");
+						String stream = responseJSON.getString("stream");
+						logger.info("[涓婄骇鐐规挱]涓嬬骇宸茬粡寮�濮嬫帹娴併�� 鍥炲200OK(SDP)锛� {}/{}", app, stream);
 						//     * 0 绛夊緟璁惧鎺ㄦ祦涓婃潵
 						//     * 1 涓嬬骇宸茬粡鎺ㄦ祦锛岀瓑寰呬笂绾у钩鍙板洖澶峚ck
 						//     * 2 鎺ㄦ祦涓�
@@ -325,46 +331,66 @@
 							e.printStackTrace();
 						}
 					});
+					sendRtpItem.setApp("rtp");
 					if ("Playback".equals(sessionName)) {
 						sendRtpItem.setPlay(false);
-						sendRtpItem.setStreamId(ssrc);
+						SSRCInfo ssrcInfo = mediaServerService.openRTPServer(mediaServerItem, null, true);
+						sendRtpItem.setStreamId(ssrcInfo.getStream());
+						// 鍐欏叆redis锛� 瓒呮椂鏃跺洖澶�
+						redisCatchStorage.updateSendRTPSever(sendRtpItem);
 						SimpleDateFormat format = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
-						playService.playBack(device.getDeviceId(), channelId, format.format(start), format.format(end),result -> {
-							if (result.getCode() != 0){
-								logger.warn("褰曞儚鍥炴斁澶辫触");
-								if (result.getEvent() != null) {
-									errorEvent.response(result.getEvent());
+						playService.playBack(mediaServerItem, ssrcInfo, device.getDeviceId(), channelId, format.format(start),
+								format.format(end), null, result -> {
+								if (result.getCode() != 0){
+									logger.warn("褰曞儚鍥炴斁澶辫触");
+									if (result.getEvent() != null) {
+										errorEvent.response(result.getEvent());
+									}
+									redisCatchStorage.deleteSendRTPServer(platform.getServerGBId(), channelId, callIdHeader.getCallId(), null);
+									try {
+										responseAck(evt, Response.REQUEST_TIMEOUT);
+									} catch (SipException e) {
+										e.printStackTrace();
+									} catch (InvalidArgumentException e) {
+										e.printStackTrace();
+									} catch (ParseException e) {
+										e.printStackTrace();
+									}
+								}else {
+									if (result.getMediaServerItem() != null) {
+										hookEvent.response(result.getMediaServerItem(), result.getResponse());
+									}
 								}
-								redisCatchStorage.deleteSendRTPServer(platform.getServerGBId(), channelId, callIdHeader.getCallId(), null);
-								try {
-									responseAck(evt, Response.REQUEST_TIMEOUT);
-								} catch (SipException e) {
-									e.printStackTrace();
-								} catch (InvalidArgumentException e) {
-									e.printStackTrace();
-								} catch (ParseException e) {
-									e.printStackTrace();
-								}
-							}else {
-								if (result.getMediaServerItem() != null) {
-									hookEvent.response(result.getMediaServerItem(), result.getResponse());
-								}
-							}
-						});
+							});
 					}else {
 						sendRtpItem.setPlay(true);
-						StreamInfo streamInfo = redisCatchStorage.queryPlayByDevice(device.getDeviceId(), channelId);
-						if (streamInfo == null) {
+						SsrcTransaction playTransaction = sessionManager.getSsrcTransaction(device.getDeviceId(), channelId, "play", null);
+						if (playTransaction != null) {
+							Boolean streamReady = zlmrtpServerFactory.isStreamReady(mediaServerItem, "rtp", playTransaction.getStream());
+							if (!streamReady) {
+								playTransaction = null;
+							}
+						}
+						if (playTransaction == null) {
+							SSRCInfo ssrcInfo = mediaServerService.openRTPServer(mediaServerItem, null, true);
 							if (mediaServerItem.isRtpEnable()) {
 								sendRtpItem.setStreamId(String.format("%s_%s", device.getDeviceId(), channelId));
+							}else {
+								sendRtpItem.setStreamId(ssrcInfo.getStream());
 							}
-							sendRtpItem.setPlay(false);
-							playService.play(mediaServerItem,device.getDeviceId(), channelId, hookEvent, errorEvent, ()->{
+							// 鍐欏叆redis锛� 瓒呮椂鏃跺洖澶�
+							redisCatchStorage.updateSendRTPSever(sendRtpItem);
+							playService.play(mediaServerItem, ssrcInfo, device, channelId, hookEvent, errorEvent, (code, msg)->{
 								redisCatchStorage.deleteSendRTPServer(platform.getServerGBId(), channelId, callIdHeader.getCallId(), null);
-							});
+							}, null);
 						}else {
-							sendRtpItem.setStreamId(streamInfo.getStream());
-							hookEvent.response(mediaServerItem, null);
+							sendRtpItem.setStreamId(playTransaction.getStream());
+							// 鍐欏叆redis锛� 瓒呮椂鏃跺洖澶�
+							redisCatchStorage.updateSendRTPSever(sendRtpItem);
+							JSONObject jsonObject = new JSONObject();
+							jsonObject.put("app", sendRtpItem.getApp());
+							jsonObject.put("stream", sendRtpItem.getStreamId());
+							hookEvent.response(mediaServerItem, jsonObject);
 						}
 					}
 				}else if (gbStream != null) {

--
Gitblit v1.8.0