panlinlin
2021-05-08 e48fa711a3664bece9b3e58840a75fe7c05bc47c
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;
   }
}