Thirdparty
/
wvp-GB28181-pro
mirror of https://github.com/648540858/wvp-GB28181-pro
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

临时提交

pull/1642/head
648540858 2024-09-04 23:06:27 +08:00
parent ad5b66eb71
commit 568abbcd9d
4 changed files with 110 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/main/java/com/genersoft/iot/vmp/media/service/IMediaServerService.java
View File

@ -160,4 +160,7 @@ public interface IMediaServerService {
void stopProxy(MediaServer mediaServer, String streamKey);
StreamInfo getMediaByAppAndStream(String app, String stream);
int createRTPServer(MediaServer mediaServerItem, String streamId, long ssrc, Integer port, boolean onlyAuto, boolean disableAudio, boolean reUsePort, Integer tcpMode);
}

16
src/main/java/com/genersoft/iot/vmp/media/service/impl/MediaServerServiceImpl.java
View File

@ -186,6 +186,22 @@ public class MediaServerServiceImpl implements IMediaServerService {
return new SSRCInfo(rtpServerPort, ssrc, streamId);
}
@Override
public int createRTPServer(MediaServer mediaServer, String streamId, long ssrc, Integer port, boolean onlyAuto, boolean disableAudio, boolean reUsePort, Integer tcpMode) {
int rtpServerPort;
if (mediaServer.isRtpEnable()) {
IMediaNodeServerService mediaNodeServerService = nodeServerServiceMap.get(mediaServer.getType());
if (mediaNodeServerService == null) {
log.info("[openRTPServer] 失败, mediaServer的类型： {}，未找到对应的实现类", mediaServer.getType());
return 0;
}
rtpServerPort = mediaNodeServerService.createRTPServer(mediaServer, streamId, ssrc, port, onlyAuto, disableAudio, reUsePort, tcpMode);
} else {
rtpServerPort = mediaServer.getRtpProxyPort();
}
return rtpServerPort;
}
@Override
public void closeRTPServer(MediaServer mediaServer, String streamId) {
if (mediaServer == null) {

24
src/main/java/com/genersoft/iot/vmp/service/bean/RTPServerParam.java
View File

@ -6,14 +6,18 @@ import lombok.Data;
@Data
public class RTPServerParam {
MediaServer mediaServerItem;
String streamId;
String presetSsrc;
boolean ssrcCheck;
boolean isPlayback;
Integer port;
Boolean onlyAuto;
Boolean disableAudio;
Boolean reUsePort;
Integer tcpMode;
private MediaServer mediaServerItem;
private String streamId;
private String presetSsrc;
private boolean ssrcCheck;
private boolean playback;
private Integer port;
private boolean onlyAuto;
private boolean disableAudio;
private boolean reUsePort;
/**
* tcp模式，0时为不启用tcp监听，1时为启用tcp监听，2时为tcp主动连接模式
*/
private Integer tcpMode;
}

79
src/main/java/com/genersoft/iot/vmp/service/impl/RtpServerServiceImpl.java
View File

@ -1,29 +1,42 @@
package com.genersoft.iot.vmp.service.impl;
import com.genersoft.iot.vmp.common.InviteInfo;
import com.genersoft.iot.vmp.common.InviteSessionType;
import com.genersoft.iot.vmp.common.StreamInfo;
import com.genersoft.iot.vmp.conf.DynamicTask;
import com.genersoft.iot.vmp.conf.UserSetting;
import com.genersoft.iot.vmp.conf.exception.SsrcTransactionNotFoundException;
import com.genersoft.iot.vmp.gb28181.bean.AudioBroadcastCatch;
import com.genersoft.iot.vmp.gb28181.bean.Device;
import com.genersoft.iot.vmp.gb28181.bean.InviteStreamType;
import com.genersoft.iot.vmp.gb28181.bean.SendRtpItem;
import com.genersoft.iot.vmp.gb28181.session.SSRCFactory;
import com.genersoft.iot.vmp.media.event.hook.Hook;
import com.genersoft.iot.vmp.media.event.hook.HookType;
import com.genersoft.iot.vmp.media.event.media.MediaArrivalEvent;
import com.genersoft.iot.vmp.media.event.media.MediaDepartureEvent;
import com.genersoft.iot.vmp.media.service.IMediaNodeServerService;
import com.genersoft.iot.vmp.media.service.IMediaServerService;
import com.genersoft.iot.vmp.service.IReceiveRtpServerService;
import com.genersoft.iot.vmp.service.bean.ErrorCallback;
import com.genersoft.iot.vmp.service.bean.InviteErrorCode;
import com.genersoft.iot.vmp.service.bean.RTPServerParam;
import com.genersoft.iot.vmp.service.bean.SSRCInfo;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.event.EventListener;
import org.springframework.scheduling.annotation.Async;
import org.springframework.security.core.parameters.P;
import org.springframework.stereotype.Service;
import org.springframework.util.Assert;
import javax.sip.InvalidArgumentException;
import javax.sip.SipException;
import java.text.ParseException;
import java.util.List;
import java.util.UUID;
@Slf4j
@Service
public class RtpServerServiceImpl implements IReceiveRtpServerService {
@ -33,6 +46,12 @@ public class RtpServerServiceImpl implements IReceiveRtpServerService {
@Autowired
private DynamicTask dynamicTask;
@Autowired
private SSRCFactory ssrcFactory;
@Autowired
private UserSetting userSetting;
/**
* 流到来的处理
*/
@ -53,9 +72,65 @@ public class RtpServerServiceImpl implements IReceiveRtpServerService {
@Override
public SSRCInfo openRTPServer(RTPServerParam rtpServerParam, ErrorCallback<StreamInfo> callback) {
// 开启流到来的监听
if (callback == null) {
log.warn("[开启RTP收流] 失败，回调为NULL");
return null;
}
if (rtpServerParam.getMediaServerItem() == null) {
log.warn("[开启RTP收流] 失败，媒体节点为NULL");
return null;
}
// 获取mediaServer可用的ssrc
final String ssrc;
if (rtpServerParam.getPresetSsrc() != null) {
ssrc = rtpServerParam.getPresetSsrc();
}else {
if (rtpServerParam.isPlayback()) {
ssrc = ssrcFactory.getPlayBackSsrc(rtpServerParam.getMediaServerItem().getId());
}else {
ssrc = ssrcFactory.getPlaySsrc(rtpServerParam.getMediaServerItem().getId());
}
}
final String streamId;
if (rtpServerParam.getStreamId() == null) {
streamId = String.format("%08x", Long.parseLong(ssrc)).toUpperCase();
}else {
streamId = rtpServerParam.getStreamId();
}
if (rtpServerParam.isSsrcCheck() && rtpServerParam.getTcpMode() > 0) {
// 目前zlm不支持 tcp模式更新ssrc，暂时关闭ssrc校验
log.warn("[openRTPServer] 平台对接时下级可能自定义ssrc，但是tcp模式zlm收流目前无法更新ssrc，可能收流超时，此时请使用udp收流或者关闭ssrc校验");
}
int rtpServerPort;
if (rtpServerParam.getMediaServerItem().isRtpEnable()) {
rtpServerPort = mediaServerService.createRTPServer(rtpServerParam.getMediaServerItem(), streamId,
rtpServerParam.isSsrcCheck() ? Long.parseLong(ssrc) : 0, rtpServerParam.getPort(), rtpServerParam.isOnlyAuto(),
rtpServerParam.isDisableAudio(), rtpServerParam.isReUsePort(), rtpServerParam.getTcpMode());
} else {
rtpServerPort = rtpServerParam.getMediaServerItem().getRtpProxyPort();
}
if (rtpServerPort == 0) {
callback.run(InviteErrorCode.ERROR_FOR_RESOURCE_EXHAUSTION.getCode(), InviteErrorCode.ERROR_FOR_RESOURCE_EXHAUSTION.getMsg(), null);
return null;
}
SSRCInfo ssrcInfo = new SSRCInfo(rtpServerPort, ssrc, streamId);
// 设置流超时的定时任务
// 调用节点，开启端口监听
String timeOutTaskKey = UUID.randomUUID().toString();
dynamicTask.startDelay(timeOutTaskKey, () -> {
// 收流超时
// 释放ssrc
if (rtpServerParam.getPresetSsrc() == null) {
ssrcFactory.releaseSsrc(rtpServerParam.getMediaServerItem().getId(), ssrc);
}
// 关闭收流端口
mediaServerService.closeRTPServer(rtpServerParam.getMediaServerItem(), streamId);
}, userSetting.getPlayTimeout());
// 开启流到来的监听
}