增加SSE前端推送支持

Lawrence

2021-01-20 9789c55ef944ffdbfd7b2bdaf9d2a210f6572a32

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

@@ -7,10 +7,16 @@
import javax.sip.InvalidArgumentException;

import javax.sip.RequestEvent;

import javax.sip.SipException;

import javax.sip.header.FromHeader;

import javax.sip.message.Request;

import javax.sip.message.Response;



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

import com.genersoft.iot.vmp.gb28181.transmit.cmd.impl.SIPCommanderFroPlatform;

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

import com.genersoft.iot.vmp.vmanager.platform.bean.ChannelReduce;

import gov.nist.javax.sip.address.AddressImpl;

import gov.nist.javax.sip.address.SipUri;

import org.dom4j.Document;

import org.dom4j.DocumentException;

import org.dom4j.Element;

@@ -20,10 +26,6 @@
import org.springframework.beans.factory.annotation.Autowired;



import com.genersoft.iot.vmp.common.VideoManagerConstants;

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

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

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

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

import com.genersoft.iot.vmp.gb28181.event.DeviceOffLineDetector;

import com.genersoft.iot.vmp.gb28181.event.EventPublisher;

import com.genersoft.iot.vmp.gb28181.transmit.callback.DeferredResultHolder;

@@ -46,6 +48,8 @@
   private final static Logger logger = LoggerFactory.getLogger(MessageRequestProcessor.class);



   private SIPCommander cmder;



   private SIPCommanderFroPlatform cmderFroPlatform;



   private IVideoManagerStorager storager;



