[chg]发送数据增加数量过滤

label-tool-keliu/src/main/java/com/viontech/label/tool/keliu/controller/KeliuController.java

@@ -24,15 +24,18 @@ import java.io.BufferedReader;
 import java.io.IOException;
 import java.io.InputStream;
 import java.io.InputStreamReader;
+import java.util.ArrayList;
 import java.util.Date;
 import java.util.HashMap;
 import java.util.LinkedList;
 import java.util.List;
+import java.util.Map;
 import java.util.concurrent.Future;
 import java.util.concurrent.LinkedBlockingDeque;
 import java.util.concurrent.ThreadPoolExecutor;
 import java.util.concurrent.TimeUnit;
 import java.util.concurrent.atomic.AtomicLong;
+import java.util.stream.Collectors;
 
 /**
  * .
@@ -79,8 +82,15 @@ public class KeliuController {
             try {
                 AtomicLong count = new AtomicLong();
                 List<FaceRecognition> faceRecognitions = keliuRepository.getFaceRecognitionsByDateAndMallId(date, mallId, direction, deviceSerialNum, personType, channelSerialNum, startTime, endTime);
-                log.info("上传数据量:" + faceRecognitions.size());
-                for (FaceRecognition faceRecognition : faceRecognitions) {
+                Map<String, List<FaceRecognition>> faceRecognitionMap = faceRecognitions.stream().collect(Collectors.groupingBy(FaceRecognition::getPersonUnid));
+                List<FaceRecognition> faceRecognitionList = new ArrayList<>();
+                faceRecognitionMap.forEach((k, v) -> {
+                    if (v.size() >= 8) {
+                        faceRecognitionList.addAll(v);
+                    }
+                });
+                log.info("上传数据量:" + faceRecognitionList.size());
+                for (FaceRecognition faceRecognition : faceRecognitionList) {
                     Future<JsonMessageUtil.JsonMessage> submit = threadPoolExecutor.submit(() -> {
                         try {
                             byte[] bodyPic = fileService.getFile(faceRecognition.getBodyPath());
@@ -88,7 +98,7 @@ public class KeliuController {
                             HttpEntity<MultiValueMap<String, Object>> requestEntity = getRequestEntity(faceRecognition, bodyPic, bodyFeature, packId, taskId);
                             ResponseEntity<JsonMessageUtil.JsonMessage> exchange = restTemplate.exchange(vionConfig.getTargetUrl(), HttpMethod.POST, requestEntity, JsonMessageUtil.JsonMessage.class);
                             JsonMessageUtil.JsonMessage body = exchange.getBody();
-                            log.info("unid:{},msg:{},count:{}/{}", faceRecognition.getUnid(), body.getMsg(), count.incrementAndGet(), faceRecognitions.size());
+                            log.info("unid:{},msg:{},count:{}/{}", faceRecognition.getUnid(), body.getMsg(), count.incrementAndGet(), faceRecognitionList.size());
                             return body;
                         } catch (Exception e) {
                             log.info("", e);