From f10b458fc919fbcadc4936bf3a7886088e20ac71 Mon Sep 17 00:00:00 2001
From: 648540858 <456panlinlin>
Date: 星期五, 08 四月 2022 14:44:49 +0800
Subject: [PATCH] 优化订阅信息的发送与取消订阅

---
 src/main/java/com/genersoft/iot/vmp/gb28181/transmit/event/request/impl/message/response/cmd/CatalogResponseMessageHandler.java |  241 +++++++++++++++++++++++++++++-------------------
 1 files changed, 146 insertions(+), 95 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 7349a7a..cc54ec9 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
@@ -2,18 +2,22 @@
 
 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.conf.UserSetting;
+import com.genersoft.iot.vmp.gb28181.bean.*;
 import com.genersoft.iot.vmp.gb28181.event.DeviceOffLineDetector;
 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.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 com.genersoft.iot.vmp.utils.GpsUtil;
+import com.genersoft.iot.vmp.vmanager.bean.WVPResult;
 import org.dom4j.DocumentException;
 import org.dom4j.Element;
 import org.slf4j.Logger;
@@ -21,15 +25,17 @@
 import org.springframework.beans.factory.InitializingBean;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Component;
+import org.springframework.util.StringUtils;
 
 import javax.sip.InvalidArgumentException;
 import javax.sip.RequestEvent;
 import javax.sip.SipException;
 import javax.sip.message.Response;
 import java.text.ParseException;
+import java.text.SimpleDateFormat;
+import java.util.ArrayList;
 import java.util.Iterator;
