进一步优化录像列表获取算法

Lawrence

2021-01-12 dd0a234c051c451143137ebb5bfbd2d789a6d2f5

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

@@ -10,6 +10,7 @@
import javax.sip.message.Request;

import javax.sip.message.Response;



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

import org.dom4j.Document;

import org.dom4j.DocumentException;

import org.dom4j.Element;

@@ -34,7 +35,7 @@
import com.genersoft.iot.vmp.storager.IVideoManagerStorager;

import com.genersoft.iot.vmp.utils.redis.RedisUtil;

import org.springframework.util.StringUtils;



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

/**

 * @Description:MESSAGE请求处理器

 * @author: swwheihei

@@ -47,6 +48,8 @@
   private SIPCommander cmder;



   private IVideoManagerStorager storager;



   private IRedisCatchStorage redisCatchStorage;



   private EventPublisher publisher;



@@ -64,6 +67,7 @@
   private static final String MESSAGE_DEVICE_INFO = "DeviceInfo";

   private static final String MESSAGE_ALARM = "Alarm";

   private static final String MESSAGE_RECORD_INFO = "RecordInfo";

   private static final String MESSAGE_MEDIA_STATUS = "MediaStatus";

   // private static final String MESSAGE_BROADCAST = "Broadcast";

   // private static final String MESSAGE_DEVICE_STATUS = "DeviceStatus";

   // private static final String MESSAGE_MOBILE_POSITION = "MobilePosition";

@@ -98,6 +102,11 @@
         } else if (MESSAGE_RECORD_INFO.equals(cmd)) {

            logger.info("接收到RecordInfo消息");

            processMessageRecordInfo(evt);

         }else if (MESSAGE_MEDIA_STATUS.equals(cmd)) {

            logger.info("接收到MediaStatus消息");

            processMessageMediaStatus(evt);

         } else {

            logger.info("接收到消息：" + cmd);

         }

      } catch (DocumentException e) {

         e.printStackTrace();

@@ -178,10 +187,11 @@
               DeviceChannel deviceChannel = new DeviceChannel();

               deviceChannel.setName(channelName);

               deviceChannel.setChannelId(channelDeviceId);

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

               // 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")) {

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

                  deviceChannel.setStatus(0);

               }



@@ -191,30 +201,57 @@
               deviceChannel.setCivilCode(XmlUtil.getText(itemDevice, "CivilCode"));

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

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

               deviceChannel.setParental(itemDevice.element("Parental") == null ? 0

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

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

               deviceChannel.setSafetyWay(itemDevice.element("SafetyWay") == null ? 0

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

               deviceChannel.setRegisterWay(itemDevice.element("RegisterWay") == null ? 1

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

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

               deviceChannel.setCertifiable(itemDevice.element("Certifiable") == null ? 0

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

               deviceChannel.setErrCode(itemDevice.element("ErrCode") == null ? 0

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

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

               deviceChannel.setPort(itemDevice.element("Port") == null ? 0

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

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

               deviceChannel.setLongitude(itemDevice.element("Longitude") == null ? 0.00

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

               deviceChannel.setLatitude(itemDevice.element("Latitude") == null ? 0.00

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

               deviceChannel.setPTZType(itemDevice.element("PTZType") == null ? 0

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

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

            }

@@ -260,7 +297,7 @@
            device.setStreamMode("UDP");

         }

         storager.updateDevice(device);

         cmder.catalogQuery(device);

         //cmder.catalogQuery(device, null);

         // 回复200 OK

         responseAck(evt);

         if (offLineDetector.isOnline(deviceId)) {

@@ -281,12 +318,16 @@
      try {

         Element rootElement = getRootElement(evt);

         String deviceId = XmlUtil.getText(rootElement, "DeviceID");

         // 回复200 OK

         responseAck(evt);

         if (offLineDetector.isOnline(deviceId)) {

            publisher.onlineEventPublish(deviceId, VideoManagerConstants.EVENT_ONLINE_KEEPLIVE);

         } else {

         // 检查设备是否存在， 不存在则不回复

         if (storager.exists(deviceId)) {

            // 回复200 OK

            responseAck(evt);

            if (offLineDetector.isOnline(deviceId)) {

               publisher.onlineEventPublish(deviceId, VideoManagerConstants.EVENT_ONLINE_KEEPLIVE);

            } else {

            }

         }



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

         e.printStackTrace();

      }

@@ -301,13 +342,18 @@
      try {

         // 回复200 OK

         responseAck(evt);

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

         RecordInfo recordInfo = new RecordInfo();

         Element rootElement = getRootElement(evt);

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

         String deviceId = deviceIdElement.getText().toString();

         recordInfo.setDeviceId(deviceId);

         recordInfo.setName(XmlUtil.getText(rootElement, "Name"));

         recordInfo.setSumNum(Integer.parseInt(XmlUtil.getText(rootElement, "SumNum")));

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

            recordInfo.setSumNum(0);

         } else {

            recordInfo.setSumNum(Integer.parseInt(XmlUtil.getText(rootElement, "SumNum")));

         }

         String sn = XmlUtil.getText(rootElement, "SN");

         Element recordListElement = rootElement.element("RecordList");

         if (recordListElement == null || recordInfo.getSumNum() == 0) {

@@ -351,31 +397,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());

               }

            }

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

@@ -399,6 +435,35 @@
      }

   }





   private void processMessageMediaStatus(RequestEvent evt){

      try {

         // 回复200 OK

         responseAck(evt);

         Element rootElement = getRootElement(evt);

         String deviceId = XmlUtil.getText(rootElement, "DeviceID");

         String NotifyType =XmlUtil.getText(rootElement, "NotifyType");

         if (NotifyType.equals("121")){

            logger.info("媒体播放完毕，通知关流");

            StreamInfo streamInfo = redisCatchStorage.queryPlaybackByDevice(deviceId, "*");

            if (streamInfo != null) {

               redisCatchStorage.stopPlayback(streamInfo);

               cmder.streamByeCmd(streamInfo.getStreamId());

            }

         }

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

         e.printStackTrace();

      }

   }





   /***

    * 回复200 OK

    * @param evt

    * @throws SipException

    * @throws InvalidArgumentException

    * @throws ParseException

    */

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

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

      getServerTransaction(evt).sendResponse(response);

@@ -436,4 +501,11 @@
      this.offLineDetector = offLineDetector;

   }



   public IRedisCatchStorage getRedisCatchStorage() {

      return redisCatchStorage;

   }



   public void setRedisCatchStorage(IRedisCatchStorage redisCatchStorage) {

      this.redisCatchStorage = redisCatchStorage;

   }

}