添加拉流代理与国标关联, 支持代理rtsp/rtmp/...，转发到国标

panlinlin

2021-04-01 7dc8fd4a1e8c5afb6fad53454935419c239838c0

 src/main/java/com/genersoft/iot/vmp/gb28181/transmit/request/impl/InviteRequestProcessor.java

@@ -12,13 +12,12 @@
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.gb28181.bean.*;

import com.genersoft.iot.vmp.gb28181.event.SipSubscribe;

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.ZLMHttpHookSubscribe;

import com.genersoft.iot.vmp.media.zlm.ZLMRTPServerFactory;

import com.genersoft.iot.vmp.storager.IRedisCatchStorage;

import com.genersoft.iot.vmp.storager.IVideoManagerStorager;

@@ -30,6 +29,8 @@
import org.slf4j.LoggerFactory;



import java.text.ParseException;

import java.util.HashMap;

import java.util.Map;

import java.util.Vector;



/**    

@@ -93,12 +94,14 @@
         if (platform != null) {

            // 查询平台下是否有该通道

            DeviceChannel channel = storager.queryChannelInParentPlatform(requesterId, channelId);

            if (channel == null) {

            GbStream gbStream = storager.queryStreamInParentPlatform(requesterId, channelId);

            // 不是通道可能是直播流

            if (channel != null || gbStream != null ) {

               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，呼叫转接中

            }

            // 解析sdp消息, 使用jainsip 自带的sdp解析方式

            String contentString = new String(request.getRawContent());

@@ -153,14 +156,17 @@
            String addressStr = sdp.getOrigin().getAddress();

            //String sessionName = sdp.getSessionName().getValue();

            logger.info("[上级点播]用户：{}， 地址：{}:{}， ssrc：{}", username, addressStr, port, ssrc);



            Device device = storager.queryVideoDeviceByPlatformIdAndChannelId(requesterId, channelId);

            Device device  = null;

            // 通过 channel 和 gbStream 是否为null 值判断来源是直播流合适国标

            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,

               SendRtpItem sendRtpItem = zlmrtpServerFactory.createSendRtpItem(addressStr, port, ssrc, requesterId,

                     device.getDeviceId(), channelId,

                  mediaTransmissionTCP);

            if (tcpActive != null) {

               sendRtpItem.setTcpActive(tcpActive);

@@ -176,6 +182,7 @@
            // 通知下级推流，

            PlayResult playResult = playService.play(device.getDeviceId(), channelId, (responseJSON)->{

               // 收到推流， 回复200OK, 等待ack

                  if (sendRtpItem == null) return;

               sendRtpItem.setStatus(1);

               redisCatchStorage.updateSendRTPSever(sendRtpItem);

               // TODO 添加对tcp的支持

@@ -214,6 +221,55 @@
            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("服务器端口资源不足");

                  responseAck(evt, Response.BUSY_HERE);

                  return;

               }



               // 写入redis， 超时时回复

               redisCatchStorage.updateSendRTPSever(sendRtpItem);



               // 检测直播流是否在线

               Boolean streamReady = zlmrtpServerFactory.isStreamReady(gbStream.getApp(), gbStream.getStream());

               if (streamReady) {

                  sendRtpItem.setStatus(1);

                  redisCatchStorage.updateSendRTPSever(sendRtpItem);

                  // TODO 添加对tcp的支持

                  MediaServerConfig 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");



                  try {

                     responseAck(evt, content.toString());

                  } catch (SipException e) {

                     e.printStackTrace();

                  } catch (InvalidArgumentException e) {

                     e.printStackTrace();

                  } catch (ParseException e) {

                     e.printStackTrace();

                  }

               }

            }



         } else {

            // 非上级平台请求，查询是否设备请求（通常为接收语音广播的设备）

            Device device = storager.queryVideoDevice(requesterId);

@@ -298,6 +354,7 @@
      }

   }





   /***

    * 回复状态码

    * 100 trying