|  | @@ -1,9 +1,6 @@
 | 
	
		
			
				|  |  |  package com.genersoft.iot.vmp.gb28181.transmit.event.request.impl.message.response.cmd;
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | -import com.genersoft.iot.vmp.gb28181.bean.Device;
 | 
	
		
			
				|  |  | -import com.genersoft.iot.vmp.gb28181.bean.ParentPlatform;
 | 
	
		
			
				|  |  | -import com.genersoft.iot.vmp.gb28181.bean.RecordInfo;
 | 
	
		
			
				|  |  | -import com.genersoft.iot.vmp.gb28181.bean.RecordItem;
 | 
	
		
			
				|  |  | +import com.genersoft.iot.vmp.gb28181.bean.*;
 | 
	
		
			
				|  |  |  import com.genersoft.iot.vmp.gb28181.event.EventPublisher;
 | 
	
		
			
				|  |  |  import com.genersoft.iot.vmp.gb28181.session.RecordDataCatch;
 | 
	
		
			
				|  |  |  import com.genersoft.iot.vmp.gb28181.transmit.callback.DeferredResultHolder;
 | 
	
	
		
			
				|  | @@ -19,6 +16,8 @@ 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.scheduling.concurrent.ThreadPoolTaskExecutor;
 | 
	
		
			
				|  |  |  import org.springframework.stereotype.Component;
 | 
	
		
			
				|  |  |  import org.springframework.util.StringUtils;
 | 
	
		
			
				|  |  |  
 | 
	
	
		
			
				|  | @@ -28,6 +27,9 @@ import javax.sip.SipException;
 | 
	
		
			
				|  |  |  import javax.sip.message.Response;
 | 
	
		
			
				|  |  |  import java.text.ParseException;
 | 
	
		
			
				|  |  |  import java.util.*;
 | 
	
		
			
				|  |  | +import java.util.concurrent.BlockingQueue;
 | 
	
		
			
				|  |  | +import java.util.concurrent.ConcurrentLinkedQueue;
 | 
	
		
			
				|  |  | +import java.util.concurrent.LinkedBlockingQueue;
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  import static com.genersoft.iot.vmp.gb28181.utils.XmlUtil.getText;
 | 
	
		
			
				|  |  |  
 | 
	
	
		
			
				|  | @@ -42,6 +44,9 @@ public class RecordInfoResponseMessageHandler extends SIPRequestProcessorParent
 | 
	
		
			
				|  |  |      private final String cmdType = "RecordInfo";
 | 
	
		
			
				|  |  |      private final static String CACHE_RECORDINFO_KEY = "CACHE_RECORDINFO_";
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | +    private ConcurrentLinkedQueue<HandlerCatchData> taskQueue = new ConcurrentLinkedQueue<>();
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +    private boolean taskQueueHandlerRun = false;
 | 
	
		
			
				|  |  |      @Autowired
 | 
	
		
			
				|  |  |      private ResponseMessageHandler responseMessageHandler;
 | 
	
		
			
				|  |  |  
 | 
	
	
		
			
				|  | @@ -51,11 +56,13 @@ public class RecordInfoResponseMessageHandler extends SIPRequestProcessorParent
 | 
	
		
			
				|  |  |      @Autowired
 | 
	
		
			
				|  |  |      private DeferredResultHolder deferredResultHolder;
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  |      @Autowired
 | 
	
		
			
				|  |  |      private EventPublisher eventPublisher;
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | +    @Qualifier("taskExecutor")
 | 
	
		
			
				|  |  | +    @Autowired
 | 
	
		
			
				|  |  | +    private ThreadPoolTaskExecutor taskExecutor;
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  |      @Override
 | 
	
		
			
				|  |  |      public void afterPropertiesSet() throws Exception {
 | 
	
		
			
				|  |  |          responseMessageHandler.addHandler(cmdType, this);
 | 
	
	
		
			
				|  | @@ -67,75 +74,89 @@ public class RecordInfoResponseMessageHandler extends SIPRequestProcessorParent
 | 
	
		
			
				|  |  |          // 回复200 OK
 | 
	
		
			
				|  |  |          try {
 | 
	
		
			
				|  |  |              responseAck(evt, Response.OK);
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -            rootElement = getRootElement(evt, device.getCharset());
 | 
	
		
			
				|  |  | -            String sn = getText(rootElement, "SN");
 | 
	
		
			
				|  |  | -            RecordInfo recordInfo = new RecordInfo();
 | 
	
		
			
				|  |  | -            recordInfo.setDeviceId(device.getDeviceId());
 | 
	
		
			
				|  |  | -            recordInfo.setSn(sn);
 | 
	
		
			
				|  |  | -            recordInfo.setName(getText(rootElement, "Name"));
 | 
	
		
			
				|  |  | -            String sumNumStr = getText(rootElement, "SumNum");
 | 
	
		
			
				|  |  | -            int sumNum = 0;
 | 
	
		
			
				|  |  | -            if (!StringUtils.isEmpty(sumNumStr)) {
 | 
	
		
			
				|  |  | -                sumNum = Integer.parseInt(sumNumStr);
 | 
	
		
			
				|  |  | -            }
 | 
	
		
			
				|  |  | -            recordInfo.setSumNum(sumNum);
 | 
	
		
			
				|  |  | -            Element recordListElement = rootElement.element("RecordList");
 | 
	
		
			
				|  |  | -            if (recordListElement == null || sumNum == 0) {
 | 
	
		
			
				|  |  | -                logger.info("无录像数据");
 | 
	
		
			
				|  |  | -                eventPublisher.recordEndEventPush(recordInfo);
 | 
	
		
			
				|  |  | -                recordDataCatch.put(device.getDeviceId(), sn, sumNum, new ArrayList<>());
 | 
	
		
			
				|  |  | -                releaseRequest(device.getDeviceId(), sn);
 | 
	
		
			
				|  |  | -            } else {
 | 
	
		
			
				|  |  | -                Iterator<Element> recordListIterator = recordListElement.elementIterator();
 | 
	
		
			
				|  |  | -                if (recordListIterator != null) {
 | 
	
		
			
				|  |  | -                    List<RecordItem> recordList = new ArrayList<>();
 | 
	
		
			
				|  |  | -                    // 遍历DeviceList
 | 
	
		
			
				|  |  | -                    while (recordListIterator.hasNext()) {
 | 
	
		
			
				|  |  | -                        Element itemRecord = recordListIterator.next();
 | 
	
		
			
				|  |  | -                        Element recordElement = itemRecord.element("DeviceID");
 | 
	
		
			
				|  |  | -                        if (recordElement == null) {
 | 
	
		
			
				|  |  | -                            logger.info("记录为空,下一个...");
 | 
	
		
			
				|  |  | -                            continue;
 | 
	
		
			
				|  |  | +            taskQueue.offer(new HandlerCatchData(evt, device, rootElement));
 | 
	
		
			
				|  |  | +            if (!taskQueueHandlerRun) {
 | 
	
		
			
				|  |  | +                taskQueueHandlerRun = true;
 | 
	
		
			
				|  |  | +                taskExecutor.execute(()->{
 | 
	
		
			
				|  |  | +                    try {
 | 
	
		
			
				|  |  | +                        while (!taskQueue.isEmpty()) {
 | 
	
		
			
				|  |  | +                            HandlerCatchData take = taskQueue.poll();
 | 
	
		
			
				|  |  | +                            Element rootElementForCharset = getRootElement(take.getEvt(), take.getDevice().getCharset());
 | 
	
		
			
				|  |  | +                            String sn = getText(rootElementForCharset, "SN");
 | 
	
		
			
				|  |  | +                            String channelId = getText(rootElementForCharset, "DeviceID");
 | 
	
		
			
				|  |  | +                            RecordInfo recordInfo = new RecordInfo();
 | 
	
		
			
				|  |  | +                            recordInfo.setChannelId(channelId);
 | 
	
		
			
				|  |  | +                            recordInfo.setDeviceId(take.getDevice().getDeviceId());
 | 
	
		
			
				|  |  | +                            recordInfo.setSn(sn);
 | 
	
		
			
				|  |  | +                            recordInfo.setName(getText(rootElementForCharset, "Name"));
 | 
	
		
			
				|  |  | +                            String sumNumStr = getText(rootElementForCharset, "SumNum");
 | 
	
		
			
				|  |  | +                            int sumNum = 0;
 | 
	
		
			
				|  |  | +                            if (!StringUtils.isEmpty(sumNumStr)) {
 | 
	
		
			
				|  |  | +                                sumNum = Integer.parseInt(sumNumStr);
 | 
	
		
			
				|  |  | +                            }
 | 
	
		
			
				|  |  | +                            recordInfo.setSumNum(sumNum);
 | 
	
		
			
				|  |  | +                            Element recordListElement = rootElementForCharset.element("RecordList");
 | 
	
		
			
				|  |  | +                            if (recordListElement == null || sumNum == 0) {
 | 
	
		
			
				|  |  | +                                logger.info("无录像数据");
 | 
	
		
			
				|  |  | +                                eventPublisher.recordEndEventPush(recordInfo);
 | 
	
		
			
				|  |  | +                                recordDataCatch.put(take.getDevice().getDeviceId(), sn, sumNum, new ArrayList<>());
 | 
	
		
			
				|  |  | +                                releaseRequest(take.getDevice().getDeviceId(), sn);
 | 
	
		
			
				|  |  | +                            } else {
 | 
	
		
			
				|  |  | +                                Iterator<Element> recordListIterator = recordListElement.elementIterator();
 | 
	
		
			
				|  |  | +                                if (recordListIterator != null) {
 | 
	
		
			
				|  |  | +                                    List<RecordItem> recordList = new ArrayList<>();
 | 
	
		
			
				|  |  | +                                    // 遍历DeviceList
 | 
	
		
			
				|  |  | +                                    while (recordListIterator.hasNext()) {
 | 
	
		
			
				|  |  | +                                        Element itemRecord = recordListIterator.next();
 | 
	
		
			
				|  |  | +                                        Element recordElement = itemRecord.element("DeviceID");
 | 
	
		
			
				|  |  | +                                        if (recordElement == null) {
 | 
	
		
			
				|  |  | +                                            logger.info("记录为空,下一个...");
 | 
	
		
			
				|  |  | +                                            continue;
 | 
	
		
			
				|  |  | +                                        }
 | 
	
		
			
				|  |  | +                                        RecordItem record = new RecordItem();
 | 
	
		
			
				|  |  | +                                        record.setDeviceId(getText(itemRecord, "DeviceID"));
 | 
	
		
			
				|  |  | +                                        record.setName(getText(itemRecord, "Name"));
 | 
	
		
			
				|  |  | +                                        record.setFilePath(getText(itemRecord, "FilePath"));
 | 
	
		
			
				|  |  | +                                        record.setFileSize(getText(itemRecord, "FileSize"));
 | 
	
		
			
				|  |  | +                                        record.setAddress(getText(itemRecord, "Address"));
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +                                        String startTimeStr = getText(itemRecord, "StartTime");
 | 
	
		
			
				|  |  | +                                        record.setStartTime(DateUtil.ISO8601Toyyyy_MM_dd_HH_mm_ss(startTimeStr));
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +                                        String endTimeStr = getText(itemRecord, "EndTime");
 | 
	
		
			
				|  |  | +                                        record.setEndTime(DateUtil.ISO8601Toyyyy_MM_dd_HH_mm_ss(endTimeStr));
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +                                        record.setSecrecy(itemRecord.element("Secrecy") == null ? 0
 | 
	
		
			
				|  |  | +                                                : Integer.parseInt(getText(itemRecord, "Secrecy")));
 | 
	
		
			
				|  |  | +                                        record.setType(getText(itemRecord, "Type"));
 | 
	
		
			
				|  |  | +                                        record.setRecorderId(getText(itemRecord, "RecorderID"));
 | 
	
		
			
				|  |  | +                                        recordList.add(record);
 | 
	
		
			
				|  |  | +                                    }
 | 
	
		
			
				|  |  | +                                    recordInfo.setRecordList(recordList);
 | 
	
		
			
				|  |  | +                                    // 发送消息,如果是上级查询此录像,则会通过这里通知给上级
 | 
	
		
			
				|  |  | +                                    eventPublisher.recordEndEventPush(recordInfo);
 | 
	
		
			
				|  |  | +                                    int count = recordDataCatch.put(take.getDevice().getDeviceId(), sn, sumNum, recordList);
 | 
	
		
			
				|  |  | +                                    logger.info("[国标录像], {}->{}: {}/{}", take.getDevice().getDeviceId(), sn, count, sumNum);
 | 
	
		
			
				|  |  | +                                }
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +                                if (recordDataCatch.isComplete(take.getDevice().getDeviceId(), sn)){
 | 
	
		
			
				|  |  | +                                    releaseRequest(take.getDevice().getDeviceId(), sn);
 | 
	
		
			
				|  |  | +                                }
 | 
	
		
			
				|  |  | +                            }
 | 
	
		
			
				|  |  |                          }
 | 
	
		
			
				|  |  | -                        RecordItem record = new RecordItem();
 | 
	
		
			
				|  |  | -                        record.setDeviceId(getText(itemRecord, "DeviceID"));
 | 
	
		
			
				|  |  | -                        record.setName(getText(itemRecord, "Name"));
 | 
	
		
			
				|  |  | -                        record.setFilePath(getText(itemRecord, "FilePath"));
 | 
	
		
			
				|  |  | -                        record.setFileSize(getText(itemRecord, "FileSize"));
 | 
	
		
			
				|  |  | -                        record.setAddress(getText(itemRecord, "Address"));
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -                        String startTimeStr = getText(itemRecord, "StartTime");
 | 
	
		
			
				|  |  | -                        record.setStartTime(DateUtil.ISO8601Toyyyy_MM_dd_HH_mm_ss(startTimeStr));
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -                        String endTimeStr = getText(itemRecord, "EndTime");
 | 
	
		
			
				|  |  | -                        record.setEndTime(DateUtil.ISO8601Toyyyy_MM_dd_HH_mm_ss(endTimeStr));
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -                        record.setSecrecy(itemRecord.element("Secrecy") == null ? 0
 | 
	
		
			
				|  |  | -                                : Integer.parseInt(getText(itemRecord, "Secrecy")));
 | 
	
		
			
				|  |  | -                        record.setType(getText(itemRecord, "Type"));
 | 
	
		
			
				|  |  | -                        record.setRecorderId(getText(itemRecord, "RecorderID"));
 | 
	
		
			
				|  |  | -                        recordList.add(record);
 | 
	
		
			
				|  |  | +                        taskQueueHandlerRun = false;
 | 
	
		
			
				|  |  | +                    }catch (DocumentException e) {
 | 
	
		
			
				|  |  | +                        throw new RuntimeException(e);
 | 
	
		
			
				|  |  |                      }
 | 
	
		
			
				|  |  | -                    recordInfo.setRecordList(recordList);
 | 
	
		
			
				|  |  | -                    // 发送消息,如果是上级查询此录像,则会通过这里通知给上级
 | 
	
		
			
				|  |  | -                    eventPublisher.recordEndEventPush(recordInfo);
 | 
	
		
			
				|  |  | -                    int count = recordDataCatch.put(device.getDeviceId(), sn, sumNum, recordList);
 | 
	
		
			
				|  |  | -                    logger.info("[国标录像], {}->{}: {}/{}", device.getDeviceId(), sn, count, sumNum);
 | 
	
		
			
				|  |  | -                }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -                if (recordDataCatch.isComplete(device.getDeviceId(), sn)){
 | 
	
		
			
				|  |  | -                    releaseRequest(device.getDeviceId(), sn);
 | 
	
		
			
				|  |  | -                }
 | 
	
		
			
				|  |  | +                });
 | 
	
		
			
				|  |  |              }
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  |          } catch (SipException e) {
 | 
	
		
			
				|  |  |              e.printStackTrace();
 | 
	
		
			
				|  |  |          } catch (InvalidArgumentException e) {
 | 
	
		
			
				|  |  |              e.printStackTrace();
 | 
	
		
			
				|  |  |          } catch (ParseException e) {
 | 
	
		
			
				|  |  |              e.printStackTrace();
 | 
	
		
			
				|  |  | -        } catch (DocumentException e) {
 | 
	
		
			
				|  |  | -            e.printStackTrace();
 | 
	
		
			
				|  |  |          }
 | 
	
		
			
				|  |  |      }
 | 
	
		
			
				|  |  |  
 |