From 7d9cc96ef54399795deb5b7fc7682e6323dc1202 Mon Sep 17 00:00:00 2001
From: 648540858 <456panlinlin>
Date: 星期五, 25 三月 2022 16:05:14 +0800
Subject: [PATCH] 优化国标录像下载,添加进度条以及自动合并文件下载,需要结合新版assist服务使用。

---
 src/main/java/com/genersoft/iot/vmp/gb28181/transmit/event/request/impl/message/query/cmd/CatalogQueryMessageHandler.java |   54 ++++++++++++++++++++++++++++++++++++++++++++++--------
 1 files changed, 46 insertions(+), 8 deletions(-)

diff --git a/src/main/java/com/genersoft/iot/vmp/gb28181/transmit/event/request/impl/message/query/cmd/CatalogQueryMessageHandler.java b/src/main/java/com/genersoft/iot/vmp/gb28181/transmit/event/request/impl/message/query/cmd/CatalogQueryMessageHandler.java
index 98ed2dc..959432c 100644
--- a/src/main/java/com/genersoft/iot/vmp/gb28181/transmit/event/request/impl/message/query/cmd/CatalogQueryMessageHandler.java
+++ b/src/main/java/com/genersoft/iot/vmp/gb28181/transmit/event/request/impl/message/query/cmd/CatalogQueryMessageHandler.java
@@ -1,10 +1,7 @@
 package com.genersoft.iot.vmp.gb28181.transmit.event.request.impl.message.query.cmd;
 
 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.GbStream;
-import com.genersoft.iot.vmp.gb28181.bean.ParentPlatform;
+import com.genersoft.iot.vmp.gb28181.bean.*;
 import com.genersoft.iot.vmp.gb28181.event.EventPublisher;
 import com.genersoft.iot.vmp.gb28181.transmit.callback.DeferredResultHolder;
 import com.genersoft.iot.vmp.gb28181.transmit.cmd.impl.SIPCommanderFroPlatform;
@@ -73,17 +70,55 @@
             List<ChannelReduce> channelReduces = storager.queryChannelListInParentPlatform(parentPlatform.getServerGBId());
             // 鏌ヨ鍏宠仈鐨勭洿鎾�氶亾
             List<GbStream> gbStreams = storager.queryGbStreamListInPlatform(parentPlatform.getServerGBId());
-            int size = channelReduces.size() + gbStreams.size();
+            // 鍥炲鐩綍淇℃伅
+            List<PlatformCatalog> catalogs =  storager.queryCatalogInPlatform(parentPlatform.getServerGBId());
+            int size = catalogs.size() + channelReduces.size() + gbStreams.size();
+            if (catalogs.size() > 0) {
+                for (PlatformCatalog catalog : catalogs) {
+                    if (catalog.getParentId().equals(parentPlatform.getServerGBId())) {
+                        catalog.setParentId(parentPlatform.getDeviceGBId());
+                    }
+                    DeviceChannel deviceChannel = new DeviceChannel();
+                    deviceChannel.setChannelId(catalog.getId());
+                    deviceChannel.setName(catalog.getName());
+                    deviceChannel.setLongitude(0.0);
+                    deviceChannel.setLatitude(0.0);
+                    deviceChannel.setDeviceId(parentPlatform.getDeviceGBId());
+                    deviceChannel.setManufacture("wvp-pro");
+                    deviceChannel.setStatus(1);
+                    deviceChannel.setParental(1);
+                    deviceChannel.setParentId(catalog.getParentId());
+                    deviceChannel.setRegisterWay(1);
+                    deviceChannel.setCivilCode(parentPlatform.getDeviceGBId().substring(0,6));
+                    deviceChannel.setModel("live");
+                    deviceChannel.setOwner("wvp-pro");
+                    deviceChannel.setSecrecy("0");
+                    cmderFroPlatform.catalogQuery(deviceChannel, parentPlatform, sn, fromHeader.getTag(), size);
+                    // 闃叉鍙戦�佽繃蹇�
+                    Thread.sleep(100);
+                }
+            }
             // 鍥炲绾ц仈鐨勯�氶亾
             if (channelReduces.size() > 0) {
                 for (ChannelReduce channelReduce : channelReduces) {
+                    if (channelReduce.getCatalogId().equals(parentPlatform.getServerGBId())) {
+                        channelReduce.setCatalogId(parentPlatform.getDeviceGBId());
+                    }
                     DeviceChannel deviceChannel = storager.queryChannel(channelReduce.getDeviceId(), channelReduce.getChannelId());
+                    deviceChannel.setParental(0);
+                    deviceChannel.setParentId(channelReduce.getCatalogId());
+                    deviceChannel.setCivilCode(parentPlatform.getDeviceGBId().substring(0, 6));
                     cmderFroPlatform.catalogQuery(deviceChannel, parentPlatform, sn, fromHeader.getTag(), size);
+                    // 闃叉鍙戦�佽繃蹇�
+                    Thread.sleep(100);
                 }
             }
             // 鍥炲鐩存挱鐨勯�氶亾
             if (gbStreams.size() > 0) {
                 for (GbStream gbStream : gbStreams) {
+                    if (gbStream.getCatalogId().equals(parentPlatform.getServerGBId())) {
+                        gbStream.setCatalogId(parentPlatform.getDeviceGBId());
+                    }
                     DeviceChannel deviceChannel = new DeviceChannel();
                     deviceChannel.setChannelId(gbStream.getGbId());
                     deviceChannel.setName(gbStream.getName());
@@ -92,16 +127,17 @@
                     deviceChannel.setDeviceId(parentPlatform.getDeviceGBId());
                     deviceChannel.setManufacture("wvp-pro");
                     deviceChannel.setStatus(gbStream.isStatus()?1:0);
-    //							deviceChannel.setParentId(parentPlatform.getDeviceGBId());
+                    deviceChannel.setParentId(gbStream.getCatalogId());
                     deviceChannel.setRegisterWay(1);
-                    deviceChannel.setCivilCode(config.getDomain());
+                    deviceChannel.setCivilCode(parentPlatform.getDeviceGBId().substring(0,6));
                     deviceChannel.setModel("live");
                     deviceChannel.setOwner("wvp-pro");
                     deviceChannel.setParental(0);
                     deviceChannel.setSecrecy("0");
-                    deviceChannel.setSecrecy("0");
 
                     cmderFroPlatform.catalogQuery(deviceChannel, parentPlatform, sn, fromHeader.getTag(), size);
+                    // 闃叉鍙戦�佽繃蹇�
+                    Thread.sleep(100);
                 }
             }
             if (size == 0) {
@@ -114,6 +150,8 @@
             e.printStackTrace();
         } catch (ParseException e) {
             e.printStackTrace();
+        } catch (InterruptedException e) {
+            e.printStackTrace();
         }
 
     }

--
Gitblit v1.8.0