xiaoxie
2022-04-18 6bb8f45e270681dbcb70aed663b406e7d905e091
src/main/java/com/genersoft/iot/vmp/gb28181/transmit/event/request/impl/SubscribeRequestProcessor.java
@@ -1,18 +1,33 @@
package com.genersoft.iot.vmp.gb28181.transmit.event.request.impl;
import com.genersoft.iot.vmp.common.VideoManagerConstants;
import com.genersoft.iot.vmp.conf.DynamicTask;
import com.genersoft.iot.vmp.conf.UserSetting;
import com.genersoft.iot.vmp.gb28181.bean.CmdType;
import com.genersoft.iot.vmp.gb28181.bean.ParentPlatform;
import com.genersoft.iot.vmp.gb28181.bean.SubscribeHolder;
import com.genersoft.iot.vmp.gb28181.bean.SubscribeInfo;
import com.genersoft.iot.vmp.gb28181.task.impl.MobilePositionSubscribeHandlerTask;
import com.genersoft.iot.vmp.gb28181.transmit.SIPProcessorObserver;
import com.genersoft.iot.vmp.gb28181.transmit.cmd.ISIPCommanderForPlatform;
import com.genersoft.iot.vmp.gb28181.transmit.event.request.ISIPRequestProcessor;
import com.genersoft.iot.vmp.gb28181.transmit.event.request.SIPRequestProcessorParent;
import com.genersoft.iot.vmp.gb28181.utils.SipUtils;
import com.genersoft.iot.vmp.gb28181.utils.XmlUtil;
import com.genersoft.iot.vmp.storager.IRedisCatchStorage;
import com.genersoft.iot.vmp.storager.IVideoManagerStorage;
import gov.nist.javax.sip.SipProviderImpl;
import org.dom4j.DocumentException;
import org.dom4j.Element;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.InitializingBean;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.context.annotation.Lazy;
import org.springframework.stereotype.Component;
import javax.sip.InvalidArgumentException;
import javax.sip.RequestEvent;
import javax.sip.ServerTransaction;
import javax.sip.SipException;
import javax.sip.*;
import javax.sip.header.ExpiresHeader;
import javax.sip.message.Request;
import javax.sip.message.Response;
@@ -30,6 +45,34 @@
   @Autowired
   private SIPProcessorObserver sipProcessorObserver;
   @Autowired
   private IRedisCatchStorage redisCatchStorage;
   @Autowired
   private ISIPCommanderForPlatform sipCommanderForPlatform;
   @Autowired
   private IVideoManagerStorage storager;
   @Lazy
   @Autowired
   @Qualifier(value="tcpSipProvider")
   private SipProviderImpl tcpSipProvider;
   @Lazy
   @Autowired
   @Qualifier(value="udpSipProvider")
   private SipProviderImpl udpSipProvider;
   @Autowired
   private DynamicTask dynamicTask;
   @Autowired
   private UserSetting userSetting;
   @Autowired
   private SubscribeHolder subscribeHolder;
   @Override
   public void afterPropertiesSet() throws Exception {
      // 添加消息处理的订阅
@@ -46,30 +89,155 @@
      Request request = evt.getRequest();
      try {
         Response response = null;
         response = getMessageFactory().createResponse(200, request);
         if (response != null) {
            ExpiresHeader expireHeader = getHeaderFactory().createExpiresHeader(30);
            response.setExpires(expireHeader);
         }
         logger.info("response : " + response.toString());
         ServerTransaction transaction = getServerTransaction(evt);
         if (transaction != null) {
            transaction.sendResponse(response);
            transaction.getDialog().delete();
            transaction.terminate();
         Element rootElement = getRootElement(evt);
         String cmd = XmlUtil.getText(rootElement, "CmdType");
         if (CmdType.MOBILE_POSITION.equals(cmd)) {
            processNotifyMobilePosition(evt, rootElement);
//         } else if (CmdType.ALARM.equals(cmd)) {
//            logger.info("接收到Alarm订阅");
//            processNotifyAlarm(evt, rootElement);
         } else if (CmdType.CATALOG.equals(cmd)) {
            processNotifyCatalogList(evt, rootElement);
         } else {
            logger.info("processRequest serverTransactionId is null.");
         }
            logger.info("接收到消息:" + cmd);
            Response response = null;
            response = getMessageFactory().createResponse(200, request);
            if (response != null) {
               ExpiresHeader expireHeader = getHeaderFactory().createExpiresHeader(30);
               response.setExpires(expireHeader);
            }
            logger.info("response : " + response.toString());
            ServerTransaction transaction = getServerTransaction(evt);
            if (transaction != null) {
               transaction.sendResponse(response);
               transaction.getDialog().delete();
               transaction.terminate();
            } else {
               logger.info("processRequest serverTransactionId is null.");
            }
         }
      } catch (ParseException e) {
         e.printStackTrace();
      } catch (SipException e) {
         e.printStackTrace();
      } catch (InvalidArgumentException e) {
         e.printStackTrace();
      } catch (DocumentException e) {
         e.printStackTrace();
      }
   }
   /**
    * 处理移动位置订阅消息
    */
   private void processNotifyMobilePosition(RequestEvent evt, Element rootElement) throws SipException {
      String platformId = SipUtils.getUserIdFromFromHeader(evt.getRequest());
      String deviceID = XmlUtil.getText(rootElement, "DeviceID");
      ParentPlatform platform = storager.queryParentPlatByServerGBId(platformId);
      SubscribeInfo subscribeInfo = new SubscribeInfo(evt, platformId);
      if (platform == null) {
         return;
      }
      if (evt.getServerTransaction() == null) {
         ServerTransaction serverTransaction = platform.getTransport().equals("TCP") ? tcpSipProvider.getNewServerTransaction(evt.getRequest())
               : udpSipProvider.getNewServerTransaction(evt.getRequest());
         subscribeInfo.setTransaction(serverTransaction);
         Dialog dialog = serverTransaction.getDialog();
         dialog.terminateOnBye(false);
         subscribeInfo.setDialog(dialog);
      }
      String sn = XmlUtil.getText(rootElement, "SN");
      logger.info("[回复 移动位置订阅]: {}", platformId);
      StringBuilder resultXml = new StringBuilder(200);
      resultXml.append("<?xml version=\"1.0\" ?>\r\n")
            .append("<Response>\r\n")
            .append("<CmdType>MobilePosition</CmdType>\r\n")
            .append("<SN>" + sn + "</SN>\r\n")
            .append("<DeviceID>" + deviceID + "</DeviceID>\r\n")
            .append("<Result>OK</Result>\r\n")
            .append("</Response>\r\n");
      if (subscribeInfo.getExpires() > 0) {
         String interval = XmlUtil.getText(rootElement, "Interval"); // GPS上报时间间隔
         if (interval == null) {
            subscribeInfo.setGpsInterval(5);
         }else {
            subscribeInfo.setGpsInterval(Integer.parseInt(interval));
         }
         subscribeInfo.setSn(sn);
         subscribeHolder.putMobilePositionSubscribe(platformId, subscribeInfo);
//         if (subscribeHolder.getMobilePositionSubscribe(platformId) == null ) {
//            subscribeHolder.putMobilePositionSubscribe(platformId, subscribeInfo);
//         }else {
//            if (subscribeHolder.getMobilePositionSubscribe(platformId).getDialog() != null
//                  && subscribeHolder.getMobilePositionSubscribe(platformId).getDialog().getState() != null
//                  && !subscribeHolder.getMobilePositionSubscribe(platformId).getDialog().getState().equals(DialogState.CONFIRMED)) {
//               subscribeHolder.putMobilePositionSubscribe(platformId, subscribeInfo);
//            }
//         }
      }else if (subscribeInfo.getExpires() == 0) {
         subscribeHolder.removeMobilePositionSubscribe(platformId);
      }
      try {
         ParentPlatform parentPlatform = storager.queryParentPlatByServerGBId(platformId);
         responseXmlAck(evt, resultXml.toString(), parentPlatform);
      } catch (SipException e) {
         e.printStackTrace();
      } catch (InvalidArgumentException e) {
         e.printStackTrace();
      } catch (ParseException e) {
         e.printStackTrace();
      }
   }
   private void processNotifyAlarm(RequestEvent evt, Element rootElement) {
   }
   private void processNotifyCatalogList(RequestEvent evt, Element rootElement) throws SipException {
      String platformId = SipUtils.getUserIdFromFromHeader(evt.getRequest());
      String deviceID = XmlUtil.getText(rootElement, "DeviceID");
      ParentPlatform platform = storager.queryParentPlatByServerGBId(platformId);
      SubscribeInfo subscribeInfo = new SubscribeInfo(evt, platformId);
      if (evt.getServerTransaction() == null) {
         ServerTransaction serverTransaction = platform.getTransport().equals("TCP") ? tcpSipProvider.getNewServerTransaction(evt.getRequest())
               : udpSipProvider.getNewServerTransaction(evt.getRequest());
         subscribeInfo.setTransaction(serverTransaction);
         Dialog dialog = serverTransaction.getDialog();
         dialog.terminateOnBye(false);
         subscribeInfo.setDialog(dialog);
      }
      String sn = XmlUtil.getText(rootElement, "SN");
      logger.info("[回复 目录订阅]: {}/{}", platformId, deviceID);
      StringBuilder resultXml = new StringBuilder(200);
      resultXml.append("<?xml version=\"1.0\" ?>\r\n")
            .append("<Response>\r\n")
            .append("<CmdType>Catalog</CmdType>\r\n")
            .append("<SN>" + sn + "</SN>\r\n")
            .append("<DeviceID>" + deviceID + "</DeviceID>\r\n")
            .append("<Result>OK</Result>\r\n")
            .append("</Response>\r\n");
      if (subscribeInfo.getExpires() > 0) {
         subscribeHolder.putCatalogSubscribe(platformId, subscribeInfo);
      }else if (subscribeInfo.getExpires() == 0) {
         subscribeHolder.removeCatalogSubscribe(platformId);
      }
      try {
         ParentPlatform parentPlatform = storager.queryParentPlatByServerGBId(platformId);
         responseXmlAck(evt, resultXml.toString(), parentPlatform);
      } catch (SipException e) {
         e.printStackTrace();
      } catch (InvalidArgumentException e) {
         e.printStackTrace();
      } catch (ParseException e) {
         e.printStackTrace();
      }
   }
}