修复回放、停止预览bug，增加ZLM hook监听，长时间无人观看，停止摄像头推流。需开启ZLM的hook并配置url

songww

2020-05-14 805fa42a2c010d92b1c8fc29a60d32fd99fb4ea3

 src/main/java/com/genersoft/iot/vmp/gb28181/transmit/cmd/impl/SIPCommander.java

@@ -1,19 +1,22 @@
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;

@@ -27,13 +30,16 @@
public class SIPCommander implements ISIPCommander {

   

   @Autowired

   private SipConfig config;

   private SipConfig sipConfig;

   

   @Autowired

   private SIPRequestHeaderProvider headerProvider;

   

   @Autowired

   private SipLayer sipLayer;

	
   @Autowired

   private VideoStreamSessionManager streamSession;

   

   /**

    * 云台方向放控制，使用配置文件中的默认镜头移动速度

@@ -46,7 +52,7 @@
    */

   @Override

   public boolean ptzdirectCmd(Device device, String channelId, int leftRight, int upDown) {

      return ptzCmd(device, channelId, leftRight, upDown, 0, config.getSpeed(), 0);

      return ptzCmd(device, channelId, leftRight, upDown, 0, sipConfig.getSpeed(), 0);

   }



   /**

@@ -72,7 +78,7 @@
    */  

   @Override

   public boolean ptzZoomCmd(Device device, String channelId, int inOut) {

      return ptzCmd(device, channelId, 0, 0, inOut, 0, config.getSpeed());

      return ptzCmd(device, channelId, 0, 0, inOut, 0, sipConfig.getSpeed());

   }



   /**

@@ -135,25 +141,70 @@
   public String playStreamCmd(Device device, String channelId) {

      try {

         

         //生成ssrc标识数据流 10位数字

         String ssrc = "";

         Random random = new Random();

         // ZLMediaServer最大识别7FFFFFFF即2147483647，所以随机数不能超过这个数

         ssrc = String.valueOf(random.nextInt(2147483647));

         String ssrc = streamSession.createPlaySsrc();

         String transport = device.getTransport();

         //

         StringBuffer content = new StringBuffer(200);

           content.append("v=0\r\n");

           content.append("o="+channelId+" 0 0 IN IP4 "+config.getSipIp()+"\r\n");

           content.append("o="+channelId+" 0 0 IN IP4 "+sipConfig.getSipIp()+"\r\n");

           content.append("s=Play\r\n");

           content.append("c=IN IP4 "+config.getMediaIp()+"\r\n");

           content.append("c=IN IP4 "+sipConfig.getMediaIp()+"\r\n");

           content.append("t=0 0\r\n");

           if("TCP".equals(transport)) {

              content.append("m=video "+sipConfig.getMediaPort()+" TCP/RTP/AVP 96 98 97\r\n");

         }

           if("UDP".equals(transport)) {

              content.append("m=video "+sipConfig.getMediaPort()+" RTP/AVP 96 98 97\r\n");

         }

           content.append("a=recvonly\r\n");

           content.append("a=rtpmap:96 PS/90000\r\n");

           content.append("a=rtpmap:98 H264/90000\r\n");

           content.append("a=rtpmap:97 MPEG4/90000\r\n");

           if("TCP".equals(transport)){

                content.append("a=setup:passive\r\n");

                content.append("a=connection:new\r\n");

           }

           content.append("y="+ssrc+"\r\n");//ssrc

	        
           Request request = headerProvider.createInviteRequest(device, channelId, content.toString(), null, "live", null);

	
           ClientTransaction transaction = transmitRequest(device, request);

           streamSession.put(ssrc, transaction);

         return ssrc;

      } catch ( SipException | ParseException | InvalidArgumentException e) {

         e.printStackTrace();

         return null;

      } 
   }

	
   /**

    * 请求回放视频流

    * 
    * @param device  视频设备

    * @param channelId  预览通道

    * @param startTime 开始时间,格式要求：yyyy-MM-dd HH:mm:ss

    * @param endTime 结束时间,格式要求：yyyy-MM-dd HH:mm:ss

    */ 
   @Override

