From 764d04b497356ba6bcbb75fd42b51eca750f7223 Mon Sep 17 00:00:00 2001
From: 648540858 <648540858@qq.com>
Date: 星期三, 29 五月 2024 15:02:51 +0800
Subject: [PATCH] 调整上级观看消息的发送

---
 src/main/java/com/genersoft/iot/vmp/gb28181/transmit/event/request/impl/message/response/cmd/CatalogResponseMessageHandler.java |  261 ++++++++++++++++++++++++++--------------------------
 1 files changed, 131 insertions(+), 130 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
old mode 100644
new mode 100755
index 5fcc3ad..c16d7f5
--- 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,25 +1,23 @@
 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.gb28181.event.EventPublisher;
-import com.genersoft.iot.vmp.gb28181.transmit.callback.DeferredResultHolder;
-import com.genersoft.iot.vmp.gb28181.transmit.callback.RequestMessage;
+import com.genersoft.iot.vmp.gb28181.bean.*;
+import com.genersoft.iot.vmp.gb28181.session.CatalogDataCatch;
 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.NumericUtil;
-import com.genersoft.iot.vmp.storager.IVideoManagerStorager;
+import com.genersoft.iot.vmp.gb28181.utils.SipUtils;
+import com.genersoft.iot.vmp.gb28181.utils.XmlUtil;
+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 javax.sip.InvalidArgumentException;
@@ -30,32 +28,37 @@
 import java.util.ArrayList;
 import java.util.Iterator;
 import java.util.List;
+import java.util.concurrent.ConcurrentLinkedQueue;
+import java.util.concurrent.atomic.AtomicBoolean;
 
-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";
 
     @Autowired
     private ResponseMessageHandler responseMessageHandler;
 
-    @Autowired
-    private IVideoManagerStorager storager;
+    private final ConcurrentLinkedQueue<HandlerCatchData> taskQueue = new ConcurrentLinkedQueue<>();
 
     @Autowired
-    private DeferredResultHolder deferredResultHolder;
+    private IVideoManagerStorage storager;
 
     @Autowired
-    private DeviceOffLineDetector offLineDetector;
+    private CatalogDataCatch catalogDataCatch;
+
+    @Qualifier("taskExecutor")
+    @Autowired
+    private ThreadPoolTaskExecutor taskExecutor;
 
     @Autowired
-    private SipConfig config;
-
-    @Autowired
-    private EventPublisher publisher;
+    private SipConfig sipConfig;
+    private AtomicBoolean processing = new AtomicBoolean(false);
 
     @Override
     public void afterPropertiesSet() throws Exception {
@@ -64,122 +67,120 @@
 
     @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) {
-                        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(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);
-                    } else {
-                        deviceChannel.setParental(Integer.parseInt(getText(itemDevice, "Parental")));
-                    }
-                    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();
+            responseAck((SIPRequest) evt.getRequest(), Response.OK);
+        } catch (SipException | InvalidArgumentException | ParseException e) {
+            logger.error("[鍛戒护鍙戦�佸け璐 鐩綍鏌ヨ鍥炲: {}", e.getMessage());
         }
+        // 宸茬粡寮�鍚秷鎭鐞嗗垯璺宠繃
+        if (processing.compareAndSet(false, true)) {
+            taskExecutor.execute(() -> {
+                while (!taskQueue.isEmpty()) {
+                    // 鍏ㄥ眬寮傚父鎹曡幏锛屼繚璇佷笅涓�鏉″彲浠ュ緱鍒板鐞�
+                    try {
+                        HandlerCatchData take = taskQueue.poll();
+                        Element rootElement = null;
+                        try {
+                            rootElement = getRootElement(take.getEvt(), take.getDevice().getCharset());
+                        } catch (DocumentException e) {
+                            logger.error("[xml瑙f瀽] 澶辫触锛� ", e);
+                            continue;
+                        }
+                        if (rootElement == null) {
+                            logger.warn("[ 鏀跺埌閫氶亾 ] content cannot be null, {}", evt.getRequest());
+                            continue;
+                        }
+                        Element deviceListElement = rootElement.element("DeviceList");
+                        Element sumNumElement = rootElement.element("SumNum");
+                        Element snElement = rootElement.element("SN");
+                        int sumNum = Integer.parseInt(sumNumElement.getText());
+
+                        if (sumNum == 0) {
+                            logger.info("[鏀跺埌閫氶亾]璁惧:{}鐨�: 0涓�", take.getDevice().getDeviceId());
+                            // 鏁版嵁宸茬粡瀹屾暣鎺ユ敹
+                            storager.cleanChannelsForDevice(take.getDevice().getDeviceId());
+                            catalogDataCatch.setChannelSyncEnd(take.getDevice().getDeviceId(), null);
+                        } else {
+                            Iterator<Element> deviceListIterator = deviceListElement.elementIterator();
+                            if (deviceListIterator != null) {
+                                List<DeviceChannel> channelList = new ArrayList<>();
+                                List<String> parentChannelIds = new ArrayList<>();
+                                // 閬嶅巻DeviceList
+                                while (deviceListIterator.hasNext()) {
+                                    Element itemDevice = deviceListIterator.next();
+                                    Element channelDeviceElement = itemDevice.element("DeviceID");
+                                    if (channelDeviceElement == null) {
+                                        continue;
+                                    }
+                                    DeviceChannel channel = XmlUtil.channelContentHandler(itemDevice, device, null);
+                                    if (channel == null) {
+                                        logger.info("[鏀跺埌鐩綍璁㈤槄]锛氫絾鏄В鏋愬け璐� {}", new String(evt.getRequest().getRawContent()));
+                                        continue;
+                                    }
+                                    if (channel.getParentId() != null && channel.getParentId().equals(sipConfig.getId())) {
+                                        channel.setParentId(null);
+                                    }
+                                    SipUtils.updateGps(channel, device.getGeoCoordSys());
+                                    channel.setDeviceId(take.getDevice().getDeviceId());
+
+                                    channelList.add(channel);
+                                }
+                                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);
+                                    }
+                                }
+                            }
+
+                        }
+                    } catch (Exception e) {
+                        logger.warn("[鏀跺埌閫氶亾] 鍙戠幇鏈鐞嗙殑寮傚父, \r\n{}", evt.getRequest());
+                        logger.error("[鏀跺埌閫氶亾] 寮傚父鍐呭锛� ", e);
+                    }
+                }
+                processing.set(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