@@ -160,113 +164,144 @@
   private void processMessageCatalogList(RequestEvent evt) {

      try {

         Element rootElement = getRootElement(evt);

         String name = rootElement.getName();

         Element deviceIdElement = rootElement.element("DeviceID");

         String deviceId = deviceIdElement.getText();

         Element deviceListElement = rootElement.element("DeviceList");

         if (deviceListElement == null) {

            return;

         }

         Iterator<Element> deviceListIterator = deviceListElement.elementIterator();

         if (deviceListIterator != null) {

            Device device = storager.queryVideoDevice(deviceId);

            if (device == null) {



         FromHeader fromHeader = (FromHeader) evt.getRequest().getHeader(FromHeader.NAME);

         AddressImpl address = (AddressImpl) fromHeader.getAddress();

         SipUri uri = (SipUri) address.getURI();

         String platformId = uri.getUser();

         // if (deviceListElement == null) { // 存在DeviceList则为响应 catalog， 不存在DeviceList则为查询请求

         if (name.equalsIgnoreCase("Query")) { // 区分是Response——查询响应，还是Query——查询请求

               // TODO 后续将代码拆分

            ParentPlatform parentPlatform = storager.queryParentPlatById(platformId);

            if (parentPlatform == null) {

               response404Ack(evt);

               return;

            }

            // 遍历DeviceList

            while (deviceListIterator.hasNext()) {

               Element itemDevice = deviceListIterator.next();

               Element channelDeviceElement = itemDevice.element("DeviceID");

               if (channelDeviceElement == null) {

                  continue;

               }

               String channelDeviceId = channelDeviceElement.getText();

               Element channdelNameElement = itemDevice.element("Name");

               String channelName = channdelNameElement != null ? channdelNameElement.getTextTrim().toString() : "";

               Element statusElement = itemDevice.element("Status");

               String status = statusElement != null ? statusElement.getText().toString() : "ON";

               DeviceChannel deviceChannel = new DeviceChannel();

               deviceChannel.setName(channelName);

               deviceChannel.setChannelId(channelDeviceId);

               // ONLINE OFFLINE  HIKVISION DS-7716N-E4 NVR的兼容性处理

               if (status.equals("ON") || status.equals("On") || status.equals("ONLINE")) {

                  deviceChannel.setStatus(1);

               }

               if (status.equals("OFF") || status.equals("Off") || status.equals("OFFLINE")) {

                  deviceChannel.setStatus(0);

            }else {

               // 回复200 OK

               responseAck(evt);



               Element snElement = rootElement.element("SN");

               String sn = snElement.getText();

               // 准备回复通道信息

               List<ChannelReduce> channelReduces = storager.queryChannelListInParentPlatform(parentPlatform.getServerGBId());

               if (channelReduces.size() >0 ) {

                  for (ChannelReduce channelReduce : channelReduces) {

                     DeviceChannel deviceChannel = storager.queryChannel(channelReduce.getDeviceId(), channelReduce.getChannelId());

                     cmderFroPlatform.catalogQuery(deviceChannel, parentPlatform, sn, fromHeader.getTag(), channelReduces.size());

                  }

               }



               deviceChannel.setManufacture(XmlUtil.getText(itemDevice, "Manufacturer"));

               deviceChannel.setModel(XmlUtil.getText(itemDevice, "Model"));

               deviceChannel.setOwner(XmlUtil.getText(itemDevice, "Owner"));

               deviceChannel.setCivilCode(XmlUtil.getText(itemDevice, "CivilCode"));

               deviceChannel.setBlock(XmlUtil.getText(itemDevice, "Block"));

               deviceChannel.setAddress(XmlUtil.getText(itemDevice, "Address"));

               if (XmlUtil.getText(itemDevice, "Parental") == null || XmlUtil.getText(itemDevice, "Parental") == "") {

                  deviceChannel.setParental(0);

               } else {

                  deviceChannel.setParental(Integer.parseInt(XmlUtil.getText(itemDevice, "Parental")));

               } 
               deviceChannel.setParentId(XmlUtil.getText(itemDevice, "ParentID"));

               if (XmlUtil.getText(itemDevice, "SafetyWay") == null || XmlUtil.getText(itemDevice, "SafetyWay")== "") {

                  deviceChannel.setSafetyWay(0);

               } else {

                  deviceChannel.setSafetyWay(Integer.parseInt(XmlUtil.getText(itemDevice, "SafetyWay")));

               }

               if (XmlUtil.getText(itemDevice, "RegisterWay") == null || XmlUtil.getText(itemDevice, "RegisterWay") =="") {

                  deviceChannel.setRegisterWay(1);

               } else {

                  deviceChannel.setRegisterWay(Integer.parseInt(XmlUtil.getText(itemDevice, "RegisterWay")));

               }

               deviceChannel.setCertNum(XmlUtil.getText(itemDevice, "CertNum"));

               if (XmlUtil.getText(itemDevice, "Certifiable") == null || XmlUtil.getText(itemDevice, "Certifiable") == "") {

                  deviceChannel.setCertifiable(0);

               } else {

                  deviceChannel.setCertifiable(Integer.parseInt(XmlUtil.getText(itemDevice, "Certifiable")));

               }

               if (XmlUtil.getText(itemDevice, "ErrCode") == null || XmlUtil.getText(itemDevice, "ErrCode") == "") {

                  deviceChannel.setErrCode(0);

               } else {

                  deviceChannel.setErrCode(Integer.parseInt(XmlUtil.getText(itemDevice, "ErrCode")));

               }

               deviceChannel.setEndTime(XmlUtil.getText(itemDevice, "EndTime"));

               deviceChannel.setSecrecy(XmlUtil.getText(itemDevice, "Secrecy"));

               deviceChannel.setIpAddress(XmlUtil.getText(itemDevice, "IPAddress"));

               if (XmlUtil.getText(itemDevice, "Port") == null || XmlUtil.getText(itemDevice, "Port") =="") {

                  deviceChannel.setPort(0);

               } else {

                  deviceChannel.setPort(Integer.parseInt(XmlUtil.getText(itemDevice, "Port")));

               }

               deviceChannel.setPassword(XmlUtil.getText(itemDevice, "Password"));

               if (XmlUtil.getText(itemDevice, "Longitude") == null || XmlUtil.getText(itemDevice, "Longitude") == "") {

                  deviceChannel.setLongitude(0.00);

               } else {

                  deviceChannel.setLongitude(Double.parseDouble(XmlUtil.getText(itemDevice, "Longitude")));

               }

               if (XmlUtil.getText(itemDevice, "Latitude") == null || XmlUtil.getText(itemDevice, "Latitude") =="") {

                  deviceChannel.setLatitude(0.00);

               } else {

                  deviceChannel.setLatitude(Double.parseDouble(XmlUtil.getText(itemDevice, "Latitude")));

               }

               if (XmlUtil.getText(itemDevice, "PTZType") == null || XmlUtil.getText(itemDevice, "PTZType") == "") {

                  deviceChannel.setPTZType(0);

               } else {

                  deviceChannel.setPTZType(Integer.parseInt(XmlUtil.getText(itemDevice, "PTZType")));

               }

               deviceChannel.setHasAudio(true); // 默认含有音频，播放时再检查是否有音频及是否AAC

               storager.updateChannel(device.getDeviceId(), deviceChannel);

            }



            RequestMessage msg = new RequestMessage();

            msg.setDeviceId(deviceId);

            msg.setType(DeferredResultHolder.CALLBACK_CMD_CATALOG);

            msg.setData(device);

            deferredResultHolder.invokeResult(msg);

            // 回复200 OK

            responseAck(evt);

            if (offLineDetector.isOnline(deviceId)) {

               publisher.onlineEventPublish(deviceId, VideoManagerConstants.EVENT_ONLINE_KEEPLIVE);



         }else {

            Iterator<Element> deviceListIterator = deviceListElement.elementIterator();

            if (deviceListIterator != null) {

               Device device = storager.queryVideoDevice(deviceId);

               if (device == null) {

                  return;

               }

               // 遍历DeviceList

               while (deviceListIterator.hasNext()) {

                  Element itemDevice = deviceListIterator.next();

                  Element channelDeviceElement = itemDevice.element("DeviceID");

                  if (channelDeviceElement == null) {

                     continue;

                  }

                  String channelDeviceId = channelDeviceElement.getText();

                  Element channdelNameElement = itemDevice.element("Name");

                  String channelName = channdelNameElement != null ? channdelNameElement.getTextTrim().toString() : "";

                  Element statusElement = itemDevice.element("Status");

                  String status = statusElement != null ? statusElement.getText().toString() : "ON";

                  DeviceChannel deviceChannel = new DeviceChannel();

                  deviceChannel.setName(channelName);

                  deviceChannel.setChannelId(channelDeviceId);

                  // ONLINE OFFLINE  HIKVISION DS-7716N-E4 NVR的兼容性处理

                  if (status.equals("ON") || status.equals("On") || status.equals("ONLINE")) {

                     deviceChannel.setStatus(1);

                  }

                  if (status.equals("OFF") || status.equals("Off") || status.equals("OFFLINE")) {

                     deviceChannel.setStatus(0);

                  }



                  deviceChannel.setManufacture(XmlUtil.getText(itemDevice, "Manufacturer"));

                  deviceChannel.setModel(XmlUtil.getText(itemDevice, "Model"));

                  deviceChannel.setOwner(XmlUtil.getText(itemDevice, "Owner"));

                  deviceChannel.setCivilCode(XmlUtil.getText(itemDevice, "CivilCode"));

                  deviceChannel.setBlock(XmlUtil.getText(itemDevice, "Block"));

                  deviceChannel.setAddress(XmlUtil.getText(itemDevice, "Address"));

                  if (XmlUtil.getText(itemDevice, "Parental") == null || XmlUtil.getText(itemDevice, "Parental") == "") {

                     deviceChannel.setParental(0);

                  } else {

                     deviceChannel.setParental(Integer.parseInt(XmlUtil.getText(itemDevice, "Parental")));

                  }

                  deviceChannel.setParentId(XmlUtil.getText(itemDevice, "ParentID"));

                  if (XmlUtil.getText(itemDevice, "SafetyWay") == null || XmlUtil.getText(itemDevice, "SafetyWay")== "") {

                     deviceChannel.setSafetyWay(0);

                  } else {

                     deviceChannel.setSafetyWay(Integer.parseInt(XmlUtil.getText(itemDevice, "SafetyWay")));

                  }

                  if (XmlUtil.getText(itemDevice, "RegisterWay") == null || XmlUtil.getText(itemDevice, "RegisterWay") =="") {

                     deviceChannel.setRegisterWay(1);

                  } else {

                     deviceChannel.setRegisterWay(Integer.parseInt(XmlUtil.getText(itemDevice, "RegisterWay")));

                  }

                  deviceChannel.setCertNum(XmlUtil.getText(itemDevice, "CertNum"));

                  if (XmlUtil.getText(itemDevice, "Certifiable") == null || XmlUtil.getText(itemDevice, "Certifiable") == "") {

                     deviceChannel.setCertifiable(0);

                  } else {

                     deviceChannel.setCertifiable(Integer.parseInt(XmlUtil.getText(itemDevice, "Certifiable")));

                  }

                  if (XmlUtil.getText(itemDevice, "ErrCode") == null || XmlUtil.getText(itemDevice, "ErrCode") == "") {

                     deviceChannel.setErrCode(0);

                  } else {

                     deviceChannel.setErrCode(Integer.parseInt(XmlUtil.getText(itemDevice, "ErrCode")));

                  }

                  deviceChannel.setEndTime(XmlUtil.getText(itemDevice, "EndTime"));

                  deviceChannel.setSecrecy(XmlUtil.getText(itemDevice, "Secrecy"));

                  deviceChannel.setIpAddress(XmlUtil.getText(itemDevice, "IPAddress"));

                  if (XmlUtil.getText(itemDevice, "Port") == null || XmlUtil.getText(itemDevice, "Port") =="") {

                     deviceChannel.setPort(0);

                  } else {

                     deviceChannel.setPort(Integer.parseInt(XmlUtil.getText(itemDevice, "Port")));

                  }

                  deviceChannel.setPassword(XmlUtil.getText(itemDevice, "Password"));

                  if (XmlUtil.getText(itemDevice, "Longitude") == null || XmlUtil.getText(itemDevice, "Longitude") == "") {

                     deviceChannel.setLongitude(0.00);

                  } else {

                     deviceChannel.setLongitude(Double.parseDouble(XmlUtil.getText(itemDevice, "Longitude")));

                  }

                  if (XmlUtil.getText(itemDevice, "Latitude") == null || XmlUtil.getText(itemDevice, "Latitude") =="") {

                     deviceChannel.setLatitude(0.00);

                  } else {

                     deviceChannel.setLatitude(Double.parseDouble(XmlUtil.getText(itemDevice, "Latitude")));

                  }

                  if (XmlUtil.getText(itemDevice, "PTZType") == null || XmlUtil.getText(itemDevice, "PTZType") == "") {

                     deviceChannel.setPTZType(0);

                  } else {

                     deviceChannel.setPTZType(Integer.parseInt(XmlUtil.getText(itemDevice, "PTZType")));

                  }

                  deviceChannel.setHasAudio(true); // 默认含有音频，播放时再检查是否有音频及是否AAC

                  storager.updateChannel(device.getDeviceId(), deviceChannel);

               }



               RequestMessage msg = new RequestMessage();

               msg.setDeviceId(deviceId);

               msg.setType(DeferredResultHolder.CALLBACK_CMD_CATALOG);

               msg.setData(device);

               deferredResultHolder.invokeResult(msg);

               // 回复200 OK

               responseAck(evt);

               if (offLineDetector.isOnline(deviceId)) {

                  publisher.onlineEventPublish(deviceId, VideoManagerConstants.EVENT_ONLINE_KEEPLIVE);

               }

            }

         }



      } catch (DocumentException | SipException | InvalidArgumentException | ParseException e) {

         e.printStackTrace();

      }

@@ -289,19 +324,41 @@
            // storager.queryChannel(deviceId)

            return;

         }

         device.setName(XmlUtil.getText(rootElement, "DeviceName"));

         device.setManufacturer(XmlUtil.getText(rootElement, "Manufacturer"));

         device.setModel(XmlUtil.getText(rootElement, "Model"));

         device.setFirmware(XmlUtil.getText(rootElement, "Firmware"));

         if (StringUtils.isEmpty(device.getStreamMode())) {

            device.setStreamMode("UDP");



         DeviceAlarm deviceAlarm = new DeviceAlarm();

         deviceAlarm.setDeviceId(deviceId);

         deviceAlarm.setAlarmPriority(XmlUtil.getText(rootElement, "AlarmPriority"));

         deviceAlarm.setAlarmMethod(XmlUtil.getText(rootElement, "AlarmMethod"));

         deviceAlarm.setAlarmTime(XmlUtil.getText(rootElement, "AlarmTime"));

         if (XmlUtil.getText(rootElement, "AlarmDescription") == null) {

            deviceAlarm.setAlarmDescription("");

         } else {

            deviceAlarm.setAlarmDescription(XmlUtil.getText(rootElement, "AlarmDescription"));

         }

         storager.updateDevice(device);

         cmder.catalogQuery(device, null);

         if (XmlUtil.getText(rootElement, "Longitude") == null || XmlUtil.getText(rootElement, "Longitude") == "") {

            deviceAlarm.setLongitude(0.00);

         } else {

            deviceAlarm.setLongitude(Double.parseDouble(XmlUtil.getText(rootElement, "Longitude")));

         }

         if (XmlUtil.getText(rootElement, "Latitude") == null || XmlUtil.getText(rootElement, "Latitude") =="") {

            deviceAlarm.setLatitude(0.00);

         } else {

            deviceAlarm.setLatitude(Double.parseDouble(XmlUtil.getText(rootElement, "Latitude")));

         }



         // device.setName(XmlUtil.getText(rootElement, "DeviceName"));

         // device.setManufacturer(XmlUtil.getText(rootElement, "Manufacturer"));

         // device.setModel(XmlUtil.getText(rootElement, "Model"));

         // device.setFirmware(XmlUtil.getText(rootElement, "Firmware"));

         // if (StringUtils.isEmpty(device.getStreamMode())) {

         //    device.setStreamMode("UDP");

         // }

         // storager.updateDevice(device);

         //cmder.catalogQuery(device, null);

         // 回复200 OK

         responseAck(evt);

         if (offLineDetector.isOnline(deviceId)) {

            publisher.onlineEventPublish(deviceId, VideoManagerConstants.EVENT_ONLINE_KEEPLIVE);

            publisher.deviceAlarmEventPublish(deviceAlarm);

         }

      } catch (DocumentException | SipException | InvalidArgumentException | ParseException e) {

         // } catch (DocumentException e) {

@@ -342,6 +399,7 @@
      try {

         // 回复200 OK

         responseAck(evt);

         String uuid = UUID.randomUUID().toString().replace("-", "");

         RecordInfo recordInfo = new RecordInfo();

         Element rootElement = getRootElement(evt);

         Element deviceIdElement = rootElement.element("DeviceID");

@@ -396,31 +454,21 @@
            if (recordInfo.getSumNum() > 0 && recordList.size() > 0 && recordList.size() < recordInfo.getSumNum()) {

               // 为防止连续请求该设备的录像数据，返回数据错乱，特增加sn进行区分

               String cacheKey = CACHE_RECORDINFO_KEY + deviceId + sn;

               // TODO 暂时直接操作redis存储，后续封装专用缓存接口，改为本地内存缓存

               if (redis.hasKey(cacheKey)) {

                  List<RecordItem> previousList = (List<RecordItem>) redis.get(cacheKey);

                  if (previousList != null && previousList.size() > 0) {

                     recordList.addAll(previousList);

                  }

                  // 本分支表示录像列表被拆包，且加上之前的数据还是不够,保存缓存返回，等待下个包再处理

                  if (recordList.size() < recordInfo.getSumNum()) {

                     logger.info("已获取" + recordList.size() + "项录像数据，共" + recordInfo.getSumNum() + "项");

                     redis.set(cacheKey, recordList, 90);

                     return;

                  } else {

                     // 本分支表示录像被拆包，但加上之前的数据够足够，返回响应

                     // 因设备心跳有监听redis过期机制，为提高性能，此处手动删除

                     logger.info("录像数据已全部获取");

                     redis.del(cacheKey);

                  }

               } else {

                  // 本分支有两种可能：1、录像列表被拆包，且是第一个包,直接保存缓存返回，等待下个包再处理

                  // 2、之前有包，但超时清空了，那么这次sn批次的响应数据已经不完整，等待过期时间后redis自动清空数据

                  logger.info("已获取" + recordList.size() + "项录像数据，共" + recordInfo.getSumNum() + "项");

                  logger.info("等待后续的包...");



                  redis.set(cacheKey, recordList, 90);

               redis.set(cacheKey + "_" + uuid, recordList, 90);

               List<Object> cacheKeys = redis.scan(cacheKey + "_*");

               List<RecordItem> totalRecordList = new ArrayList<RecordItem>();

               for (int i = 0; i < cacheKeys.size(); i++) {

                  totalRecordList.addAll((List<RecordItem>) redis.get(cacheKeys.get(i).toString()));

               }

               if (totalRecordList.size() < recordInfo.getSumNum()) {

                  logger.info("已获取" + totalRecordList.size() + "项录像数据，共" + recordInfo.getSumNum() + "项");

                  return;

               }

               logger.info("录像数据已全部获取，共" + recordInfo.getSumNum() + "项");

               recordInfo.setRecordList(totalRecordList);

               for (int i = 0; i < cacheKeys.size(); i++) {

                  redis.del(cacheKeys.get(i).toString());

               }

            }

            // 自然顺序排序, 元素进行升序排列

@@ -460,7 +508,6 @@
               cmder.streamByeCmd(streamInfo.getStreamId());

            }

         }



      } catch (ParseException | SipException | InvalidArgumentException | DocumentException e) {

         e.printStackTrace();

      }

@@ -476,6 +523,18 @@
    */

   private void responseAck(RequestEvent evt) throws SipException, InvalidArgumentException, ParseException {

      Response response = getMessageFactory().createResponse(Response.OK, evt.getRequest());

      getServerTransaction(evt).sendResponse(response);

   }



   /***

    * 回复404

    * @param evt

    * @throws SipException

    * @throws InvalidArgumentException

    * @throws ParseException

    */

   private void response404Ack(RequestEvent evt) throws SipException, InvalidArgumentException, ParseException {

      Response response = getMessageFactory().createResponse(Response.NOT_FOUND, evt.getRequest());

      getServerTransaction(evt).sendResponse(response);

   }



@@ -518,4 +577,12 @@
   public void setRedisCatchStorage(IRedisCatchStorage redisCatchStorage) {

      this.redisCatchStorage = redisCatchStorage;

   }



   public SIPCommanderFroPlatform getCmderFroPlatform() {

      return cmderFroPlatform;

   }



   public void setCmderFroPlatform(SIPCommanderFroPlatform cmderFroPlatform) {

      this.cmderFroPlatform = cmderFroPlatform;

   }

}