From a69ece9e3bb32741a6a980648f5c935c54eda682 Mon Sep 17 00:00:00 2001
From: fuliqi <fuliqi@qq.com>
Date: 星期六, 31 八月 2024 11:12:37 +0800
Subject: [PATCH] 宇视初始化修改

---
 ycl-server/src/main/java/com/ycl/task/OsdTask.java |  140 +++++++++++++++++++++++++++++-----------------
 1 files changed, 88 insertions(+), 52 deletions(-)

diff --git a/ycl-server/src/main/java/com/ycl/task/OsdTask.java b/ycl-server/src/main/java/com/ycl/task/OsdTask.java
index 511633b..75443d2 100644
--- a/ycl-server/src/main/java/com/ycl/task/OsdTask.java
+++ b/ycl-server/src/main/java/com/ycl/task/OsdTask.java
@@ -7,6 +7,8 @@
 import com.ycl.api.DH.module.LoginModule;
 import com.ycl.api.DH.utils.DHApi;
 import com.ycl.api.HK.HKApi;
+import com.ycl.api.YS.YSApi;
+import com.ycl.api.YS.YSInit;
 import com.ycl.platform.domain.entity.TMonitor;
 import com.ycl.platform.domain.result.OSDResult;
 import com.ycl.platform.domain.result.UY.MonitorQualifyResult;
@@ -49,6 +51,10 @@
     public String DHUserName;
     @Value("${DH.password}")
     public String DHPassword;
+    @Value("${YS.userName}")
+    public String YSUserName;
+    @Value("${YS.password}")
+    public String YSPassword;
     @Autowired
     private MongoTemplate mongoTemplate;
     @Autowired
@@ -60,7 +66,7 @@
             24,
             5000,
             TimeUnit.SECONDS,
-            new ArrayBlockingQueue<>(10),
+            new ArrayBlockingQueue<>(1000),
             new ThreadPoolExecutor.CallerRunsPolicy()
     );
 
