From 0ccc9a0dcadfb32aa8c63b718518ec0d216fae8f Mon Sep 17 00:00:00 2001
From: xiangpei <xiangpei@timesnew.cn>
Date: 星期四, 05 九月 2024 17:06:16 +0800
Subject: [PATCH] 保存到mongo时设置标签信息

---
 ycl-server/src/main/java/com/ycl/task/HKTask.java |   86 +++++++++++++++++++++++++++++++++++++-----
 1 files changed, 75 insertions(+), 11 deletions(-)

diff --git a/ycl-server/src/main/java/com/ycl/task/HKTask.java b/ycl-server/src/main/java/com/ycl/task/HKTask.java
index d310a07..85c3811 100644
--- a/ycl-server/src/main/java/com/ycl/task/HKTask.java
+++ b/ycl-server/src/main/java/com/ycl/task/HKTask.java
@@ -3,16 +3,11 @@
 import com.alibaba.fastjson2.JSONObject;
 import com.mongodb.client.result.DeleteResult;
 import com.ycl.feign.HkApiUtil;
-import com.ycl.feign.HkApiUtil;
 import com.ycl.platform.domain.param.HK.*;
 import com.ycl.platform.domain.result.HK.*;
-import com.ycl.platform.service.ICheckIndexCarService;
-import com.ycl.platform.service.ICheckIndexFaceService;
-import com.ycl.platform.service.ICheckIndexVideoService;
-import com.ycl.platform.service.IYwThresholdService;
+import com.ycl.platform.service.*;
 import com.ycl.utils.DateUtils;
 import constant.ApiConstants;
-import com.ycl.feign.HKClient;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.beans.factory.annotation.Value;
@@ -24,6 +19,7 @@
 
 import java.util.Date;
 import java.util.List;
+import java.util.Objects;
 
 //娴峰悍瀵规帴鏁版嵁浠诲姟
 @Slf4j
@@ -41,11 +37,7 @@
     @Autowired
     private IYwThresholdService ywThresholdService;
     @Autowired
-    private ICheckIndexFaceService checkIndexFaceService;
-    @Autowired
-    private ICheckIndexCarService checkIndexCarService;
-    @Autowired
-    private ICheckIndexVideoService checkIndexVideoService;
+    private YwPointService pointService;
 
     //杞﹁締璁惧鍏ㄦ鎸囨爣鐩戞祴缁撴灉
     public void vehicleDeviceInspectionTask() {
@@ -60,6 +52,12 @@
             //濡傛灉瀛樺湪涔嬪墠鐨勬暟鎹厛鍒犻櫎
             Query query = new Query(Criteria.where("mongoCreateTime").gte(DateUtils.getDayStart(new Date())).lt(DateUtils.getDayEnd(new Date())));
             DeleteResult result = mongoTemplate.remove(query, VehicleDeviceInspectionResult.class);
+            list.stream().forEach(item -> {
+                if (Objects.nonNull(item.getExternalIndexCode())) {
+                    item.setNo(item.getExternalIndexCode());
+                }
+            });
+            pointService.setDeviceTagByGB(list);
             //瀛樻斁鍦╩ongo涓�
             mongoTemplate.insertAll(list);
             //鍚屾鐨勬暟鎹繘琛屽伐鍗曢槇鍊煎鐞�
@@ -83,6 +81,12 @@
             //濡傛灉瀛樺湪涔嬪墠鐨勬暟鎹厛鍒犻櫎
             Query query = new Query(Criteria.where("mongoCreateTime").gte(DateUtils.getDayStart(new Date())).lt(DateUtils.getDayEnd(new Date())));
             DeleteResult result = mongoTemplate.remove(query, FaceDeviceInspectionResult.class);
+            list.stream().forEach(item -> {
+                if (Objects.nonNull(item.getExternalIndexCode())) {
+                    item.setNo(item.getExternalIndexCode());
+                }
+            });
+            pointService.setDeviceTagByGB(list);
             //瀛樻斁鍦╩ongo涓�
             mongoTemplate.insertAll(list);
             //鍚屾鐨勬暟鎹繘琛屽伐鍗曢槇鍊煎鐞�
@@ -110,6 +114,12 @@
                     .where("mongoCreateTime").gte(DateUtils.getDayStart(new Date())).lt(DateUtils.getDayEnd(new Date()))
                     .and("dataType").is(ApiConstants.HK_DataType_CAR));
             DeleteResult result = mongoTemplate.remove(query, SnapshotDataMonitorResult.class);
+            carList.stream().forEach(item -> {
+                if (Objects.nonNull(item.getExternalIndexCode())) {
+                    item.setNo(item.getExternalIndexCode());
+                }
+            });
+            pointService.setDeviceTagByGB(carList);
             //瀛樻斁鍦╩ongo涓�
             carList.forEach(item -> item.setDataType(ApiConstants.HK_DataType_CAR));
             mongoTemplate.insertAll(carList);
@@ -129,6 +139,12 @@
                     .where("mongoCreateTime").gte(DateUtils.getDayStart(new Date())).lt(DateUtils.getDayEnd(new Date()))
                     .and("dataType").is(ApiConstants.HK_DataType_FACE));
             DeleteResult result = mongoTemplate.remove(query, SnapshotDataMonitorResult.class);
