From 0225ef3c66a2cc57dfc572b85848178bedd22fa3 Mon Sep 17 00:00:00 2001
From: “mitao” <“2763622819@qq.com”>
Date: 星期四, 02 四月 2026 17:00:01 +0800
Subject: [PATCH] fix:委托方大屏项目列表本月检查风险计算

---
 ruoyi-system/src/main/java/com/ruoyi/system/service/impl/DeviceUsageServiceImpl.java |  389 +++++++++++++++++++++---------------------------------
 1 files changed, 153 insertions(+), 236 deletions(-)

diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/DeviceUsageServiceImpl.java b/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/DeviceUsageServiceImpl.java
index 596e933..63ee79f 100644
--- a/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/DeviceUsageServiceImpl.java
+++ b/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/DeviceUsageServiceImpl.java
@@ -1,8 +1,7 @@
 package com.ruoyi.system.service.impl;
 
 
-import com.alibaba.fastjson2.JSON;
-import com.alibaba.fastjson2.JSONObject;
+import cn.hutool.core.date.DateUtil;
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
@@ -12,6 +11,7 @@
 import com.ruoyi.system.query.*;
 import com.ruoyi.system.service.ClientParamService;
 import com.ruoyi.system.service.DeviceUsageService;
+import com.ruoyi.system.service.ProjectCategoryCalculator;
 import com.ruoyi.system.vo.*;
 import com.ruoyi.system.vo.DeviceCountVo;
 import com.ruoyi.system.vo.DeviceListVo;
@@ -22,7 +22,6 @@
 
 import java.math.BigDecimal;
 import java.text.ParseException;
-import java.text.SimpleDateFormat;
 import java.time.LocalDate;
 import java.time.LocalDateTime;
 import java.time.YearMonth;
@@ -51,6 +50,8 @@
     private TaskMapper taskMapper;
     @Autowired
     private RecordMapper recordMapper;