@@ -71,6 +77,10 @@
      * 鍚屾鍝佺墝鍒癿ysql
      */
     public void getOSD() {
+        log.info("寮�濮嬭幏鍙朞SD");
+//            YSApi.getOsd("51.95.66.42",YSUserName,YSPassword);
+
+        DHApi.getOsd("51.95.67.189",DHUserName,DHPassword);
         //鎵归噺淇敼娴峰悍鍝佺墝闆嗗悎
         List<String> hkList = new ArrayList<>();
         //鎵归噺淇敼澶у崕鍝佺墝闆嗗悎
@@ -83,60 +93,84 @@
         List<MonitorQualifyResult> oneMachineFileResults = mongoTemplate.find(query, MonitorQualifyResult.class);
         //闇�瑕佽�冩牳鐨勬暟鎹�
         List<String> serialNumbers = oneMachineFileResults.stream().map(result -> result.getSerialNumber().getValue()).collect(Collectors.toList());
-
-        List<TMonitor> monitors = monitorMapper.selectByNumbers(serialNumbers);
-        for (TMonitor monitor : monitors) {
-            executorService.submit(() -> {
-                if (DeviceType.HK.getType().equals(monitor.getDeviceType())) {
-                    //娴峰悍鍝佺墝
-                    OSDResult osd = HKApi.getOsdByIP(monitor.getIp(), HKUserName, HKPassword);
-                    if (checkSuccess(osdResultList, monitor, osd)) return;
-                } else if (DeviceType.DH.getType().equals(monitor.getDeviceType())) {
-                    //澶у崕鍝佺墝
-                    OSDResult osd = DHApi.getOsd(monitor.getIp(), DHUserName, DHPassword);
-                    if (checkSuccess(osdResultList, monitor, osd)) return;
-                } else if (DeviceType.YS.getType().equals(monitor.getDeviceType())) {
-                    //TODO锛氬畤瑙哸pi
-                }
-                //鏈煡鍝佺墝鎴栬�卆pi璋冪敤澶辫触锛屾尐涓墽琛屾墍鏈塧pi
-                OSDResult osd = tryAllApi(monitor, hkList, dhList, ysList);
-                if (osd != null) {
-                    synchronized (osdResultList) {
-                        osdResultList.add(osd);
+        if(!CollectionUtils.isEmpty(serialNumbers)) {
+            List<TMonitor> monitors = monitorMapper.selectByNumbers(serialNumbers);
+            monitors = monitors.stream().filter(monitor -> !StringUtils.isEmpty(monitor.getIp()) && !"127.0.0.1".equals(monitor.getIp())).collect(Collectors.toList());
+//            monitors = monitors.stream().filter(monitor -> monitor.getId() < 7000).collect(Collectors.toList());
+            log.info("澶勭悊鏁版嵁澶у皬{}", monitors.size());
+            for (TMonitor monitor : monitors) {
+                executorService.submit(() -> {
+                    if (DeviceType.HK.getType().equals(monitor.getDeviceType())) {
+                        //娴峰悍鍝佺墝
+                        OSDResult osd = HKApi.getOsdByIP(monitor.getIp(), HKUserName, HKPassword);
+                        if (checkSuccess(osdResultList, monitor, osd)) {
+                            log.info("娴峰悍璋冪敤鎴愬姛" + osd);
+                            return;
+                        }
+                    } else if (DeviceType.DH.getType().equals(monitor.getDeviceType())) {
+                        //澶у崕鍝佺墝
+                        OSDResult osd = DHApi.getOsd(monitor.getIp(), DHUserName, DHPassword);
+                        if (checkSuccess(osdResultList, monitor, osd)) {
+                            log.info("澶у崕璋冪敤鎴愬姛" + osd);
+                            return;
+                        }
+                    } else if (DeviceType.YS.getType().equals(monitor.getDeviceType())) {
+                        //TODO锛氬畤瑙哸pi
                     }
+                    //鏈煡鍝佺墝鎴栬�卆pi璋冪敤澶辫触锛屾尐涓墽琛屾墍鏈塧pi
+                    OSDResult osd = tryAllApi(monitor, hkList, dhList, ysList);
+                    if (osd != null) {
+                        synchronized (osdResultList) {
+                            osdResultList.add(osd);
+                        }
+                    }
+                });
+            }
+            executorService.shutdown();
+            try {
+                // 绛夊緟鎵�鏈変换鍔″畬鎴愶紝鏈�澶氱瓑寰�10绉掞紙鍙互鏍规嵁闇�瑕佽皟鏁达級
+                if (!executorService.awaitTermination(10, TimeUnit.SECONDS)) {
+                    // 濡傛灉瓒呮椂锛屽皾璇曞仠姝㈠綋鍓嶆鍦ㄦ墽琛岀殑浠诲姟
+                    executorService.shutdownNow();
                 }
-            });
+            } catch (InterruptedException e) {
+                // 褰撳墠绾跨▼鍦ㄧ瓑寰呰繃绋嬩腑琚腑鏂�
+                executorService.shutdownNow();
+            }
+            //缁撴潫api鎵ц锛屽紑濮嬫牎楠�
+            List<OsdCheckResult> checkResults = new ArrayList<>();
+            Map<String, TMonitor> monitorMap = monitors.stream().collect(Collectors.toMap(TMonitor::getSerialNumber, Function.identity()));
+            for (OSDResult osdResult : osdResultList) {
+                TMonitor monitor = monitorMap.get(osdResult.getSerialNumber());
+                //灏佽OsdCheckResult
+                OsdCheckResult osdCheckResult = getOsdCheckResult(osdResult, monitor);
+                checkCorrect(osdResult, monitor, osdCheckResult);
+                checkResults.add(osdCheckResult);
+            }
+            log.info("缁撴灉鏁版嵁澶у皬:{}", checkResults.size());
+            //缁撴灉瀛樺叆mongo
+            if (!CollectionUtils.isEmpty(checkResults)) {
+                log.info("瀛樺叆mongo");
+                //濡傛灉浠婂ぉ瀛樺湪涔嬪墠鐨勬暟鎹厛鍒犻櫎
+                Query pyQuery = new Query(Criteria
+                        .where("mongoCreateTime").gte(DateUtils.getDayStart(new Date())).lt(DateUtils.getDayEnd(new Date())));
+                DeleteResult result = mongoTemplate.remove(pyQuery, OsdCheckResult.class);
+                //瀛樻斁鍦╩ongo涓�
+                mongoTemplate.insertAll(checkResults);
+                // 宸ュ崟鐢熸垚
+                uyErrorTypeCheckService.osdCheck(checkResults);
+            }
+            //淇敼鏁版嵁搴撳搧鐗�
+            if (!CollectionUtils.isEmpty(hkList)) {
+                log.info("淇敼鍝佺墝");
+                monitorMapper.batchUpdateDeviceType(hkList, DeviceType.HK.getType());
+            }
+            if (!CollectionUtils.isEmpty(dhList)) {
+                log.info("淇敼鍝佺墝");
+                monitorMapper.batchUpdateDeviceType(dhList, DeviceType.DH.getType());
+            }
+            //TODO:瀹囪
         }
-        executorService.shutdown();
-        //缁撴潫api鎵ц锛屽紑濮嬫牎楠�
-        List<OsdCheckResult> checkResults = new ArrayList<>();
-        Map<String, TMonitor> monitorMap = monitors.stream().collect(Collectors.toMap(TMonitor::getSerialNumber, Function.identity()));
-        for (OSDResult osdResult : osdResultList) {
-            TMonitor monitor = monitorMap.get(osdResult.getSerialNumber());
-            //灏佽OsdCheckResult
-            OsdCheckResult osdCheckResult = getOsdCheckResult(osdResult, monitor);
-            checkCorrect(osdResult, monitor, osdCheckResult);
-            checkResults.add(osdCheckResult);
-        }
-        //缁撴灉瀛樺叆mongo
-        if(CollectionUtils.isEmpty(checkResults)){
-            //濡傛灉浠婂ぉ瀛樺湪涔嬪墠鐨勬暟鎹厛鍒犻櫎
-            Query pyQuery = new Query(Criteria
-                    .where("mongoCreateTime").gte(DateUtils.getDayStart(new Date())).lt(DateUtils.getDayEnd(new Date())));
-            DeleteResult result = mongoTemplate.remove(pyQuery, OsdCheckResult.class);
-            //瀛樻斁鍦╩ongo涓�
-            mongoTemplate.insertAll(checkResults);
-            // 宸ュ崟鐢熸垚
-            uyErrorTypeCheckService.osdCheck(checkResults);
-        }
-        //淇敼鏁版嵁搴撳搧鐗�
-        if(!CollectionUtils.isEmpty(hkList)){
-            monitorMapper.batchUpdateDeviceType(hkList,DeviceType.HK.getType());
-        }
-        if(!CollectionUtils.isEmpty(dhList)){
-            monitorMapper.batchUpdateDeviceType(dhList,DeviceType.DH.getType());
-        }
-        //TODO:瀹囪
         log.info("缁撴潫鑾峰彇OSD");
     }
 
@@ -219,6 +253,7 @@
         OSDResult hkosd = HKApi.getOsdByIP(monitor.getIp(), HKUserName, HKPassword);
         if (hkosd != null) {
             hkosd.setSerialNumber(monitor.getSerialNumber());
+            log.info("娴峰悍璋冪敤鎴愬姛"+hkosd);
             synchronized (hkList) {
                 hkList.add(monitor.getSerialNumber());
             }
@@ -228,6 +263,7 @@
         OSDResult dhosd = DHApi.getOsd(monitor.getIp(), DHUserName, DHPassword);
         if (dhosd != null) {
             dhosd.setSerialNumber(monitor.getSerialNumber());
+            log.info("澶у崕璋冪敤鎴愬姛"+dhosd);
             synchronized (dhList) {
                 dhList.add(monitor.getSerialNumber());
             }

--
Gitblit v1.8.0