songww
2020-05-13 6ecd801c2365feb4e65f6684065aa97f11615797
src/main/java/com/genersoft/iot/vmp/gb28181/transmit/cmd/impl/SIPCommander.java
@@ -1,25 +1,25 @@
package com.genersoft.iot.vmp.gb28181.transmit.cmd.impl;
import java.text.ParseException;
import java.util.Random;
import javax.sip.ClientTransaction;
import javax.sip.Dialog;
import javax.sip.InvalidArgumentException;
import javax.sip.SipException;
import javax.sip.TransactionDoesNotExistException;
import javax.sip.header.ViaHeader;
import javax.sip.message.Request;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.ApplicationEventPublisher;
import org.springframework.stereotype.Component;
import com.genersoft.iot.vmp.conf.SipConfig;
import com.genersoft.iot.vmp.gb28181.SipLayer;
import com.genersoft.iot.vmp.gb28181.bean.Device;
import com.genersoft.iot.vmp.gb28181.session.VideoStreamSessionManager;
import com.genersoft.iot.vmp.gb28181.transmit.cmd.ISIPCommander;
import com.genersoft.iot.vmp.gb28181.transmit.cmd.SIPRequestHeaderProvider;
import com.genersoft.iot.vmp.gb28181.utils.DateUtil;
import com.genersoft.iot.vmp.gb28181.utils.SsrcUtil;
import tk.mybatis.mapper.util.StringUtil;
/**    
 * @Description:设备能力接口,用于定义设备的控制、查询能力   
@@ -37,6 +37,9 @@
   
   @Autowired
   private SipLayer sipLayer;
   @Autowired
   private VideoStreamSessionManager streamSession;
   
   /**
    * 云台方向放控制,使用配置文件中的默认镜头移动速度
@@ -138,11 +141,11 @@
   public String playStreamCmd(Device device, String channelId) {
      try {
         
         String ssrc = SsrcUtil.getPlaySsrc();
         String ssrc = streamSession.createPlaySsrc();
         //
         StringBuffer content = new StringBuffer(200);
           content.append("v=0\r\n");
           content.append("o="+channelId+" 0 0 IN IP4 "+sipConfig.getSipIp()+"\r\n");
           content.append("o="+channelId+" 0 0 IN IP4 "+sipConfig.getMediaIp()+"\r\n");
           content.append("s=Play\r\n");
           content.append("c=IN IP4 "+sipConfig.getMediaIp()+"\r\n");
           content.append("t=0 0\r\n");
@@ -164,7 +167,8 @@
           
           Request request = headerProvider.createInviteRequest(device, content.toString(), null, "live", null);
   
           transmitRequest(device, request);
           ClientTransaction transaction = transmitRequest(device, request);
           streamSession.put(ssrc, transaction);
         return ssrc;
      } catch ( SipException | ParseException | InvalidArgumentException e) {
         e.printStackTrace();
@@ -181,16 +185,16 @@
    * @param endTime 结束时间,格式要求:yyyy-MM-dd HH:mm:ss
    */ 
   @Override
   public String playbackStreamCmd(Device device, String channelId, String recordId, String startTime, String endTime) {
   public String playbackStreamCmd(Device device, String channelId, String startTime, String endTime) {
      try {
         
         String ssrc = SsrcUtil.getPlayBackSsrc();
         String ssrc = streamSession.createPlayBackSsrc();
         //
         StringBuffer content = new StringBuffer(200);
           content.append("v=0\r\n");
           content.append("o="+channelId+" 0 0 IN IP4 "+sipConfig.getSipIp()+"\r\n");
           content.append("o="+device.getDeviceId()+" 0 0 IN IP4 "+sipConfig.getMediaIp()+"\r\n");
           content.append("s=Playback\r\n");
           content.append("u="+recordId+":3\r\n");
           content.append("u="+channelId+":3\r\n");
           content.append("c=IN IP4 "+sipConfig.getMediaIp()+"\r\n");
           content.append("t="+DateUtil.yyyy_MM_dd_HH_mm_ssToTimestamp(startTime)+" "+DateUtil.yyyy_MM_dd_HH_mm_ssToTimestamp(endTime) +"\r\n");
           if(device.getTransport().equals("TCP")) {
@@ -211,11 +215,48 @@
           
           Request request = headerProvider.createInviteRequest(device, content.toString(), null, "live", null);
   
           transmitRequest(device, request);
           ClientTransaction transaction = transmitRequest(device, request);
           streamSession.put(ssrc, transaction);
         return ssrc;
      } catch ( SipException | ParseException | InvalidArgumentException e) {
         e.printStackTrace();
         return null;
      }
   }
   /**
    * 视频流停止
    *
    * @param device  视频设备
    * @param channelId  预览通道
    */
   @Override
   public void streamByeCmd(String ssrc) {
      try {
         ClientTransaction transaction = streamSession.get(ssrc);
         if (transaction == null) {
            return;
         }
         Dialog dialog = transaction.getDialog();
         if (dialog == null) {
            return;
         }
         Request byeRequest = dialog.createRequest(Request.BYE);
         ViaHeader viaHeader = (ViaHeader) byeRequest.getHeader(ViaHeader.NAME);
         String protocol = viaHeader.getTransport();
         ClientTransaction clientTransaction = null;
         if("TCP".equals(protocol)) {
            clientTransaction = sipLayer.getTcpSipProvider().getNewClientTransaction(byeRequest);
         } else if("UDP".equals(protocol)) {
            clientTransaction = sipLayer.getUdpSipProvider().getNewClientTransaction(byeRequest);
         }
         dialog.sendRequest(clientTransaction);
      } catch (TransactionDoesNotExistException e) {
         e.printStackTrace();
      } catch (SipException e) {
         e.printStackTrace();
      }
   }
@@ -438,12 +479,15 @@
      return false;
   }
   
   private void transmitRequest(Device device, Request request) throws SipException {
      if(device.getTransport().equals("TCP")) {
         sipLayer.getTcpSipProvider().sendRequest(request);
      } else if(device.getTransport().equals("UDP")) {
         sipLayer.getUdpSipProvider().sendRequest(request);
   private ClientTransaction transmitRequest(Device device, Request request) throws SipException {
      ClientTransaction clientTransaction = null;
      if("TCP".equals(device.getTransport())) {
         clientTransaction = sipLayer.getTcpSipProvider().getNewClientTransaction(request);
      } else if("UDP".equals(device.getTransport())) {
         clientTransaction = sipLayer.getUdpSipProvider().getNewClientTransaction(request);
      }
      clientTransaction.sendRequest();
      return clientTransaction;
   }
}