+    @Autowired
+    private ProjectCategoryCalculator projectCategoryCalculator;
 
     @Override
     public List<String> getDeviceUsageList(Integer id) {
@@ -80,9 +81,6 @@
 
     @Override
     public List<EntrustProjectVo> projectList(Page<EntrustProjectVo> page, EntrustListQuery query) {
-        Integer pageNum = query.getPageNum();
-        Integer pageSize = query.getPageSize();
-
         String startTime = query.getStartTime();
         String endTime = query.getEndTime();
 
@@ -127,98 +125,27 @@
             projectIds.add(-1);
         }
         query.setProjectIds(projectIds);
+
+        Map<Integer, String> categoryMap = calculateProjectCategoryMap(projects, query);
         // 查询检查任务
         LambdaQueryWrapper<Task> taskWrapper = new LambdaQueryWrapper<>();
         taskWrapper.in(Task::getProjectId, projectIds)
                 .eq(Task::getDeleted, false);
         List<Task> taskList = taskMapper.selectList(taskWrapper);
-        List<Integer> checkTaskIds = taskList.stream().map(Task::getId).collect(Collectors.toList());
-        if (checkTaskIds.isEmpty()) {
-            checkTaskIds.add(-1);
-        }
-        // 查询重大隐患记录(level 表示隐患等级,0 表示重大隐患)
-        LambdaQueryWrapper<Record> recordWrapper = new LambdaQueryWrapper<>();
-        recordWrapper.eq(Record::getDeleted, false)
-                .in(Record::getCheckTaskId, checkTaskIds)
-                .eq(Record::getLevel, 0); // 重大隐患等级
-        List<Integer> recordIds = new ArrayList<>(recordMapper.selectList(recordWrapper).stream().map(Record::getId).toList());
-        if (recordIds.isEmpty()) {
-            recordIds.add(-1);
-        }
-
-        Map<Integer, Integer> projectMap = projectMapper.selectList(new LambdaQueryWrapper<Project>()
-                .eq(Project::getDeleted, false)).stream().collect(Collectors.toMap(Project::getId, Project::getClientId));
-        Map<Integer, String> clientParamMap = clientParamMapper.selectList(new LambdaQueryWrapper<ClientParam>())
-                .stream().collect(Collectors.toMap(ClientParam::getClientId, ClientParam::getUnderConstructionProject));
-        Map<Integer, List<Task>> taskMap = taskMapper.selectList(new LambdaQueryWrapper<Task>()
-                        .ge(Task::getCreateTime, startTime)
-                        .le(Task::getCreateTime, LocalDateTime.now())
-                        .in(Task::getId, checkTaskIds)
-                        .isNotNull(Task::getProjectId))
-                .stream().collect(Collectors.groupingBy(Task::getProjectId));
-        Map<Integer, Task> taskRecordMap = taskMapper.selectList(new LambdaQueryWrapper<Task>()
-                        .ge(Task::getCreateTime, startTime)
-                        .le(Task::getCreateTime, LocalDateTime.now())
-                        .in(Task::getId, checkTaskIds)
-                        .isNotNull(Task::getProjectId))
-                .stream().collect(Collectors.toMap(Task::getId,e->e));
-        Map<Integer, List<EvalResult>> resultMap = evalResultMapper.selectList(new LambdaQueryWrapper<EvalResult>()
-                .in(EvalResult::getCheckTaskId, checkTaskIds)
-        ).stream().collect(Collectors.groupingBy(EvalResult::getProjectId));
-        List<Record> records = recordMapper.selectList(new LambdaQueryWrapper<Record>()
-                .in(Record::getId, recordIds)
-                .eq(Record::getDeleted, false));
         List<EntrustProjectVo> deviceList = this.baseMapper.projectList(page, query);
         for (EntrustProjectVo deviceListVo : deviceList) {
-            List<EvalResult> evalResults = resultMap.get(deviceListVo.getId());
-            if (evalResults != null) {
-                float temp = 0;
-                // 计算安全检查平均分
-                float avgSafetyScore = (float) evalResults.stream()
-                        .filter(e -> e.getScore() != null)
-                        .mapToDouble(EvalResult::getScore)
-                        .average()
-                        .orElse(0.0);
-                // 计算设备检查平均分(需要解析 device_res_map)
-                List<Float> deviceScores = evalResults.stream()
-                        .filter(e -> StringUtils.hasText(e.getDeviceResMap()))
-                        .map(e -> calculateDeviceScore(e.getDeviceResMap()))
-                        .filter(score -> score != null && score > 0)
-                        .toList();
-
-                if (!deviceScores.isEmpty()) {
-                    float avgDeviceScore = (float) deviceScores.stream()
-                            .mapToDouble(Float::doubleValue)
-                            .average()
-                            .orElse(0.0);
-                    temp = (avgSafetyScore + avgDeviceScore) / 2;
-                }
-                Integer clientId = projectMap.get(deviceListVo.getId());
-                String s = clientParamMap.get(clientId);
-                JSONObject jsonObject = JSON.parseObject(s);
-                String a = jsonObject.getString("a");
-                String b = jsonObject.getString("b");
-                String c = jsonObject.getString("c");
-                if (temp >= Float.parseFloat(a)) {
-                    deviceListVo.setRisk("A类");
-                } else if (temp >= Float.parseFloat(b)) {
-                    deviceListVo.setRisk("B类");
-                } else if (temp >= Float.parseFloat(c)) {
-                    deviceListVo.setRisk("C类");
-                }
-            } else {
-                deviceListVo.setRisk(null);
-            }
-            List<Task> tasks = taskMap.get(deviceListVo.getId());
+            deviceListVo.setRisk(formatProjectCategory(categoryMap.get(deviceListVo.getId())));
+            List<Task> tasks = taskList.stream().filter(e -> e.getProjectId().equals(deviceListVo.getId())).toList();
+//            List<Task> tasks = taskMap.get(deviceListVo.getId());
             if (tasks != null && !tasks.isEmpty()) {
-                List<Integer> taskIds = tasks.stream().map(Task::getId).toList();
-                List<Record> recordList = records.stream().filter(e -> taskIds.contains(e.getCheckTaskId())).toList();
-                Record record = recordList.stream().max(Comparator.comparing(Record::getCreateTime)).orElse(null);
-                if (record != null) {
-                    Task task = taskRecordMap.get(record.getCheckTaskId());
-                    if (task!=null){
-                        deviceListVo.setCheckTime(task.getCreateTime());
-                    }
+
+                Task task1 = tasks.stream().filter(e -> e.getDate().getYear() == LocalDateTime.now().getYear() && e.getDate().getMonth().getValue() == LocalDateTime.now().getMonth().getValue()).findFirst().orElse(null);
+
+//                List<Record> recordList = records.stream().filter(e -> taskIds.contains(e.getCheckTaskId())).toList();
+//                Record record = recordList.stream().max(Comparator.comparing(Record::getCreateTime)).orElse(null);
+                if (task1 != null && (task1.getState() == 3 || Integer.valueOf(5).equals(task1.getState()) ||
+                        Integer.valueOf(6).equals(task1.getState()) || Integer.valueOf(7).equals(task1.getState()))) {
+                    deviceListVo.setCheckTime(task1.getDate().atStartOfDay());
                     deviceListVo.setCheckResult("已检查");
                 } else {
                     deviceListVo.setCheckTime(null);
@@ -234,74 +161,56 @@
         return deviceList;
     }
 
-    private static Float calculateDeviceScore(String deviceResMapJson) {
-        try {
-            // 1. 解析 JSON 字符串
-            JSONObject deviceMap = JSON.parseObject(deviceResMapJson);
-            if (deviceMap == null || deviceMap.isEmpty()) {
-                log.debug("设备评估结果映射为空");
-                return null;
-            }
+    private Map<Integer, String> calculateProjectCategoryMap(List<Project> projects, EntrustListQuery query) {
+        if (projects.isEmpty() || query.getClientId() == null || query.getCategory() == null) {
+            return Collections.emptyMap();
+        }
 
-            // 2. 提取所有设备的分数
-            List<Float> scores = new ArrayList<>();
-            for (String deviceId : deviceMap.keySet()) {
-                try {
-                    // 获取设备数据对象
-                    JSONObject deviceData = deviceMap.getJSONObject(deviceId);
-                    if (deviceData == null) {
-                        log.debug("设备数据为空,deviceId: {}", deviceId);
-                        continue;
-                    }
+        Date startDate = resolveProjectCategoryStartDate(query.getStartTime());
+        Date endDate = resolveProjectCategoryEndDate(query.getEndTime());
+        if (query.getCategory() == 1) {
+            return projectCategoryCalculator.calculateOngoingProjectCategories(
+                    projects, query.getClientId().longValue(), startDate, endDate
+            );
+        }
+        if (query.getCategory() == 2) {
+            return projectCategoryCalculator.calculateFixedAssetCategories(
+                    projects, query.getClientId().longValue(), startDate, endDate
+            );
+        }
+        return Collections.emptyMap();
+    }
 
-                    // 获取 content 对象
-                    JSONObject content = deviceData.getJSONObject("content");
-                    if (content == null) {
-                        log.debug("设备 content 为空,deviceId: {}", deviceId);
-                        continue;
-                    }
+    private Date resolveProjectCategoryStartDate(String startTime) {
+        if (!StringUtils.hasLength(startTime)) {
+            return DateUtil.beginOfMonth(new Date());
+        }
+        return DateUtil.beginOfDay(DateUtil.parse(normalizeDateText(startTime)));
+    }
 
-                    // 获取 data 对象
-                    JSONObject data = content.getJSONObject("data");
-                    if (data == null) {
-                        log.debug("设备 data 为空,deviceId: {}", deviceId);
-                        continue;
-                    }
+    private Date resolveProjectCategoryEndDate(String endTime) {
+        if (!StringUtils.hasLength(endTime)) {
+            return DateUtil.endOfMonth(new Date());
+        }
+        return DateUtil.endOfDay(DateUtil.parse(normalizeDateText(endTime)));
+    }
 
-                    // 获取 score 字段
-                    if (!data.containsKey("score")) {
-                        log.debug("设备 data 中不包含 score 字段,deviceId: {}", deviceId);
-                        continue;
-                    }
+    private String normalizeDateText(String dateText) {
+        return dateText.replace("年", "-")
+                .replace("月", "-")
+                .replace("日", "")
+                .replace("/", "-")
+                .trim();
+    }
 
-                    Float score = data.getFloat("score");
-                    if (score != null && score > 0) {
-                        scores.add(score);
-                        log.debug("成功提取设备得分,deviceId: {}, score: {}", deviceId, score);
-                    }
-                } catch (Exception e) {
-                    log.warn("提取设备得分失败,deviceId: {}", deviceId, e);
-                }
-            }
-
-            // 3. 计算平均分
-            if (scores.isEmpty()) {
-                log.debug("没有有效的设备得分数据");
-                return null;
-            }
-
-            float avgScore = (float) scores.stream()
-                    .mapToDouble(Float::doubleValue)
-                    .average()
-                    .orElse(0.0);
-
-            log.debug("设备平均得分计算完成,设备数量: {}, 平均分: {}", scores.size(), avgScore);
-            return avgScore;
-
-        } catch (Exception e) {
-            log.warn("解析设备得分失败,deviceResMapJson: {}", deviceResMapJson, e);
+    private String formatProjectCategory(String category) {
+        if (!StringUtils.hasLength(category)) {
             return null;
         }
+        if (category.endsWith("类")) {
+            return category;
+        }
+        return category + "类";
     }
 
     @Override
@@ -367,7 +276,7 @@
                         .orElse(null);
                 if (record != null) {
                     Task task = taskMapper.selectById(record.getCheckTaskId());
-                    if (task != null){
+                    if (task != null) {
                         deviceListVo.setCheckTime(task.getCreateTime());
                     }
                     deviceListVo.setCheckResult("已检查");
@@ -402,32 +311,34 @@
             startTime = "";
         }
         LambdaQueryWrapper<Project> projectWrapper = new LambdaQueryWrapper<>();
-        switch (query.getCategory()) {
-            case 1:
-                // 查询在建资产项目
-                projectWrapper.eq(Project::getDeleted, false)
-                        .ne(Project::getType, 2) // 在建工程
-                        .and(wrapper -> wrapper
-                                .eq(Project::getStatus, 1)  // 在建项目
-                                .or(w -> w
-                                        .eq(Project::getStatus, 2)  // 已完成项目
+        if(query.getCategory()!=null) {
+            switch (query.getCategory()) {
+                case 1:
+                    // 查询在建资产项目
+                    projectWrapper.eq(Project::getDeleted, false)
+                            .ne(Project::getType, 2) // 在建工程
+                            .and(wrapper -> wrapper
+                                    .eq(Project::getStatus, 1)  // 在建项目
+                                    .or(w -> w
+                                            .eq(Project::getStatus, 2)  // 已完成项目
 
-                                        .between(StringUtils.hasLength(startTime),Project::getSuccessTime, startTime, endTime)  // 完成时间 >= 开始日期// 完成时间 <= 结束日期
-                                )
-                        );
-                break;
-            case 2:
-                // 查询固定资产项目
-                projectWrapper.eq(Project::getDeleted, false)
-                        .eq(Project::getType, 2) // 在建工程
-                        .and(wrapper -> wrapper
-                                .eq(Project::getStatus, 1)  // 在建项目
-                                .or(w -> w
-                                        .eq(Project::getStatus, 2)  // 已完成项目
-                                        .between(StringUtils.hasLength(startTime),Project::getSuccessTime, startTime, endTime)  // 完成时间 >= 开始日期// 完成时间 <= 结束日期
-                                )
-                        );
-                break;
+                                            .between(StringUtils.hasLength(startTime), Project::getSuccessTime, startTime, endTime)  // 完成时间 >= 开始日期// 完成时间 <= 结束日期
+                                    )
+                            );
+                    break;
+                case 2:
+                    // 查询固定资产项目
+                    projectWrapper.eq(Project::getDeleted, false)
+                            .eq(Project::getType, 2) // 在建工程
+                            .and(wrapper -> wrapper
+                                    .eq(Project::getStatus, 1)  // 在建项目
+                                    .or(w -> w
+                                            .eq(Project::getStatus, 2)  // 已完成项目
+                                            .between(StringUtils.hasLength(startTime), Project::getSuccessTime, startTime, endTime)  // 完成时间 >= 开始日期// 完成时间 <= 结束日期
+                                    )
+                            );
+                    break;
+            }
         }
         if (query.getClientId() != null) {
             projectWrapper.eq(Project::getClientId, query.getClientId());
@@ -468,32 +379,36 @@
             startTime = "";
         }
         LambdaQueryWrapper<Project> projectWrapper = new LambdaQueryWrapper<>();
-        switch (query.getCategory()) {
-            case 1:
-                // 查询在建资产项目
-                projectWrapper.eq(Project::getDeleted, false)
-                        .ne(Project::getType, 2) // 在建工程
-                        .and(wrapper -> wrapper
-                                .eq(Project::getStatus, 1)  // 在建项目
-                                .or(w -> w
-                                        .eq(Project::getStatus, 2)  // 已完成项目
+        if (query.getCategory() != null) {
 
-                                        .between(StringUtils.hasLength(startTime),Project::getSuccessTime, startTime, endTime)  // 完成时间 >= 开始日期// 完成时间 <= 结束日期
-                                )
-                        );
-                break;
-            case 2:
-                // 查询固定资产项目
-                projectWrapper.eq(Project::getDeleted, false)
-                        .eq(Project::getType, 2) // 在建工程
-                        .and(wrapper -> wrapper
-                                .eq(Project::getStatus, 1)  // 在建项目
-                                .or(w -> w
-                                        .eq(Project::getStatus, 2)  // 已完成项目
-                                        .between(StringUtils.hasLength(startTime),Project::getSuccessTime, startTime, endTime)  // 完成时间 >= 开始日期// 完成时间 <= 结束日期
-                                )
-                        );
-                break;
+
+            switch (query.getCategory()) {
+                case 1:
+                    // 查询在建资产项目
+                    projectWrapper.eq(Project::getDeleted, false)
+                            .ne(Project::getType, 2) // 在建工程
+                            .and(wrapper -> wrapper
+                                    .eq(Project::getStatus, 1)  // 在建项目
+                                    .or(w -> w
+                                            .eq(Project::getStatus, 2)  // 已完成项目
+
+                                            .between(StringUtils.hasLength(startTime), Project::getSuccessTime, startTime, endTime)  // 完成时间 >= 开始日期// 完成时间 <= 结束日期
+                                    )
+                            );
+                    break;
+                case 2:
+                    // 查询固定资产项目
+                    projectWrapper.eq(Project::getDeleted, false)
+                            .eq(Project::getType, 2) // 在建工程
+                            .and(wrapper -> wrapper
+                                    .eq(Project::getStatus, 1)  // 在建项目
+                                    .or(w -> w
+                                            .eq(Project::getStatus, 2)  // 已完成项目
+                                            .between(StringUtils.hasLength(startTime), Project::getSuccessTime, startTime, endTime)  // 完成时间 >= 开始日期// 完成时间 <= 结束日期
+                                    )
+                            );
+                    break;
+            }
         }
         if (query.getClientId() != null) {
             projectWrapper.eq(Project::getClientId, query.getClientId());
@@ -533,41 +448,43 @@
             startTime = "";
         }
         LambdaQueryWrapper<Project> projectWrapper = new LambdaQueryWrapper<>();
-        switch (query.getCategory()) {
-            case 1:
-                // 查询在建资产项目
-                projectWrapper
+        if(query.getCategory()!=null) {
+            switch (query.getCategory()) {
+                case 1:
+                    // 查询在建资产项目
+                    projectWrapper
 //                        .eq(Project::getDeleted, false)
-                        .ne(Project::getType, 2) // 在建工程
-                        .and(wrapper -> wrapper
-                                .eq(Project::getStatus, 1)  // 在建项目
-                                .or(w -> w
-                                        .eq(Project::getStatus, 2)  // 已完成项目
-                                        .between(StringUtils.hasLength(startTime),Project::getSuccessTime, startTime, endTime)  // 完成时间 >= 开始日期// 完成时间 <= 结束日期
-                                )
-                                        .or(w -> w
-                                                        .isNull(Project::getStatus)  // 已完成项目
-                                        .between(StringUtils.hasLength(startTime),Project::getSuccessTime, startTime, endTime)  // 完成时间 >= 开始日期// 完成时间 <= 结束日期
-                                        )
-                        );
-                break;
-            case 2:
-                // 查询固定资产项目
-                projectWrapper
+                            .ne(Project::getType, 2) // 在建工程
+                            .and(wrapper -> wrapper
+                                    .eq(Project::getStatus, 1)  // 在建项目
+                                    .or(w -> w
+                                            .eq(Project::getStatus, 2)  // 已完成项目
+                                            .between(StringUtils.hasLength(startTime), Project::getSuccessTime, startTime, endTime)  // 完成时间 >= 开始日期// 完成时间 <= 结束日期
+                                    )
+                                    .or(w -> w
+                                            .isNull(Project::getStatus)  // 已完成项目
+                                            .between(StringUtils.hasLength(startTime), Project::getSuccessTime, startTime, endTime)  // 完成时间 >= 开始日期// 完成时间 <= 结束日期
+                                    )
+                            );
+                    break;
+                case 2:
+                    // 查询固定资产项目
+                    projectWrapper
 //                        .eq(Project::getDeleted, false)
-                        .eq(Project::getType, 2) // 在建工程
-                        .and(wrapper -> wrapper
-                                .eq(Project::getStatus, 1)  // 在建项目
-                                .or(w -> w
-                                        .eq(Project::getStatus, 2)  // 已完成项目
-                                        .between(StringUtils.hasLength(startTime),Project::getSuccessTime, startTime, endTime)  // 完成时间 >= 开始日期// 完成时间 <= 结束日期
-                                )
-                                        .or(w -> w
-                                                        .isNull(Project::getStatus)  // 已完成项目
+                            .eq(Project::getType, 2) // 在建工程
+                            .and(wrapper -> wrapper
+                                            .eq(Project::getStatus, 1)  // 在建项目
+                                            .or(w -> w
+                                                    .eq(Project::getStatus, 2)  // 已完成项目
+                                                    .between(StringUtils.hasLength(startTime), Project::getSuccessTime, startTime, endTime)  // 完成时间 >= 开始日期// 完成时间 <= 结束日期
+                                            )
+                                            .or(w -> w
+                                                            .isNull(Project::getStatus)  // 已完成项目
 //                                        .between(StringUtils.hasLength(startTime),Project::getSuccessTime, startTime, endTime)  // 完成时间 >= 开始日期// 完成时间 <= 结束日期
-                                        )
-                        );
-                break;
+                                            )
+                            );
+                    break;
+            }
         }
         if (query.getClientId() != null) {
             projectWrapper.eq(Project::getClientId, query.getClientId());
@@ -580,14 +497,14 @@
         // 查询检查任务
         LambdaQueryWrapper<Task> taskWrapper = new LambdaQueryWrapper<>();
         taskWrapper.in(Task::getProjectId, projectIds)
-                ;
+        ;
         List<Task> tasks = taskMapper.selectList(taskWrapper);
         List<Integer> checkTaskIds = tasks.stream().map(Task::getId).collect(Collectors.toList());
 
         // 查询重大隐患记录(level 表示隐患等级,0 表示重大隐患)
         LambdaQueryWrapper<Record> recordWrapper = new LambdaQueryWrapper<>();
         recordWrapper.eq(Record::getDeleted, false)
-                .eq(Record::getRecordType,"error")
+                .eq(Record::getRecordType, "error")
                 .in(Record::getCheckTaskId, checkTaskIds);
         List<Integer> recordIds = recordMapper.selectList(recordWrapper).stream().map(Record::getId).toList();
         query.setRecordIds(recordIds);

--
Gitblit v1.7.1