From 1af77ab5f7c11a4b3d59c1989b51b9fca29679ce Mon Sep 17 00:00:00 2001
From: 648540858 <648540858@qq.com>
Date: 星期二, 18 十月 2022 22:18:49 +0800
Subject: [PATCH] Merge pull request #645 from IKangXu/wvp-28181-2.0

---
 src/main/java/com/genersoft/iot/vmp/gb28181/transmit/event/request/impl/message/response/cmd/CatalogResponseMessageHandler.java |  228 +++++++++++++++++++++++++++------------------------------
 1 files changed, 108 insertions(+), 120 deletions(-)

diff --git a/src/main/java/com/genersoft/iot/vmp/gb28181/transmit/event/request/impl/message/response/cmd/CatalogResponseMessageHandler.java b/src/main/java/com/genersoft/iot/vmp/gb28181/transmit/event/request/impl/message/response/cmd/CatalogResponseMessageHandler.java
index 5fcc3ad..52e0b7f 100644
--- a/src/main/java/com/genersoft/iot/vmp/gb28181/transmit/event/request/impl/message/response/cmd/CatalogResponseMessageHandler.java
+++ b/src/main/java/com/genersoft/iot/vmp/gb28181/transmit/event/request/impl/message/response/cmd/CatalogResponseMessageHandler.java
@@ -1,26 +1,30 @@
 package com.genersoft.iot.vmp.gb28181.transmit.event.request.impl.message.response.cmd;
 
-import com.genersoft.iot.vmp.common.VideoManagerConstants;
 import com.genersoft.iot.vmp.conf.SipConfig;
-import com.genersoft.iot.vmp.gb28181.bean.Device;
-import com.genersoft.iot.vmp.gb28181.bean.DeviceChannel;
-import com.genersoft.iot.vmp.gb28181.bean.ParentPlatform;
-import com.genersoft.iot.vmp.gb28181.event.DeviceOffLineDetector;
+import com.genersoft.iot.vmp.conf.UserSetting;
+import com.genersoft.iot.vmp.gb28181.bean.*;
 import com.genersoft.iot.vmp.gb28181.event.EventPublisher;
+import com.genersoft.iot.vmp.gb28181.session.CatalogDataCatch;
 import com.genersoft.iot.vmp.gb28181.transmit.callback.DeferredResultHolder;
-import com.genersoft.iot.vmp.gb28181.transmit.callback.RequestMessage;
 import com.genersoft.iot.vmp.gb28181.transmit.event.request.SIPRequestProcessorParent;
 import com.genersoft.iot.vmp.gb28181.transmit.event.request.impl.message.IMessageHandler;
 import com.genersoft.iot.vmp.gb28181.transmit.event.request.impl.message.response.ResponseMessageHandler;
+import com.genersoft.iot.vmp.gb28181.utils.Coordtransform;
 import com.genersoft.iot.vmp.gb28181.utils.NumericUtil;
-import com.genersoft.iot.vmp.storager.IVideoManagerStorager;
+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.message.SIPRequest;
 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.scheduling.concurrent.ThreadPoolTaskExecutor;
 import org.springframework.stereotype.Component;
+import org.springframework.util.StringUtils;
 
 import javax.sip.InvalidArgumentException;
 import javax.sip.RequestEvent;
@@ -30,32 +34,33 @@
 import java.util.ArrayList;
 import java.util.Iterator;
 import java.util.List;
+import java.util.concurrent.ConcurrentLinkedQueue;
 
