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

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;

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



   private IVideoManagerStorager storager;



   private IRedisCatchStorage redisCatchStorage;



   private EventPublisher publisher;



@@ -184,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);

               }



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

         }

         storager.updateDevice(device);

         cmder.catalogQuery(device);

         //cmder.catalogQuery(device, null);

         // 回复200 OK

         responseAck(evt);

         if (offLineDetector.isOnline(deviceId)) {

@@ -314,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();

      }

@@ -334,6 +342,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");

@@ -388,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());

               }

            }

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

@@ -446,10 +445,10 @@
         String NotifyType =XmlUtil.getText(rootElement, "NotifyType");

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

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

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

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

            if (streamInfo != null) {

               storager.stopPlayback(streamInfo);

               cmder.streamByeCmd(streamInfo.getSsrc());

               redisCatchStorage.stopPlayback(streamInfo);

               cmder.streamByeCmd(streamInfo.getStreamId());

            }

         }

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

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

   }



   public IRedisCatchStorage getRedisCatchStorage() {

      return redisCatchStorage;

   }



   public void setRedisCatchStorage(IRedisCatchStorage redisCatchStorage) {

      this.redisCatchStorage = redisCatchStorage;

   }

}