修复级联注册bug

64850858

2021-06-28 b24be12405ac4db1e9874367ae27c81aae172495

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

@@ -1,20 +1,42 @@
package com.genersoft.iot.vmp.gb28181.transmit.request.impl;



import javax.sip.address.SipURI;

import javax.sip.Dialog;

import javax.sip.DialogState;

import javax.sip.InvalidArgumentException;

import javax.sip.RequestEvent;

import javax.sip.SipException;

import javax.sip.header.FromHeader;

import javax.sip.header.HeaderAddress;

import javax.sip.header.ToHeader;

import javax.sip.message.Response;



import com.genersoft.iot.vmp.gb28181.bean.SendRtpItem;

import com.genersoft.iot.vmp.gb28181.transmit.cmd.ISIPCommander;

import com.genersoft.iot.vmp.gb28181.transmit.request.SIPRequestAbstractProcessor;

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

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

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;



import java.text.ParseException;

import java.util.HashMap;

import java.util.Map;



/**    

 * @Description: BYE请求处理器

 * @author: swwheihei

 * @date:   2020年5月3日 下午5:32:05     
 * @author: lawrencehj

 * @date:   2021年3月9日     
 */

public class ByeRequestProcessor extends SIPRequestAbstractProcessor {



   private Logger logger = LoggerFactory.getLogger(ByeRequestProcessor.class);



   private ISIPCommander cmder;



   private IRedisCatchStorage redisCatchStorage;



   private ZLMRTPServerFactory zlmrtpServerFactory;



   /**

    * 处理BYE请求

@@ -24,6 +46,27 @@
   public void process(RequestEvent evt) {

      try {

         responseAck(evt);

         Dialog dialog = evt.getDialog();

         if (dialog == null) return;

         if (dialog.getState().equals(DialogState.TERMINATED)) {

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

            if (sendRtpItem == null) return;

            String streamId = sendRtpItem.getStreamId();

            Map<String, Object> param = new HashMap<>();

            param.put("vhost","__defaultVhost__");

            param.put("app",sendRtpItem.getApp());

            param.put("stream",streamId);

            param.put("ssrc",sendRtpItem.getSsrc());

            logger.info("停止向上级推流：" + streamId);

            zlmrtpServerFactory.stopSendRtpStream(param);

            redisCatchStorage.deleteSendRTPServer(platformGbId, channelId);

            if (zlmrtpServerFactory.totalReaderCount(sendRtpItem.getApp(), streamId) == 0) {

               logger.info(streamId + "无其它观看者，通知设备停止推流");

               cmder.streamByeCmd(sendRtpItem.getDeviceId(), channelId);

            }

         }

      } catch (SipException e) {

         e.printStackTrace();

      } catch (InvalidArgumentException e) {

@@ -31,8 +74,6 @@
      } catch (ParseException e) {

         e.printStackTrace();

      }

      // TODO 优先级99 Bye Request消息实现，此消息一般为级联消息，上级给下级发送视频停止指令

		
   }



   /***

@@ -47,4 +88,28 @@
      getServerTransaction(evt).sendResponse(response);

   }



   public IRedisCatchStorage getRedisCatchStorage() {

      return redisCatchStorage;

   }



   public void setRedisCatchStorage(IRedisCatchStorage redisCatchStorage) {

      this.redisCatchStorage = redisCatchStorage;

   }



   public ZLMRTPServerFactory getZlmrtpServerFactory() {

      return zlmrtpServerFactory;

   }



   public void setZlmrtpServerFactory(ZLMRTPServerFactory zlmrtpServerFactory) {

      this.zlmrtpServerFactory = zlmrtpServerFactory;

   }



   public ISIPCommander getSIPCommander() {

      return cmder;

   }



   public void setSIPCommander(ISIPCommander cmder) {

      this.cmder = cmder;

   }



}