648540858
2021-11-03 b445128fb5bdc24f6703e7a40e9fd9f40919baf2
src/main/java/com/genersoft/iot/vmp/gb28181/transmit/request/impl/AckRequestProcessor.java
@@ -4,27 +4,35 @@
import java.util.Map;
import javax.sip.*;
//import javax.sip.message.Request;
import javax.sip.address.SipURI;
import javax.sip.header.FromHeader;
import javax.sip.header.HeaderAddress;
import javax.sip.header.ToHeader;
import com.genersoft.iot.vmp.common.StreamInfo;
import com.genersoft.iot.vmp.gb28181.bean.SendRtpItem;
import com.genersoft.iot.vmp.gb28181.transmit.request.SIPRequestAbstractProcessor;
import com.genersoft.iot.vmp.media.zlm.ZLMRTPServerFactory;
import com.genersoft.iot.vmp.media.zlm.dto.MediaServerItem;
import com.genersoft.iot.vmp.service.IMediaServerService;
import com.genersoft.iot.vmp.storager.IRedisCatchStorage;
import org.springframework.stereotype.Component;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
/**    
 * @Description:ACK请求处理器  
 * @author: swwheihei
 * @date:   2020年5月3日 下午5:31:45     
 */
@Component
public class AckRequestProcessor extends SIPRequestAbstractProcessor {
   private Logger logger = LoggerFactory.getLogger(AckRequestProcessor.class);
    private IRedisCatchStorage redisCatchStorage;
   private ZLMRTPServerFactory zlmrtpServerFactory;
   private IMediaServerService mediaServerService;
   /**   
    * 处理  ACK请求
@@ -37,22 +45,29 @@
      Dialog dialog = evt.getDialog();
      if (dialog == null) return;
      //DialogState state = dialog.getState();
      if (/*request.getMethod().equals(Request.INVITE) &&*/ dialog.getState()== DialogState.CONFIRMED) {
         String remoteUri = dialog.getRemoteParty().getURI().toString();
         String localUri = dialog.getLocalParty().getURI().toString();
         String platformGbId = remoteUri.substring(remoteUri.indexOf(":") + 1, remoteUri.indexOf("@"));
         String channelId = localUri.substring(remoteUri.indexOf(":") + 1, remoteUri.indexOf("@"));
      if (/*request.getMecodewwthod().equals(Request.INVITE) &&*/ dialog.getState()== DialogState.CONFIRMED) {
         String platformGbId = ((SipURI) ((HeaderAddress) evt.getRequest().getHeader(FromHeader.NAME)).getAddress().getURI()).getUser();
         String channelId = ((SipURI) ((HeaderAddress) evt.getRequest().getHeader(ToHeader.NAME)).getAddress().getURI()).getUser();
         SendRtpItem sendRtpItem =  redisCatchStorage.querySendRTPServer(platformGbId, channelId);
         String is_Udp = sendRtpItem.isTcp() ? "0" : "1";
         String deviceId = sendRtpItem.getDeviceId();
         StreamInfo streamInfo = redisCatchStorage.queryPlayByDevice(deviceId, channelId);
         sendRtpItem.setStreamId(streamInfo.getStreamId());
         StreamInfo streamInfo = null;
         if (deviceId == null) {
            streamInfo = new StreamInfo();
            streamInfo.setApp(sendRtpItem.getApp());
            streamInfo.setStreamId(sendRtpItem.getStreamId());
         }else {
            streamInfo = redisCatchStorage.queryPlayByDevice(deviceId, channelId);
            sendRtpItem.setStreamId(streamInfo.getStreamId());
            streamInfo.setApp("rtp");
         }
         redisCatchStorage.updateSendRTPSever(sendRtpItem);
         System.out.println(platformGbId);
         System.out.println(channelId);
         logger.info(platformGbId);
         logger.info(channelId);
         Map<String, Object> param = new HashMap<>();
         param.put("vhost","__defaultVhost__");
         param.put("app","rtp");
         param.put("app",streamInfo.getApp());
         param.put("stream",streamInfo.getStreamId());
         param.put("ssrc", sendRtpItem.getSsrc());
         param.put("dst_url",sendRtpItem.getIp());
@@ -65,18 +80,22 @@
         while (!rtpPushed) {
            try {
               if (System.currentTimeMillis() - startTime < 30 * 1000) {
                  if (zlmrtpServerFactory.isRtpReady(streamInfo.getStreamId())) {
                  MediaServerItem mediaInfo = mediaServerService.getOne(sendRtpItem.getMediaServerId());
                  if (zlmrtpServerFactory.isStreamReady(mediaInfo, streamInfo.getApp(), streamInfo.getStreamId())) {
                     rtpPushed = true;
                     System.out.println("已获取设备推流,开始向上级推流");
                     zlmrtpServerFactory.startSendRtpStream(param);
                     logger.info("已获取设备推流[{}/{}],开始向上级推流[{}:{}]",
                           streamInfo.getApp() ,streamInfo.getStreamId(), sendRtpItem.getIp(), sendRtpItem.getPort());
                     zlmrtpServerFactory.startSendRtpStream(mediaInfo, param);
                  } else {
                     System.out.println("等待设备推流.......");
                     Thread.sleep(2000);
                     logger.info("等待设备推流[{}/{}].......",
                           streamInfo.getApp() ,streamInfo.getStreamId());
                     Thread.sleep(1000);
                     continue;
                  }
               } else {
                  rtpPushed = true;
                  System.out.println("设备推流超时,终止向上级推流");
                  logger.info("设备推流[{}/{}]超时,终止向上级推流",
                        streamInfo.getApp() ,streamInfo.getStreamId());
               }
            } catch (InterruptedException e) {
               e.printStackTrace();
@@ -88,7 +107,7 @@
      //    CSeq csReq = (CSeq) request.getHeader(CSeq.NAME);
      //    ackRequest = dialog.createAck(csReq.getSeqNumber());
      //    dialog.sendAck(ackRequest);
      //    System.out.println("send ack to callee:" + ackRequest.toString());
      //    logger.info("send ack to callee:" + ackRequest.toString());
      // } catch (SipException e) {
      //    e.printStackTrace();
      // } catch (InvalidArgumentException e) {
@@ -112,4 +131,12 @@
   public void setZlmrtpServerFactory(ZLMRTPServerFactory zlmrtpServerFactory) {
      this.zlmrtpServerFactory = zlmrtpServerFactory;
   }
   public IMediaServerService getMediaServerService() {
      return mediaServerService;
   }
   public void setMediaServerService(IMediaServerService mediaServerService) {
      this.mediaServerService = mediaServerService;
   }
}