From 7be0d5b901b999d2347774df3cc1a64927838bb3 Mon Sep 17 00:00:00 2001
From: heheng <475597332@qq.com>
Date: 星期二, 15 四月 2025 10:10:35 +0800
Subject: [PATCH] 优化每年专项数据获取

---
 src/main/java/com/ruoyi/doublePrevention/service/impl/SPIDataCountServiceImpl.java |   76 +++++++++++++++++++++++++++++++++++++
 1 files changed, 75 insertions(+), 1 deletions(-)

diff --git a/src/main/java/com/ruoyi/doublePrevention/service/impl/SPIDataCountServiceImpl.java b/src/main/java/com/ruoyi/doublePrevention/service/impl/SPIDataCountServiceImpl.java
index 827faec..2a772bb 100644
--- a/src/main/java/com/ruoyi/doublePrevention/service/impl/SPIDataCountServiceImpl.java
+++ b/src/main/java/com/ruoyi/doublePrevention/service/impl/SPIDataCountServiceImpl.java
@@ -1,6 +1,7 @@
 package com.ruoyi.doublePrevention.service.impl;
 
 import com.ruoyi.doublePrevention.entity.DoublePreventDept;
+import com.ruoyi.doublePrevention.entity.SPI.*;
 import com.ruoyi.doublePrevention.entity.dto.DataCountDangerLevelRectifiedRespDO;
 import com.ruoyi.doublePrevention.entity.dto.DataCountDangerLevelRespDO;
 import com.ruoyi.doublePrevention.entity.dto.DataCountDangerResultRespDO;
@@ -13,7 +14,6 @@
 import com.ruoyi.doublePrevention.service.baseService.PreventRiskDangerInfoService;
 import com.ruoyi.doublePrevention.service.baseService.TrHiddenDangerCheckPointService;
 import com.ruoyi.doublePrevention.vo.ResultVO;
-import io.swagger.models.auth.In;
 import org.apache.commons.lang3.ObjectUtils;
 import org.springframework.beans.BeanUtils;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -21,6 +21,9 @@
 import org.springframework.stereotype.Service;
 import org.springframework.util.CollectionUtils;
 
+import java.time.LocalDate;
+import java.time.LocalDateTime;
+import java.time.Period;
 import java.util.Calendar;
 import java.util.Date;
 import java.util.List;
@@ -158,6 +161,77 @@
         return parameterVerificationResult;
     }
 
+    @Override
+    public ResultVO<TroubleData> getSPIData(SPIDataReqBO spiDataReqBO) {
+
+        ResultVO resultVO = new ResultVO<>();
+        resultVO.setCode("200");
+        resultVO.setMsg("查询成功");
+
+        String key = "006a4740-8f2d-4fdc-a25b-a7413a37a2ba";
+
+        if (ObjectUtils.isEmpty(spiDataReqBO.getKey())){
+            resultVO.setCode("400");
+            resultVO.setMsg("key不能为空");
+            return resultVO;
+        }
+        if (!spiDataReqBO.getKey().equals(key)){
+            resultVO.setCode("400");
+            resultVO.setMsg("key不正确");
+            return resultVO;
+        }
+        LocalDateTime startTime = LocalDate.now().withDayOfMonth(1).atStartOfDay();
+        for (int i = 0; i < 12; i++) {
+            startTime = startTime.minus(Period.ofMonths(1));
+        }
+        spiDataReqBO.setStartTime(startTime);
+        spiDataReqBO.setEndTime(LocalDateTime.now());
+
+
+        List<TroubleData> spiData = checkPointService.getSPIData(spiDataReqBO);
+
+
+        resultVO.setData(spiData);
+
+        return resultVO;
+    }
+
+    @Override
+    public ResultVO<TroubleLevel> getSPIDataForPieChart(SPIDataReqBO spiDataReqBO) {
+
+        ResultVO resultVO = new ResultVO<>();
+        resultVO.setCode("200");
+        resultVO.setMsg("查询成功");
+
+        String key = "006a4740-8f2d-4fdc-a25b-a7413a37a2ba";
+
+        if (ObjectUtils.isEmpty(spiDataReqBO.getKey())){
+            resultVO.setCode("400");
+            resultVO.setMsg("key不能为空");
+            return resultVO;
+        }
+        if (!spiDataReqBO.getKey().equals(key)){
+            resultVO.setCode("400");
+            resultVO.setMsg("key不正确");
+            return resultVO;
+        }
+
+        SPIDataForPieChartRespDTO spiDataForPieChartRespDTO = new SPIDataForPieChartRespDTO();
+
+        if (ObjectUtils.isEmpty(spiDataReqBO.getYear())){
+
+            spiDataReqBO.setYear("2024");
+            TroubleLevel troubleLevel = checkPointService.getTroubleLevel(spiDataReqBO);
+            resultVO.setData(troubleLevel);
+            return resultVO;
+        }
+
+        TroubleLevel troubleLevel = checkPointService.getTroubleLevel(spiDataReqBO);
+        resultVO.setData(troubleLevel);
+
+        return resultVO;
+    }
+
 
     /**
      * @description 参数校验

--
Gitblit v1.9.2