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