From a9bdb0a706e10d2dffb50ae5a8086dd744bbd976 Mon Sep 17 00:00:00 2001
From: 648540858 <648540858@qq.com>
Date: 星期二, 10 五月 2022 17:45:07 +0800
Subject: [PATCH] 优化语音广播流程

---
 src/main/java/com/genersoft/iot/vmp/gb28181/transmit/event/request/impl/InviteRequestProcessor.java |  130 +++++++++++++++++++++----------------------
 1 files changed, 64 insertions(+), 66 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 ae49e2c..e3a9b27 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
@@ -114,6 +114,7 @@
 	private SipConfig config;
 
 
+
 	@Override
 	public void afterPropertiesSet() throws Exception {
 		// 娣诲姞娑堟伅澶勭悊鐨勮闃�
@@ -492,7 +493,6 @@
 								gbStream.getApp(), gbStream.getStream(), channelId,
 								mediaTransmissionTCP);
 
-
 						if (sendRtpItem == null) {
 							logger.warn("鏈嶅姟鍣ㄧ鍙h祫婧愪笉瓒�");
 							responseAck(evt, Response.BUSY_HERE);
@@ -562,25 +562,16 @@
 		}
 	}
 
-	public void inviteFromDeviceHandle(RequestEvent evt, String requesterId, String channelId) throws InvalidArgumentException, ParseException, SipException, SdpException {
+	public void inviteFromDeviceHandle(RequestEvent evt, String requesterId, String channelId1) throws InvalidArgumentException, ParseException, SipException, SdpException {
 
-		// 鍏煎濂囪懇鐨勬捣搴疯繖閲屼娇鐢ㄧ殑涓嶆槸閫氶亾缂栧彿鑰屾槸鏈钩鍙扮紪鍙�
-//		if (channelId.equals(config.getId())) {
-//			List<AudioBroadcastCatch> all = audioBroadcastManager.getAll();
-//			for (AudioBroadcastCatch audioBroadcastCatch : all) {
-//				if (audioBroadcastCatch.getDeviceId().equals(requesterId)) {
-//					channelId = audioBroadcastCatch.getChannelId();
-//				}
-//			}
-//		}
-//		// 鍏煎澶辫触
-//		if (channelId.equals(config.getId())) {
-//			responseAck(evt, Response.BAD_REQUEST);
-//			return;
-//		}
 		// 闈炰笂绾у钩鍙拌姹傦紝鏌ヨ鏄惁璁惧璇锋眰锛堥�氬父涓烘帴鏀惰闊冲箍鎾殑璁惧锛�
 		Device device = redisCatchStorage.getDevice(requesterId);
-
+		AudioBroadcastCatch audioBroadcastCatch = audioBroadcastManager.get(requesterId, channelId1);
+		if (audioBroadcastCatch == null) {
+			logger.warn("鏉ヨ嚜璁惧鐨処nvite璇锋眰闈炶闊冲箍鎾紝宸插拷鐣�");
+			responseAck(evt, Response.FORBIDDEN);
+			return;
+		}
 		Request request = evt.getRequest();
 		if (device != null) {
 			logger.info("鏀跺埌璁惧" + requesterId + "鐨勮闊冲箍鎾璉nvite璇锋眰");
@@ -606,7 +597,6 @@
 
 			// 鏌ョ湅鏄惁鏀寔PS 璐熻浇96
 			int port = -1;
-			//boolean recvonly = false;
 			boolean mediaTransmissionTCP = false;
 			Boolean tcpActive = null;
 			for (int i = 0; i < mediaDescriptions.size(); i++) {
@@ -638,7 +628,6 @@
 				responseAck(evt, Response.UNSUPPORTED_MEDIA_TYPE); // 涓嶆敮鎸佺殑鏍煎紡锛屽彂415
 				return;
 			}
-			String sessionName = sdp.getSessionName().getValue();
 			String addressStr = sdp.getOrigin().getAddress();
 			logger.info("璁惧{}璇锋眰璇煶娴侊紝鍦板潃锛歿}:{}锛宻src锛歿}", requesterId, addressStr, port, ssrc);
 
@@ -649,20 +638,19 @@
 				return;
 			}
 			SendRtpItem sendRtpItem = zlmrtpServerFactory.createSendRtpItem(mediaServerItem, addressStr, port, ssrc, requesterId,
-					device.getDeviceId(), channelId,
+					device.getDeviceId(), audioBroadcastCatch.getChannelId(),
 					mediaTransmissionTCP);
-			sendRtpItem.setTcp(mediaTransmissionTCP);
-			if (tcpActive != null) {
-				sendRtpItem.setTcpActive(tcpActive);
-			}
 			if (sendRtpItem == null) {
 				logger.warn("鏈嶅姟鍣ㄧ鍙h祫婧愪笉瓒�");
 				responseAck(evt, Response.BUSY_HERE);
 				return;
 			}
-
+			sendRtpItem.setTcp(mediaTransmissionTCP);
+			if (tcpActive != null) {
+				sendRtpItem.setTcpActive(tcpActive);
+			}
 			String app = "broadcast";
-			String stream = device.getDeviceId() + "_" + channelId;
+			String stream = device.getDeviceId() + "_" + audioBroadcastCatch.getChannelId();
 
 			CallIdHeader callIdHeader = (CallIdHeader) request.getHeader(CallIdHeader.NAME);
 			sendRtpItem.setPlayType(InviteStreamType.PLAY);
@@ -671,6 +659,9 @@
 			sendRtpItem.setStatus(1);
 			sendRtpItem.setApp(app);
 			sendRtpItem.setStreamId(stream);
+			sendRtpItem.setPt(8);
+			sendRtpItem.setUsePs(false);
+			sendRtpItem.setOnlyAudio(true);
 			redisCatchStorage.updateSendRTPSever(sendRtpItem);
 
 			// hook鐩戝惉绛夊緟璁惧鎺ㄦ祦涓婃潵
@@ -682,10 +673,9 @@
 			subscribeKey.put("schema", "rtmp");
 			subscribeKey.put("mediaServerId", mediaServerItem.getId());
 			String finalSsrc = ssrc;
-			String waiteStreamTimeoutTaskKey = "waite-stream-" + device.getDeviceId() + channelId;
+			// 娴佸凡缁忓瓨鍦ㄦ椂鐩存帴鎺ㄦ祦
 			if (zlmrtpServerFactory.isStreamReady(mediaServerItem, app, stream)) {
 				logger.info("鍙戠幇宸茬粡鍦ㄦ帹娴�");
-				dynamicTask.stop(waiteStreamTimeoutTaskKey);
 				sendRtpItem.setStatus(2);
 				redisCatchStorage.updateSendRTPSever(sendRtpItem);
 				StringBuffer content = new StringBuffer(200);
@@ -706,6 +696,10 @@
 				parentPlatform.setServerGBId(device.getDeviceId());
 				try {
 					responseSdpAck(evt, content.toString(), parentPlatform);
+					Dialog dialog = evt.getDialog();
+					audioBroadcastCatch.setDialog((SIPDialog) dialog);
+					audioBroadcastCatch.setRequest((SIPRequest) request);
+					audioBroadcastManager.update(audioBroadcastCatch);
 				} catch (SipException e) {
 					throw new RuntimeException(e);
 				} catch (InvalidArgumentException e) {
@@ -714,21 +708,19 @@
 					throw new RuntimeException(e);
 				}
 			}else {
+				// 娴佷笉瀛樺湪鏃剁洃鍚祦涓婄嚎
 				// 璁剧疆绛夊緟鎺ㄦ祦鐨勮秴鏃�; 榛樿20s
-				String finalChannelId = channelId;
+				String waiteStreamTimeoutTaskKey = "waite-stream-" + device.getDeviceId() + audioBroadcastCatch.getChannelId();
 				dynamicTask.startDelay(waiteStreamTimeoutTaskKey, ()->{
 					logger.info("绛夊緟鎺ㄦ祦瓒呮椂: {}/{}", app, stream);
-					if (audioBroadcastManager.exit(device.getDeviceId(), finalChannelId)) {
-						audioBroadcastManager.del(device.getDeviceId(), finalChannelId);
-					}else {
-						// 鍏煎娴峰悍浣跨敤浜嗛敊璇殑閫氶亾ID鐨勬儏鍐�
-						audioBroadcastManager.delByDeviceId(device.getDeviceId());
-					}
-
+					subscribe.removeSubscribe(ZLMHttpHookSubscribe.HookType.on_stream_changed, subscribeKey);
+					playService.stopAudioBroadcast(device.getDeviceId(), audioBroadcastCatch.getChannelId());
 					// 鍙戦�乥ye
 					try {
-						cmder.streamByeCmd((SIPDialog)evt.getServerTransaction().getDialog(), (SIPRequest) evt.getRequest(), null);
+						responseAck(evt, Response.BUSY_HERE);
 					} catch (SipException e) {
+						throw new RuntimeException(e);
+					} catch (InvalidArgumentException e) {
 						throw new RuntimeException(e);
 					} catch (ParseException e) {
 						throw new RuntimeException(e);
@@ -737,37 +729,43 @@
 
 				subscribe.addSubscribe(ZLMHttpHookSubscribe.HookType.on_stream_changed, subscribeKey,
 						(MediaServerItem mediaServerItemInUse, JSONObject json)->{
-							sendRtpItem.setStatus(2);
-							redisCatchStorage.updateSendRTPSever(sendRtpItem);
-							StringBuffer content = new StringBuffer(200);
-							content.append("v=0\r\n");
-							content.append("o="+ finalChannelId +" 0 0 IN IP4 "+mediaServerItem.getSdpIp()+"\r\n");
-							content.append("s=Play\r\n");
-							content.append("c=IN IP4 "+mediaServerItem.getSdpIp()+"\r\n");
-							content.append("t=0 0\r\n");
-							content.append("m=video "+ sendRtpItem.getLocalPort()+" RTP/AVP 8\r\n");
-							content.append("a=sendonly\r\n");
-							content.append("a=rtpmap:8 PCMA/8000\r\n");
-							content.append("y="+ finalSsrc + "\r\n");
-							content.append("f=v/////a/1/8/1\r\n");
+					logger.info("鏀跺埌璇煶瀵硅鎺ㄦ祦");
+					try {
+						sendRtpItem.setStatus(2);
+						redisCatchStorage.updateSendRTPSever(sendRtpItem);
+						StringBuffer content = new StringBuffer(200);
+						content.append("v=0\r\n");
+						content.append("o="+ config.getId() +" "+ sdp.getOrigin().getSessionId() +" " + sdp.getOrigin().getSessionVersion()  + " IN IP4 "+mediaServerItem.getSdpIp()+"\r\n");
+						content.append("s=Play\r\n");
+						content.append("c=IN IP4 "+mediaServerItem.getSdpIp()+"\r\n");
+						content.append("t=0 0\r\n");
+						content.append("m=audio "+ sendRtpItem.getLocalPort()+" RTP/AVP 8\r\n");
+						content.append("a=sendonly\r\n");
+						content.append("a=rtpmap:8 PCMA/8000\r\n");
+						content.append("y="+ finalSsrc + "\r\n");
+						content.append("f=v/////a/1/8/1\r\n");
 
-							ParentPlatform parentPlatform = new ParentPlatform();
-							parentPlatform.setServerIP(device.getIp());
-							parentPlatform.setServerPort(device.getPort());
-							parentPlatform.setServerGBId(device.getDeviceId());
-							try {
-								responseSdpAck(evt, content.toString(), parentPlatform);
-							} catch (SipException e) {
-								throw new RuntimeException(e);
-							} catch (InvalidArgumentException e) {
-								throw new RuntimeException(e);
-							} catch (ParseException e) {
-								throw new RuntimeException(e);
-							}
-						});
+						ParentPlatform parentPlatform = new ParentPlatform();
+						parentPlatform.setServerIP(device.getIp());
+						parentPlatform.setServerPort(device.getPort());
+						parentPlatform.setServerGBId(device.getDeviceId());
+
+						responseSdpAck(evt, content.toString(), parentPlatform);
+						Dialog dialog = evt.getDialog();
+						audioBroadcastCatch.setDialog((SIPDialog) dialog);
+						audioBroadcastCatch.setRequest((SIPRequest) request);
+						audioBroadcastManager.update(audioBroadcastCatch);
+					} catch (SipException e) {
+						throw new RuntimeException(e);
+					} catch (InvalidArgumentException e) {
+						throw new RuntimeException(e);
+					} catch (ParseException e) {
+						throw new RuntimeException(e);
+					} catch (SdpParseException e) {
+						throw new RuntimeException(e);
+					}
+				});
 			}
-			String timeOutTaskKey = "audio-broadcast-" + device.getDeviceId() + channelId;
-			dynamicTask.stop(timeOutTaskKey);
 			String key = DeferredResultHolder.CALLBACK_CMD_BROADCAST + device.getDeviceId();
 			WVPResult<AudioBroadcastResult> wvpResult = new WVPResult<>();
 			wvpResult.setCode(0);

--
Gitblit v1.8.0