+            faceList.stream().forEach(item -> {
+                if (Objects.nonNull(item.getExternalIndexCode())) {
+                    item.setNo(item.getExternalIndexCode());
+                }
+            });
+            pointService.setDeviceTagByGB(faceList);
             //瀛樻斁鍦╩ongo涓�
             faceList.forEach(item -> item.setDataType(ApiConstants.HK_DataType_FACE));
             mongoTemplate.insertAll(faceList);
@@ -150,6 +166,12 @@
             Query query = new Query(Criteria
                     .where("mongoCreateTime").gte(DateUtils.getDayStart(new Date())).lt(DateUtils.getDayEnd(new Date())));
             DeleteResult result = mongoTemplate.remove(query, MonitoringDetailResult.class);
+            faceList.stream().forEach(item -> {
+                if (Objects.nonNull(item.getExternalIndexCode())) {
+                    item.setNo(item.getExternalIndexCode());
+                }
+            });
+            pointService.setDeviceTagByGB(faceList);
             //瀛樻斁鍦╩ongo涓�
             mongoTemplate.insertAll(faceList);
         }
@@ -170,6 +192,12 @@
             Query query = new Query(Criteria
                     .where("mongoCreateTime").gte(DateUtils.getDayStart(new Date())).lt(DateUtils.getDayEnd(new Date())));
             DeleteResult result = mongoTemplate.remove(query, CrossDetailResult.class);
+            faceList.stream().forEach(item -> {
+                if (Objects.nonNull(item.getExternalIndexCode())) {
+                    item.setNo(item.getExternalIndexCode());
+                }
+            });
+            pointService.setDeviceTagByGB(faceList);
             //瀛樻斁鍦╩ongo涓�
             mongoTemplate.insertAll(faceList);
         }
@@ -192,6 +220,12 @@
             Query query = new Query(Criteria
                     .where("mongoCreateTime").gte(DateUtils.getDayStart(new Date())).lt(DateUtils.getDayEnd(new Date())));
             DeleteResult result = mongoTemplate.remove(query, DataIntegrityMonitoringResult.class);
+            faceList.stream().forEach(item -> {
+                if (Objects.nonNull(item.getExternalIndexCode())) {
+                    item.setNo(item.getExternalIndexCode());
+                }
+            });
+            pointService.setDeviceTagByGB(faceList);
             //瀛樻斁鍦╩ongo涓�
             mongoTemplate.insertAll(faceList);
         }
@@ -214,6 +248,12 @@
             Query query = new Query(Criteria
                     .where("mongoCreateTime").gte(DateUtils.getDayStart(new Date())).lt(DateUtils.getDayEnd(new Date())));
             DeleteResult result = mongoTemplate.remove(query, AttrRecognitionMonitorResult.class);
+            faceList.stream().forEach(item -> {
+                if (Objects.nonNull(item.getExternalIndexCode())) {
+                    item.setNo(item.getExternalIndexCode());
+                }
+            });
+            pointService.setDeviceTagByGB(faceList);
             //瀛樻斁鍦╩ongo涓�
             mongoTemplate.insertAll(faceList);
         }
@@ -241,6 +281,12 @@
             DeleteResult result = mongoTemplate.remove(query, SnapshotDelayMonitorResult.class);
             //瀛樻斁鍦╩ongo涓�
             carList.forEach(item -> item.setDataType(ApiConstants.HK_DataType_CAR));
+            carList.stream().forEach(item -> {
+                if (Objects.nonNull(item.getExternalIndexCode())) {
+                    item.setNo(item.getExternalIndexCode());
+                }
+            });
+            pointService.setDeviceTagByGB(carList);
             mongoTemplate.insertAll(carList);
         }
 
@@ -281,6 +327,12 @@
             Query query = new Query(Criteria
                     .where("mongoCreateTime").gte(DateUtils.getDayStart(new Date())).lt(DateUtils.getDayEnd(new Date())));
             DeleteResult result = mongoTemplate.remove(query, PicAccessResult.class);
+            faceList.stream().forEach(item -> {
+                if (Objects.nonNull(item.getExternalIndexCode())) {
+                    item.setNo(item.getExternalIndexCode());
+                }
+            });
+            pointService.setDeviceTagByGB(faceList);
             //瀛樻斁鍦╩ongo涓�
             mongoTemplate.insertAll(faceList);
         }
@@ -303,6 +355,12 @@
             Query query = new Query(Criteria
                     .where("mongoCreateTime").gte(DateUtils.getDayStart(new Date())).lt(DateUtils.getDayEnd(new Date())));
             DeleteResult result = mongoTemplate.remove(query, VehicleDeviceSamplingResult.class);
+            faceList.stream().forEach(item -> {
+                if (Objects.nonNull(item.getExternalIndexCode())) {
+                    item.setNo(item.getExternalIndexCode());
+                }
+            });
+            pointService.setDeviceTagByGB(faceList);
             //瀛樻斁鍦╩ongo涓�
             mongoTemplate.insertAll(faceList);
         }
@@ -324,6 +382,12 @@
             Query query = new Query(Criteria
                     .where("mongoCreateTime").gte(DateUtils.getDayStart(new Date())).lt(DateUtils.getDayEnd(new Date())));
             DeleteResult result = mongoTemplate.remove(query, FaceDeviceSamplingResult.class);
+            faceList.stream().forEach(item -> {
+                if (Objects.nonNull(item.getExternalIndexCode())) {
+                    item.setNo(item.getExternalIndexCode());
+                }
+            });
+            pointService.setDeviceTagByGB(faceList);
             //瀛樻斁鍦╩ongo涓�
             mongoTemplate.insertAll(faceList);
         }

--
Gitblit v1.8.0