-
-import static com.genersoft.iot.vmp.gb28181.utils.XmlUtil.getText;
+import java.util.List;
 
 @Component
 public class CatalogResponseMessageHandler extends SIPRequestProcessorParent implements InitializingBean, IMessageHandler {
@@ -37,14 +43,19 @@
     private Logger logger = LoggerFactory.getLogger(CatalogResponseMessageHandler.class);
     private final String cmdType = "Catalog";
 
+    private SimpleDateFormat format = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
+
     @Autowired
     private ResponseMessageHandler responseMessageHandler;
 
     @Autowired
-    private IVideoManagerStorager storager;
+    private IVideoManagerStorage storager;
 
     @Autowired
     private DeferredResultHolder deferredResultHolder;
+
+    @Autowired
+    private CatalogDataCatch catalogDataCatch;
 
     @Autowired
     private DeviceOffLineDetector offLineDetector;
@@ -54,6 +65,13 @@
 
     @Autowired
     private EventPublisher publisher;
+
+    //by brewswang
+    @Autowired
+    private UserSetting userSetting;
+
+    @Autowired
+    private IRedisCatchStorage redisCatchStorage;
 
     @Override
     public void afterPropertiesSet() throws Exception {
@@ -67,97 +85,65 @@
         try {
             rootElement = getRootElement(evt, device.getCharset());
             Element deviceListElement = rootElement.element("DeviceList");
-            Iterator<Element> deviceListIterator = deviceListElement.elementIterator();
-            if (deviceListIterator != null) {
-
-                // 閬嶅巻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
-                    storager.updateChannel(device.getDeviceId(), deviceChannel);
-                }
-
+            Element sumNumElement = rootElement.element("SumNum");
+            if (sumNumElement == null || deviceListElement == null) {
+                responseAck(evt, Response.BAD_REQUEST, "xml error");
+                return;
+            }
+            int sumNum = Integer.parseInt(sumNumElement.getText());
+            if (sumNum == 0) {
+                // 鏁版嵁宸茬粡瀹屾暣鎺ユ敹
+                storager.cleanChannelsForDevice(device.getDeviceId());
                 RequestMessage msg = new RequestMessage();
                 msg.setKey(key);
-                msg.setData(device);
+                WVPResult<Object> result = new WVPResult<>();
+                result.setCode(0);
+                result.setData(device);
+                msg.setData(result);
+                result.setMsg("鏇存柊鎴愬姛锛屽叡0鏉�");
                 deferredResultHolder.invokeAllResult(msg);
+                catalogDataCatch.del(key);
+            }else {
+                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;
+                        }
+                        //by brewswang
+                        if (NumericUtil.isDouble(XmlUtil.getText(itemDevice, "Longitude"))) {//濡傛灉鍖呭惈浣嶇疆淇℃伅锛屽氨鏇存柊涓�涓嬩綅缃�
+                            processNotifyMobilePosition(evt, itemDevice);
+                        }
+                        
+                        DeviceChannel deviceChannel = XmlUtil.channelContentHander(itemDevice);
+                        deviceChannel.setDeviceId(device.getDeviceId());
+                        logger.debug("鏀跺埌鏉ヨ嚜璁惧銆恵}銆戠殑閫氶亾: {}銆恵}銆�", device.getDeviceId(), deviceChannel.getName(), deviceChannel.getChannelId());
+                        channelList.add(deviceChannel);
+                    }
+
+                    catalogDataCatch.put(key, sumNum, device, channelList);
+                    if (catalogDataCatch.get(key).size() == sumNum) {
+                        // 鏁版嵁宸茬粡瀹屾暣鎺ユ敹
+                        boolean resetChannelsResult = storager.resetChannels(device.getDeviceId(), catalogDataCatch.get(key));
+                        RequestMessage msg = new RequestMessage();
+                        msg.setKey(key);
+                        WVPResult<Object> result = new WVPResult<>();
+                        result.setCode(0);
+                        result.setData(device);
+                        if (resetChannelsResult || sumNum ==0) {
+                            result.setMsg("鏇存柊鎴愬姛锛屽叡" + sumNum + "鏉★紝宸叉洿鏂�" + catalogDataCatch.get(key).size() + "鏉�");
+                        }else {
+                            result.setMsg("鎺ユ敹鎴愬姛锛屽啓鍏ュけ璐ワ紝鍏�" + sumNum + "鏉★紝宸叉帴鏀�" + catalogDataCatch.get(key).size() + "鏉�");
+                        }
+                        msg.setData(result);
+                        deferredResultHolder.invokeAllResult(msg);
+                        catalogDataCatch.del(key);
+                    }
+                }
                 // 鍥炲200 OK
                 responseAck(evt, Response.OK);
                 if (offLineDetector.isOnline(device.getDeviceId())) {
@@ -179,4 +165,69 @@
     public void handForPlatform(RequestEvent evt, ParentPlatform parentPlatform, Element rootElement) {
 
     }
+
+    /**
+     * 澶勭悊璁惧浣嶇疆鐨勬洿鏂�
+     *
+     * @param evt, itemDevice
+     */
+    private void processNotifyMobilePosition(RequestEvent evt, Element itemDevice) {
+        try {
+            // 鍥炲 200 OK
+            Element rootElement = getRootElement(evt);
+            MobilePosition mobilePosition = new MobilePosition();
+            Element deviceIdElement = rootElement.element("DeviceID");
+            String deviceId = deviceIdElement.getTextTrim().toString();
+            Device device = redisCatchStorage.getDevice(deviceId);
+            if (device != null) {
+                if (!StringUtils.isEmpty(device.getName())) {
+                    mobilePosition.setDeviceName(device.getName());
+                }
+            }
+            mobilePosition.setDeviceId(XmlUtil.getText(rootElement, "DeviceID"));
+
+            String time = XmlUtil.getText(itemDevice, "Time");
+            if(time==null){
+                time =  XmlUtil.getText(itemDevice, "EndTime");
+            }
+            mobilePosition.setTime(time);
+            String longitude = XmlUtil.getText(itemDevice, "Longitude");
+            if(longitude!=null) {
+                mobilePosition.setLongitude(Double.parseDouble(longitude));
+            }
+            String latitude = XmlUtil.getText(itemDevice, "Latitude");
+            if(latitude!=null) {
+                mobilePosition.setLatitude(Double.parseDouble(latitude));
+            }
+            if (NumericUtil.isDouble(XmlUtil.getText(itemDevice, "Speed"))) {
+                mobilePosition.setSpeed(Double.parseDouble(XmlUtil.getText(itemDevice, "Speed")));
+            } else {
+                mobilePosition.setSpeed(0.0);
+            }
+            if (NumericUtil.isDouble(XmlUtil.getText(itemDevice, "Direction"))) {
+                mobilePosition.setDirection(Double.parseDouble(XmlUtil.getText(itemDevice, "Direction")));
+            } else {
+                mobilePosition.setDirection(0.0);
+            }
+            if (NumericUtil.isDouble(XmlUtil.getText(itemDevice, "Altitude"))) {
+                mobilePosition.setAltitude(Double.parseDouble(XmlUtil.getText(itemDevice, "Altitude")));
+            } else {
+                mobilePosition.setAltitude(0.0);
+            }
+            mobilePosition.setReportSource("Mobile Position");
+            BaiduPoint bp = new BaiduPoint();
+            bp = GpsUtil.Wgs84ToBd09(String.valueOf(mobilePosition.getLongitude()), String.valueOf(mobilePosition.getLatitude()));
+            logger.info("鐧惧害鍧愭爣锛�" + bp.getBdLng() + ", " + bp.getBdLat());
+            mobilePosition.setGeodeticSystem("BD-09");
+            mobilePosition.setCnLng(bp.getBdLng());
+            mobilePosition.setCnLat(bp.getBdLat());
+            if (!userSetting.getSavePositionHistory()) {
+                storager.clearMobilePositionsByDeviceId(deviceId);
+            }
+            storager.insertMobilePosition(mobilePosition);
+            responseAck(evt, Response.OK);
+        } catch (DocumentException | SipException | InvalidArgumentException | ParseException e) {
+            e.printStackTrace();
+        }
+    }
 }

--
Gitblit v1.8.0