|
|
@@ -0,0 +1,203 @@
|
|
|
+package com.genersoft.iot.vmp.service.redisMsg.control;
|
|
|
+
|
|
|
+import com.alibaba.fastjson2.JSON;
|
|
|
+import com.alibaba.fastjson2.JSONObject;
|
|
|
+import com.genersoft.iot.vmp.conf.UserSetting;
|
|
|
+import com.genersoft.iot.vmp.conf.redis.RedisRpcConfig;
|
|
|
+import com.genersoft.iot.vmp.conf.redis.bean.RedisRpcMessage;
|
|
|
+import com.genersoft.iot.vmp.conf.redis.bean.RedisRpcRequest;
|
|
|
+import com.genersoft.iot.vmp.conf.redis.bean.RedisRpcResponse;
|
|
|
+import com.genersoft.iot.vmp.gb28181.bean.SendRtpItem;
|
|
|
+import com.genersoft.iot.vmp.gb28181.session.SSRCFactory;
|
|
|
+import com.genersoft.iot.vmp.media.zlm.SendRtpPortManager;
|
|
|
+import com.genersoft.iot.vmp.media.zlm.ZLMServerFactory;
|
|
|
+import com.genersoft.iot.vmp.media.zlm.ZlmHttpHookSubscribe;
|
|
|
+import com.genersoft.iot.vmp.media.zlm.dto.HookSubscribeFactory;
|
|
|
+import com.genersoft.iot.vmp.media.zlm.dto.HookSubscribeForStreamChange;
|
|
|
+import com.genersoft.iot.vmp.media.zlm.dto.MediaServerItem;
|
|
|
+import com.genersoft.iot.vmp.media.zlm.dto.hook.HookParam;
|
|
|
+import com.genersoft.iot.vmp.service.IMediaServerService;
|
|
|
+import com.genersoft.iot.vmp.storager.IRedisCatchStorage;
|
|
|
+import com.genersoft.iot.vmp.vmanager.bean.WVPResult;
|
|
|
+import org.slf4j.Logger;
|
|
|
+import org.slf4j.LoggerFactory;
|
|
|
+import org.springframework.beans.factory.annotation.Autowired;
|
|
|
+import org.springframework.data.redis.core.RedisTemplate;
|
|
|
+import org.springframework.stereotype.Component;
|
|
|
+
|
|
|
+/**
|
|
|
+ * 其他wvp发起的rpc调用,这里的方法被 RedisRpcConfig 通过反射寻找对应的方法名称调用
|
|
|
+ */
|
|
|
+@Component
|
|
|
+public class RedisRpcController {
|
|
|
+
|
|
|
+ private final static Logger logger = LoggerFactory.getLogger(RedisRpcController.class);
|
|
|
+
|
|
|
+ @Autowired
|
|
|
+ private SSRCFactory ssrcFactory;
|
|
|
+
|
|
|
+ @Autowired
|
|
|
+ private IMediaServerService mediaServerService;
|
|
|
+
|
|
|
+ @Autowired
|
|
|
+ private SendRtpPortManager sendRtpPortManager;
|
|
|
+
|
|
|
+ @Autowired
|
|
|
+ private IRedisCatchStorage redisCatchStorage;
|
|
|
+
|
|
|
+ @Autowired
|
|
|
+ private UserSetting userSetting;
|
|
|
+
|
|
|
+ @Autowired
|
|
|
+ private ZlmHttpHookSubscribe hookSubscribe;
|
|
|
+
|
|
|
+ @Autowired
|
|
|
+ private ZLMServerFactory zlmServerFactory;
|
|
|
+
|
|
|
+
|
|
|
+ @Autowired
|
|
|
+ private RedisTemplate<Object, Object> redisTemplate;
|
|
|
+
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 获取发流的信息
|
|
|
+ */
|
|
|
+ public RedisRpcResponse getSendRtpItem(RedisRpcRequest request) {
|
|
|
+ SendRtpItem sendRtpItem = JSON.parseObject(request.getParam().toString(), SendRtpItem.class);
|
|
|
+ logger.info("[redis-rpc] 获取发流的信息: {}/{}", sendRtpItem.getApp(), sendRtpItem.getStream() );
|
|
|
+ // 查询本级是否有这个流
|
|
|
+ MediaServerItem mediaServerItem = mediaServerService.getMediaServerByAppAndStream(sendRtpItem.getApp(), sendRtpItem.getStream());
|
|
|
+ if (mediaServerItem == null) {
|
|
|
+ RedisRpcResponse response = request.getResponse();
|
|
|
+ response.setStatusCode(200);
|
|
|
+ }
|
|
|
+ // 自平台内容
|
|
|
+ int localPort = sendRtpPortManager.getNextPort(mediaServerItem);
|
|
|
+ if (localPort == 0) {
|
|
|
+ logger.info("[redis-rpc] getSendRtpItem->服务器端口资源不足" );
|
|
|
+ RedisRpcResponse response = request.getResponse();
|
|
|
+ response.setStatusCode(200);
|
|
|
+ }
|
|
|
+ // 写入redis, 超时时回复
|
|
|
+ sendRtpItem.setStatus(1);
|
|
|
+ sendRtpItem.setServerId(userSetting.getServerId());
|
|
|
+ sendRtpItem.setLocalIp(mediaServerItem.getSdpIp());
|
|
|
+ if (sendRtpItem.getSsrc() == null) {
|
|
|
+ // 上级平台点播时不使用上级平台指定的ssrc,使用自定义的ssrc,参考国标文档-点播外域设备媒体流SSRC处理方式
|
|
|
+ String ssrc = "Play".equalsIgnoreCase(sendRtpItem.getSessionName()) ? ssrcFactory.getPlaySsrc(mediaServerItem.getId()) : ssrcFactory.getPlayBackSsrc(mediaServerItem.getId());
|
|
|
+ sendRtpItem.setSsrc(ssrc);
|
|
|
+ }
|
|
|
+ redisCatchStorage.updateSendRTPSever(sendRtpItem);
|
|
|
+ RedisRpcResponse response = request.getResponse();
|
|
|
+ response.setStatusCode(200);
|
|
|
+ response.setBody(sendRtpItem);
|
|
|
+ return response;
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 监听流上线
|
|
|
+ */
|
|
|
+ public RedisRpcResponse waitePushStreamOnline(RedisRpcRequest request) {
|
|
|
+ SendRtpItem sendRtpItem = JSON.parseObject(request.getParam().toString(), SendRtpItem.class);
|
|
|
+ logger.info("[redis-rpc] 监听流上线: {}/{}", sendRtpItem.getApp(), sendRtpItem.getStream() );
|
|
|
+ // 查询本级是否有这个流
|
|
|
+ MediaServerItem mediaServerItem = mediaServerService.getMediaServerByAppAndStream(sendRtpItem.getApp(), sendRtpItem.getStream());
|
|
|
+ if (mediaServerItem != null) {
|
|
|
+ logger.info("[redis-rpc] 监听流上线时发现流已存在直接返回: {}/{}", sendRtpItem.getApp(), sendRtpItem.getStream() );
|
|
|
+ RedisRpcResponse response = request.getResponse();
|
|
|
+ response.setBody(sendRtpItem);
|
|
|
+ response.setStatusCode(200);
|
|
|
+ }
|
|
|
+ // 监听流上线。 流上线直接发送sendRtpItem消息给实际的信令处理者
|
|
|
+ HookSubscribeForStreamChange hook = HookSubscribeFactory.on_stream_changed(
|
|
|
+ sendRtpItem.getApp(), sendRtpItem.getStream(), true, "rtsp", null);
|
|
|
+
|
|
|
+ hookSubscribe.addSubscribe(hook, (MediaServerItem mediaServerItemInUse, HookParam hookParam) -> {
|
|
|
+ logger.info("[redis-rpc] 监听流上线,流已上线: {}/{}", sendRtpItem.getApp(), sendRtpItem.getStream() );
|
|
|
+ // 读取redis中的上级点播信息,生成sendRtpItm发送出去
|
|
|
+ if (sendRtpItem.getSsrc() == null) {
|
|
|
+ // 上级平台点播时不使用上级平台指定的ssrc,使用自定义的ssrc,参考国标文档-点播外域设备媒体流SSRC处理方式
|
|
|
+ String ssrc = "Play".equalsIgnoreCase(sendRtpItem.getSessionName()) ? ssrcFactory.getPlaySsrc(mediaServerItemInUse.getId()) : ssrcFactory.getPlayBackSsrc(mediaServerItemInUse.getId());
|
|
|
+ sendRtpItem.setSsrc(ssrc);
|
|
|
+ }
|
|
|
+ sendRtpItem.setMediaServerId(mediaServerItemInUse.getId());
|
|
|
+ sendRtpItem.setLocalIp(mediaServerItemInUse.getSdpIp());
|
|
|
+ sendRtpItem.setServerId(userSetting.getServerId());
|
|
|
+ RedisRpcResponse response = request.getResponse();
|
|
|
+ response.setBody(sendRtpItem);
|
|
|
+ response.setStatusCode(200);
|
|
|
+ // 手动发送结果
|
|
|
+ sendResponse(response);
|
|
|
+
|
|
|
+ });
|
|
|
+ return null;
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 开始发流
|
|
|
+ */
|
|
|
+ public RedisRpcResponse startSendRtp(RedisRpcRequest request) {
|
|
|
+ SendRtpItem sendRtpItem = JSON.parseObject(request.getParam().toString(), SendRtpItem.class);
|
|
|
+ MediaServerItem mediaServerItem = mediaServerService.getOne(sendRtpItem.getMediaServerId());
|
|
|
+ if (mediaServerItem == null) {
|
|
|
+ logger.info("[redis-rpc] startSendRtp->未找到MediaServer: {}", sendRtpItem.getMediaServerId() );
|
|
|
+ RedisRpcResponse response = request.getResponse();
|
|
|
+ response.setStatusCode(200);
|
|
|
+ }
|
|
|
+
|
|
|
+ Boolean streamReady = zlmServerFactory.isStreamReady(mediaServerItem, sendRtpItem.getApp(), sendRtpItem.getStream());
|
|
|
+ if (!streamReady) {
|
|
|
+ logger.info("[redis-rpc] startSendRtp->流不在线: {}/{}", sendRtpItem.getApp(), sendRtpItem.getStream() );
|
|
|
+ RedisRpcResponse response = request.getResponse();
|
|
|
+ response.setStatusCode(200);
|
|
|
+ }
|
|
|
+ JSONObject jsonObject = zlmServerFactory.startSendRtp(mediaServerItem, sendRtpItem);
|
|
|
+ RedisRpcResponse response = request.getResponse();
|
|
|
+ response.setStatusCode(200);
|
|
|
+ if (jsonObject.getInteger("code") == 0) {
|
|
|
+ WVPResult wvpResult = WVPResult.success();
|
|
|
+ response.setBody(wvpResult);
|
|
|
+ }else {
|
|
|
+ WVPResult wvpResult = WVPResult.fail(jsonObject.getInteger("code"), jsonObject.getString("msg"));
|
|
|
+ response.setBody(wvpResult);
|
|
|
+ }
|
|
|
+ return response;
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 停止发流
|
|
|
+ */
|
|
|
+ public RedisRpcResponse stopSendRtp(RedisRpcRequest request) {
|
|
|
+ SendRtpItem sendRtpItem = JSON.parseObject(request.getParam().toString(), SendRtpItem.class);
|
|
|
+ logger.info("[redis-rpc] 停止推流: {}/{}", sendRtpItem.getApp(), sendRtpItem.getStream() );
|
|
|
+ MediaServerItem mediaServerItem = mediaServerService.getOne(sendRtpItem.getMediaServerId());
|
|
|
+ if (mediaServerItem == null) {
|
|
|
+ logger.info("[redis-rpc] stopSendRtp->未找到MediaServer: {}", sendRtpItem.getMediaServerId() );
|
|
|
+ RedisRpcResponse response = request.getResponse();
|
|
|
+ response.setStatusCode(200);
|
|
|
+ }
|
|
|
+ JSONObject jsonObject = zlmServerFactory.stopSendRtpStream(mediaServerItem, sendRtpItem);
|
|
|
+ RedisRpcResponse response = request.getResponse();
|
|
|
+ response.setStatusCode(200);
|
|
|
+ if (jsonObject.getInteger("code") == 0) {
|
|
|
+ logger.info("[redis-rpc] 停止推流成功: {}/{}", sendRtpItem.getApp(), sendRtpItem.getStream() );
|
|
|
+ WVPResult wvpResult = WVPResult.success();
|
|
|
+ response.setBody(wvpResult);
|
|
|
+ }else {
|
|
|
+ int code = jsonObject.getInteger("code");
|
|
|
+ String msg = jsonObject.getString("msg");
|
|
|
+ logger.info("[redis-rpc] 停止推流失败: {}/{}, code: {}, msg: {}", sendRtpItem.getApp(), sendRtpItem.getStream(),code, msg );
|
|
|
+ WVPResult wvpResult = WVPResult.fail(code, msg);
|
|
|
+ response.setBody(wvpResult);
|
|
|
+ }
|
|
|
+ return response;
|
|
|
+ }
|
|
|
+
|
|
|
+ private void sendResponse(RedisRpcResponse response){
|
|
|
+ response.setToId(userSetting.getServerId());
|
|
|
+ RedisRpcMessage message = new RedisRpcMessage();
|
|
|
+ message.setResponse(response);
|
|
|
+ redisTemplate.convertAndSend(RedisRpcConfig.REDIS_REQUEST_CHANNEL_KEY, message);
|
|
|
+ }
|
|
|
+}
|