   public String playbackStreamCmd(Device device, String channelId, String startTime, String endTime) {

      try {

			
         String ssrc = streamSession.createPlayBackSsrc();

         //

         StringBuffer content = new StringBuffer(200);

           content.append("v=0\r\n");

           content.append("o="+sipConfig.getSipId()+" 0 0 IN IP4 "+sipConfig.getSipIp()+"\r\n");

           content.append("s=Playback\r\n");

           content.append("u="+channelId+":0\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")) {

              content.append("m=video "+config.getMediaPort()+" TCP/RTP/AVP 96 98 97\r\n");

              content.append("m=video "+sipConfig.getMediaPort()+" TCP/RTP/AVP 96 98 97\r\n");

         }

           if(device.getTransport().equals("UDP")) {

              content.append("m=video "+config.getMediaPort()+" RTP/AVP 96 98 97\r\n");

              content.append("m=video "+sipConfig.getMediaPort()+" RTP/AVP 96 98 97\r\n");

         }

           content.append("a=sendrecv\r\n");

           content.append("a=recvonly\r\n");

           content.append("a=rtpmap:96 PS/90000\r\n");

           content.append("a=rtpmap:98 H264/90000\r\n");

           content.append("a=rtpmap:97 MPEG4/90000\r\n");

@@ -163,14 +214,51 @@
           }

           content.append("y="+ssrc+"\r\n");//ssrc

           

           Request request = headerProvider.createInviteRequest(device, content.toString(), null, "live", null);

           Request request = headerProvider.createPlaybackInviteRequest(device, channelId, content.toString(), null, "playback", 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().toUpperCase();

         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();

      }

   }



   /**

@@ -323,22 +411,23 @@
    * @param endTime 结束时间,格式要求：yyyy-MM-dd HH:mm:ss

    */  

   @Override

   public boolean recordInfoQuery(Device device, String startTime, String endTime) {

   public boolean recordInfoQuery(Device device, String channelId, String startTime, String endTime) {

      

      try {

         StringBuffer catalogXml = new StringBuffer(200);

         catalogXml.append("<?xml version=\"1.0\" encoding=\"GB2312\"?>");

         catalogXml.append("<Query>");

         catalogXml.append("<CmdType>RecordInfo</CmdType>");

         catalogXml.append("<SN>" + (int)((Math.random()*9+1)*100000) + "</SN>");

         catalogXml.append("<DeviceID>" + device.getDeviceId() + "</DeviceID>");

         catalogXml.append("<StartTime>" + DateUtil.yyyy_MM_dd_HH_mm_ssToISO8601(startTime) + "</StartTime>");

         catalogXml.append("<EndTime>" + DateUtil.yyyy_MM_dd_HH_mm_ssToISO8601(endTime) + "</EndTime>");

         StringBuffer recordInfoXml = new StringBuffer(200);

         recordInfoXml.append("<?xml version=\"1.0\" encoding=\"GB2312\"?>");

         recordInfoXml.append("<Query>");

         recordInfoXml.append("<CmdType>RecordInfo</CmdType>");

         recordInfoXml.append("<SN>" + (int)((Math.random()*9+1)*100000) + "</SN>");

         recordInfoXml.append("<DeviceID>" + channelId + "</DeviceID>");

         recordInfoXml.append("<StartTime>" + DateUtil.yyyy_MM_dd_HH_mm_ssToISO8601(startTime) + "</StartTime>");

         recordInfoXml.append("<EndTime>" + DateUtil.yyyy_MM_dd_HH_mm_ssToISO8601(endTime) + "</EndTime>");

         recordInfoXml.append("<Secrecy>0</Secrecy>");

         // 大华NVR要求必须增加一个值为all的文本元素节点Type

         catalogXml.append("<Type>all</Type>");

         catalogXml.append("</Query>");

         recordInfoXml.append("<Type>all</Type>");

         recordInfoXml.append("</Query>");

         

         Request request = headerProvider.createMessageRequest(device, catalogXml.toString(), "ViaRecordInfoBranch", "FromRecordInfoTag", "ToRecordInfoTag");

         Request request = headerProvider.createMessageRequest(device, recordInfoXml.toString(), "ViaRecordInfoBranch", "FromRecordInfoTag", "ToRecordInfoTag");

         transmitRequest(device, request);

      } catch (SipException | ParseException | InvalidArgumentException e) {

         e.printStackTrace();

@@ -391,11 +480,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;

   }



}