-import static com.genersoft.iot.vmp.gb28181.utils.XmlUtil.getText;
-
+/**
+ * 鐩綍鏌ヨ鐨勫洖澶�
+ */
 @Component
 public class CatalogResponseMessageHandler extends SIPRequestProcessorParent implements InitializingBean, IMessageHandler {
 
     private Logger logger = LoggerFactory.getLogger(CatalogResponseMessageHandler.class);
     private final String cmdType = "Catalog";
 
+    private boolean taskQueueHandlerRun = false;
+
     @Autowired
     private ResponseMessageHandler responseMessageHandler;
 
-    @Autowired
-    private IVideoManagerStorager storager;
+    private ConcurrentLinkedQueue<HandlerCatchData> taskQueue = new ConcurrentLinkedQueue<>();
 
     @Autowired
-    private DeferredResultHolder deferredResultHolder;
+    private IVideoManagerStorage storager;
 
     @Autowired
-    private DeviceOffLineDetector offLineDetector;
+    private CatalogDataCatch catalogDataCatch;
 
+    @Qualifier("taskExecutor")
     @Autowired
-    private SipConfig config;
-
-    @Autowired
-    private EventPublisher publisher;
+    private ThreadPoolTaskExecutor taskExecutor;
 
     @Override
     public void afterPropertiesSet() throws Exception {
@@ -64,122 +69,105 @@
 
     @Override
     public void handForDevice(RequestEvent evt, Device device, Element element) {
-        String key = DeferredResultHolder.CALLBACK_CMD_CATALOG + device.getDeviceId();
-        Element rootElement = null;
+        taskQueue.offer(new HandlerCatchData(evt, device, element));
+        // 鍥炲200 OK
         try {
-            rootElement = getRootElement(evt, device.getCharset());
-            Element deviceListElement = rootElement.element("DeviceList");
-            Iterator<Element> deviceListIterator = deviceListElement.elementIterator();
-            if (deviceListIterator != null) {
-                List<DeviceChannel> channelList = new ArrayList<>();
-                // 閬嶅巻DeviceList
-                while (deviceListIterator.hasNext()) {
-                    Element itemDevice = deviceListIterator.next();
-                    Element channelDeviceElement = itemDevice.element("DeviceID");
-                    if (channelDeviceElement == null) {
+            responseAck((SIPRequest) evt.getRequest(), Response.OK);
+        } catch (SipException | InvalidArgumentException | ParseException e) {
+            logger.error("[鍛戒护鍙戦�佸け璐 鐩綍鏌ヨ鍥炲: {}", e.getMessage());
+        }
+        if (!taskQueueHandlerRun) {
+            taskQueueHandlerRun = true;
+            taskExecutor.execute(() -> {
+                while (!taskQueue.isEmpty()) {
+                    HandlerCatchData take = taskQueue.poll();
+                    Element rootElement = null;
+                    try {
+                        rootElement = getRootElement(take.getEvt(), take.getDevice().getCharset());
+                    } catch (DocumentException e) {
+                        logger.error("[xml瑙f瀽] 澶辫触锛� ", e);
                         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 (rootElement == null) {
+                        logger.warn("[ 鏀跺埌閫氶亾 ] content cannot be null, {}", evt.getRequest());
+                        continue;
                     }
-                    if (status.equals("OFF") || status.equals("Off") || status.equals("OFFLINE")) {
-                        deviceChannel.setStatus(0);
-                    }
+                    Element deviceListElement = rootElement.element("DeviceList");
+                    Element sumNumElement = rootElement.element("SumNum");
+                    Element snElement = rootElement.element("SN");
+                    int sumNum = Integer.parseInt(sumNumElement.getText());
 
-                    deviceChannel.setManufacture(getText(itemDevice, "Manufacturer"));
-                    deviceChannel.setModel(getText(itemDevice, "Model"));
-                    deviceChannel.setOwner(getText(itemDevice, "Owner"));
-                    deviceChannel.setCivilCode(getText(itemDevice, "CivilCode"));
-                    deviceChannel.setBlock(getText(itemDevice, "Block"));
-                    deviceChannel.setAddress(getText(itemDevice, "Address"));
-                    if (getText(itemDevice, "Parental") == null || getText(itemDevice, "Parental") == "") {
-                        deviceChannel.setParental(0);
+                    if (sumNum == 0) {
+                        logger.info("[鏀跺埌閫氶亾]璁惧:{}鐨�: 0涓�", take.getDevice().getDeviceId());
+                        // 鏁版嵁宸茬粡瀹屾暣鎺ユ敹
+                        storager.cleanChannelsForDevice(take.getDevice().getDeviceId());
+                        catalogDataCatch.setChannelSyncEnd(take.getDevice().getDeviceId(), null);
                     } else {
-                        deviceChannel.setParental(Integer.parseInt(getText(itemDevice, "Parental")));
+                        Iterator<Element> deviceListIterator = deviceListElement.elementIterator();
+                        if (deviceListIterator != null) {
+                            List<DeviceChannel> channelList = new ArrayList<>();
+                            // 閬嶅巻DeviceList
+                            while (deviceListIterator.hasNext()) {
+                                Element itemDevice = deviceListIterator.next();
+                                Element channelDeviceElement = itemDevice.element("DeviceID");
+                                if (channelDeviceElement == null) {
+                                    continue;
+                                }
+                                DeviceChannel deviceChannel = XmlUtil.channelContentHander(itemDevice, device, null);
+                                deviceChannel.setDeviceId(take.getDevice().getDeviceId());
+
+                                channelList.add(deviceChannel);
+                            }
+                            int sn = Integer.parseInt(snElement.getText());
+                            catalogDataCatch.put(take.getDevice().getDeviceId(), sn, sumNum, take.getDevice(), channelList);
+                            logger.info("[鏀跺埌閫氶亾]璁惧: {} -> {}涓紝{}/{}", take.getDevice().getDeviceId(), channelList.size(), catalogDataCatch.get(take.getDevice().getDeviceId()) == null ? 0 : catalogDataCatch.get(take.getDevice().getDeviceId()).size(), sumNum);
+                            if (catalogDataCatch.get(take.getDevice().getDeviceId()).size() == sumNum) {
+                                // 鏁版嵁宸茬粡瀹屾暣鎺ユ敹锛� 姝ゆ椂鍙兘瀛樺湪鏌愪釜璁惧绂荤嚎鍙樹笂绾跨殑鎯呭喌锛屼絾鏄�冭檻鍒版�ц兘锛屾澶勪笉鍋氬鐞嗭紝
+                                // 鐩墠鏀寔璁惧閫氶亾涓婄嚎閫氱煡鏃跺拰璁惧涓婄嚎鏃跺悜涓婄骇閫氱煡
+                                boolean resetChannelsResult = storager.resetChannels(take.getDevice().getDeviceId(), catalogDataCatch.get(take.getDevice().getDeviceId()));
+                                if (!resetChannelsResult) {
+                                    String errorMsg = "鎺ユ敹鎴愬姛锛屽啓鍏ュけ璐ワ紝鍏�" + sumNum + "鏉★紝宸叉帴鏀�" + catalogDataCatch.get(take.getDevice().getDeviceId()).size() + "鏉�";
+                                    catalogDataCatch.setChannelSyncEnd(take.getDevice().getDeviceId(), errorMsg);
+                                } else {
+                                    catalogDataCatch.setChannelSyncEnd(take.getDevice().getDeviceId(), null);
+                                }
+                            }
+                        }
+
                     }
-                    deviceChannel.setParentId(getText(itemDevice, "ParentID"));
-                    if (getText(itemDevice, "SafetyWay") == null || getText(itemDevice, "SafetyWay") == "") {
-                        deviceChannel.setSafetyWay(0);
-                    } else {
-                        deviceChannel.setSafetyWay(Integer.parseInt(getText(itemDevice, "SafetyWay")));
-                    }
-                    if (getText(itemDevice, "RegisterWay") == null || getText(itemDevice, "RegisterWay") == "") {
-                        deviceChannel.setRegisterWay(1);
-                    } else {
-                        deviceChannel.setRegisterWay(Integer.parseInt(getText(itemDevice, "RegisterWay")));
-                    }
-                    deviceChannel.setCertNum(getText(itemDevice, "CertNum"));
-                    if (getText(itemDevice, "Certifiable") == null || getText(itemDevice, "Certifiable") == "") {
-                        deviceChannel.setCertifiable(0);
-                    } else {
-                        deviceChannel.setCertifiable(Integer.parseInt(getText(itemDevice, "Certifiable")));
-                    }
-                    if (getText(itemDevice, "ErrCode") == null || getText(itemDevice, "ErrCode") == "") {
-                        deviceChannel.setErrCode(0);
-                    } else {
-                        deviceChannel.setErrCode(Integer.parseInt(getText(itemDevice, "ErrCode")));
-                    }
-                    deviceChannel.setEndTime(getText(itemDevice, "EndTime"));
-                    deviceChannel.setSecrecy(getText(itemDevice, "Secrecy"));
-                    deviceChannel.setIpAddress(getText(itemDevice, "IPAddress"));
-                    if (getText(itemDevice, "Port") == null || getText(itemDevice, "Port") == "") {
-                        deviceChannel.setPort(0);
-                    } else {
-                        deviceChannel.setPort(Integer.parseInt(getText(itemDevice, "Port")));
-                    }
-                    deviceChannel.setPassword(getText(itemDevice, "Password"));
-                    if (NumericUtil.isDouble(getText(itemDevice, "Longitude"))) {
-                        deviceChannel.setLongitude(Double.parseDouble(getText(itemDevice, "Longitude")));
-                    } else {
-                        deviceChannel.setLongitude(0.00);
-                    }
-                    if (NumericUtil.isDouble(getText(itemDevice, "Latitude"))) {
-                        deviceChannel.setLatitude(Double.parseDouble(getText(itemDevice, "Latitude")));
-                    } else {
-                        deviceChannel.setLatitude(0.00);
-                    }
-                    if (getText(itemDevice, "PTZType") == null || getText(itemDevice, "PTZType") == "") {
-                        deviceChannel.setPTZType(0);
-                    } else {
-                        deviceChannel.setPTZType(Integer.parseInt(getText(itemDevice, "PTZType")));
-                    }
-                    deviceChannel.setHasAudio(true); // 榛樿鍚湁闊抽锛屾挱鏀炬椂鍐嶆鏌ユ槸鍚︽湁闊抽鍙婃槸鍚AC
-                    // TODO 淇敼涓烘壒閲忔彃鍏�
-                    channelList.add(deviceChannel);
                 }
-                storager.updateChannels(device.getDeviceId(), channelList);
-                RequestMessage msg = new RequestMessage();
-                msg.setKey(key);
-                msg.setData(device);
-                deferredResultHolder.invokeAllResult(msg);
-                // 鍥炲200 OK
-                responseAck(evt, Response.OK);
-                if (offLineDetector.isOnline(device.getDeviceId())) {
-                    publisher.onlineEventPublish(device, VideoManagerConstants.EVENT_ONLINE_MESSAGE);
-                }
-            }
-        } catch (DocumentException e) {
-            e.printStackTrace();
-        } catch (InvalidArgumentException e) {
-            e.printStackTrace();
-        } catch (ParseException e) {
-            e.printStackTrace();
-        } catch (SipException e) {
-            e.printStackTrace();
+                taskQueueHandlerRun = false;
+            });
         }
+
     }
 
     @Override
     public void handForPlatform(RequestEvent evt, ParentPlatform parentPlatform, Element rootElement) {
 
     }
+
+    public SyncStatus getChannelSyncProgress(String deviceId) {
+        if (catalogDataCatch.get(deviceId) == null) {
+            return null;
+        } else {
+            return catalogDataCatch.getSyncStatus(deviceId);
+        }
+    }
+
+    public boolean isSyncRunning(String deviceId) {
+        if (catalogDataCatch.get(deviceId) == null) {
+            return false;
+        } else {
+            return catalogDataCatch.isSyncRunning(deviceId);
+        }
+    }
+
+    public void setChannelSyncReady(Device device, int sn) {
+        catalogDataCatch.addReady(device, sn);
+    }
+
+    public void setChannelSyncEnd(String deviceId, String errorMsg) {
+        catalogDataCatch.setChannelSyncEnd(deviceId, errorMsg);
+    }
 }

--
Gitblit v1.8.0