From 3f342a9ce225ab809d7f22e2f0e80bb63792e266 Mon Sep 17 00:00:00 2001
From: baizonghao <1719256278@qq.com>
Date: 星期二, 14 三月 2023 20:58:01 +0800
Subject: [PATCH] AI算法效能优化

---
 ycl-platform/src/main/java/com/ycl/controller/cockpit/aiIot/AIIotController.java |   21 ++++++++++++++-------
 1 files changed, 14 insertions(+), 7 deletions(-)

diff --git a/ycl-platform/src/main/java/com/ycl/controller/cockpit/aiIot/AIIotController.java b/ycl-platform/src/main/java/com/ycl/controller/cockpit/aiIot/AIIotController.java
index 678e003..f1e4ead 100644
--- a/ycl-platform/src/main/java/com/ycl/controller/cockpit/aiIot/AIIotController.java
+++ b/ycl-platform/src/main/java/com/ycl/controller/cockpit/aiIot/AIIotController.java
@@ -2,7 +2,9 @@
 
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.ycl.api.CommonResult;
+import com.ycl.dto.allot.EfficiencyDto;
 import com.ycl.entity.video.VideoPoint;
+import com.ycl.service.allot.IEfficiencyService;
 import com.ycl.service.carManage.ICarSlagcarService;
 import com.ycl.service.equipment.IHandheldTerminalService;
 import com.ycl.service.equipment.ILoudspeakerService;
@@ -26,6 +28,8 @@
 import java.math.RoundingMode;
 import java.util.ArrayList;
 import java.util.List;
+import java.util.function.Function;
+import java.util.function.ToIntFunction;
 import java.util.stream.Collectors;
 
 /**
@@ -51,6 +55,8 @@
     ILoudspeakerService loudspeakerService;
     @Autowired
     ICarSlagcarService iCarSlagcarService;
+    @Autowired
+    IEfficiencyService iEfficiencyService;
 
     @ApiOperation(value = "鐩戞祴鏁版嵁")
     @GetMapping("/detection")
@@ -95,14 +101,15 @@
     @GetMapping("/efficiency")
     public CommonResult efficiency(@Validated CockpitVO params) {
         checkApiUtil.cockpit(params);
-        List<EfficiencyVO> efficiencyVOS = new ArrayList<>();
-        for (int i = 0; i < 4; i++) {
+        List<EfficiencyDto> list = iEfficiencyService.list();
+        int sum = list.stream().mapToInt(EfficiencyDto::getNumber).sum();
+        List<EfficiencyVO> efficiencyVOS = list.stream().map((Function<EfficiencyDto, EfficiencyVO>) efficiencyDto -> {
             EfficiencyVO a = new EfficiencyVO();
-            a.setType("閬撹矾鐮存崯");
-            a.setCount(12);
-            a.setRatio(new BigDecimal("0.63").setScale(2, RoundingMode.HALF_UP));
-            efficiencyVOS.add(a);
-        }
+            a.setType(efficiencyDto.getName());
+            a.setCount(efficiencyDto.getNumber());
+            a.setRatio(new BigDecimal((double)efficiencyDto.getNumber() / (double) sum).setScale(2, RoundingMode.HALF_UP));
+            return a;
+        }).collect(Collectors.toList());
         return CommonResult.success(efficiencyVOS);
     }
 

--
Gitblit v1.8.0