From 7d1feae71bfd79b9021e516916b79f918bfcbcd9 Mon Sep 17 00:00:00 2001
From: “mitao” <“2763622819@qq.com”>
Date: 星期四, 22 一月 2026 16:36:52 +0800
Subject: [PATCH] Merge remote-tracking branch 'origin/master'

---
 ruoyi-system/src/main/java/com/ruoyi/system/service/impl/ProjectInfoServiceImpl.java |   93 ++++++++++++++++++++++++++++++++++++++++++----
 1 files changed, 84 insertions(+), 9 deletions(-)

diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/ProjectInfoServiceImpl.java b/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/ProjectInfoServiceImpl.java
index ae0efc4..e1db0fa 100644
--- a/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/ProjectInfoServiceImpl.java
+++ b/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/ProjectInfoServiceImpl.java
@@ -4,13 +4,18 @@
 import cn.hutool.core.date.DateUtil;
 import com.alibaba.fastjson2.JSONArray;
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
+import com.ruoyi.system.model.Project;
 import com.ruoyi.system.model.ProjectInfo;
 import com.ruoyi.system.mapper.ProjectInfoMapper;
 import com.ruoyi.system.model.Record;
+import com.ruoyi.system.query.ProjectDangerousTrendQuery;
 import com.ruoyi.system.query.ProjectHiddenDangerTrendListQuery;
+import com.ruoyi.system.service.ProjectCategoryCalculator;
 import com.ruoyi.system.service.ProjectInfoService;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import com.ruoyi.system.service.ProjectService;
 import com.ruoyi.system.service.RecordService;
+import com.ruoyi.system.vo.ProjectDangerousTrendVo;
 import com.ruoyi.system.vo.ProjectHiddenDangerList;
 import com.ruoyi.system.vo.ProjectHiddenDangerNextList;
 import com.ruoyi.system.vo.ProjectHiddenDangerTrendVo;
@@ -19,13 +24,13 @@
 
 import java.math.BigDecimal;
 import java.math.RoundingMode;
+import java.text.ParseException;
+import java.text.SimpleDateFormat;
 import java.time.LocalDate;
 import java.time.YearMonth;
 import java.time.temporal.ChronoUnit;
 import java.time.temporal.TemporalAdjusters;
-import java.util.ArrayList;
-import java.util.List;
-import java.util.Map;
+import java.util.*;
 
 import static java.util.stream.Collectors.groupingBy;
 
@@ -41,6 +46,12 @@
 public class ProjectInfoServiceImpl extends ServiceImpl<ProjectInfoMapper, ProjectInfo> implements ProjectInfoService {
 
 
+    @Autowired
+    private ProjectCategoryCalculator projectCategoryCalculator;
+
+    @Autowired
+    private ProjectService projectService;
+
     @Override
     public List<ProjectHiddenDangerTrendVo> getProjectHiddenDangerTrend(Integer projectId) {
         // 获取上月第一天的时间
@@ -49,9 +60,19 @@
         for (Record record : list) {
             String hierarchyNames = record.getHierarchyNames();
             JSONArray jsonArray = JSONArray.parseArray(hierarchyNames);
-            // 取出倒数第4个
-            String name = jsonArray.getString(jsonArray.size()-4);
-            record.setHierarchyName( name);
+            if(record.getDeviceId()!=null){
+                // 取出倒数第3个
+                String name = jsonArray.getString(jsonArray.size()-3);
+
+                record.setHierarchyName( name);
+            }else {
+                // 取出倒数第4个
+                String name = jsonArray.getString(jsonArray.size()-4);
+
+                record.setHierarchyName( name);
+            }
+
+
         }
 
         LocalDate thisMonth = LocalDate.now().minusMonths(0).with(TemporalAdjusters.firstDayOfMonth());
@@ -89,9 +110,18 @@
         for (Record record : list) {
             String hierarchyNames = record.getHierarchyNames();
             JSONArray jsonArray = JSONArray.parseArray(hierarchyNames);
-            // 取出倒数第4个
-            String name = jsonArray.getString(jsonArray.size()-4);
-            record.setHierarchyName( name);
+            if(record.getDeviceId()!=null){
+                // 取出倒数第3个
+                String name = jsonArray.getString(jsonArray.size()-3);
+
+                record.setHierarchyName( name);
+            }else {
+                // 取出倒数第4个
+                String name = jsonArray.getString(jsonArray.size()-4);
+
+                record.setHierarchyName( name);
+            }
+
         }
 
         // 算出跨度几个月
@@ -132,6 +162,51 @@
 
     }
 
+    @Override
+    public List<ProjectDangerousTrendVo> getProjectDangerousTrend(ProjectDangerousTrendQuery query) {
+        List<ProjectDangerousTrendVo> list = new ArrayList<>();
+        Project project = projectService.getById(query.getProjectId());
+        SimpleDateFormat simpleDateFormat = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
+        // 算出跨度几个月
+        int months = calculateMonthsBetween(query.getStartTime(),query.getEndTime());
+        ArrayList<Project> projects = new ArrayList<>();
+        projects.add( project);
+        String temp=null;
+        try {
+            for (int i = 0; i < months; i++) {
+                ProjectDangerousTrendVo vo = new ProjectDangerousTrendVo();
+                LocalDate localDate = getMonthBoundaries(query.getStartTime(), true).plusMonths(i);
+                String dateStr = localDate.toString().substring(0, 7);
+                String start = localDate+" 00:00:00";
+                String end = getMonthBoundaries(query.getStartTime(), false).plusMonths(i)+" 23:59:59";
+                Date startDate = simpleDateFormat.parse(start);
+                Date endDate = simpleDateFormat.parse(end);
+                vo.setTime(dateStr);
+
+                if(project.getType()==2){
+                    Map<Integer, String> integerStringMap = projectCategoryCalculator.calculateFixedAssetCategories(projects, Long.valueOf(project.getClientId()), startDate, endDate);
+                    String s = integerStringMap.get(query.getProjectId());
+                    vo.setLevel(s);
+                }else {
+                    Map<Integer, String> integerStringMap = projectCategoryCalculator.calculateOngoingProjectCategories(projects, Long.valueOf(project.getClientId()), startDate, endDate);
+                    String s = integerStringMap.get(query.getProjectId());
+                    vo.setLevel(s);
+                }
+                // 如果没有则沿用上次的
+                if(vo.getLevel()!=null){
+                    temp=vo.getLevel();
+                }else {
+                    vo.setLevel(temp);
+                }
+                list.add( vo);
+            }
+        }catch (ParseException e){
+            e.printStackTrace();
+        }
+
+        return list;
+    }
+
 
     public static LocalDate getMonthBoundaries(String yearMonthStr,boolean isInclusive) {
         // 解析字符串为YearMonth

--
Gitblit v1.7.1