From 8219cdfcb825b6d75c3a235a7e5195c8cac9a2ea Mon Sep 17 00:00:00 2001
From: songhuangfeng123 <shf18767906695@163.com>
Date: 星期二, 06 九月 2022 11:32:14 +0800
Subject: [PATCH] 统计接口rpc

---
 incident-manage/incident-manage-service/src/main/java/com/gkhy/safePlatform/incidentManage/service/impl/AccidentCountServiceImpl.java |   29 +++++++++++++++++++----------
 1 files changed, 19 insertions(+), 10 deletions(-)

diff --git a/incident-manage/incident-manage-service/src/main/java/com/gkhy/safePlatform/incidentManage/service/impl/AccidentCountServiceImpl.java b/incident-manage/incident-manage-service/src/main/java/com/gkhy/safePlatform/incidentManage/service/impl/AccidentCountServiceImpl.java
index b6ac38e..2395704 100644
--- a/incident-manage/incident-manage-service/src/main/java/com/gkhy/safePlatform/incidentManage/service/impl/AccidentCountServiceImpl.java
+++ b/incident-manage/incident-manage-service/src/main/java/com/gkhy/safePlatform/incidentManage/service/impl/AccidentCountServiceImpl.java
@@ -53,19 +53,24 @@
 
         getSingleTimeData(incidentManageCountRPCResp, query);
 
-        return new SearchResultVO<>(
+        SearchResultVO searchResultVO = new SearchResultVO<>(
                 false,
                 null,
                 null,
                 null,
-                1L,
+                null,
                 incidentManageCountRPCResp,
                 ResultCodes.OK
         );
+        searchResultVO.setCount(1);
+        return searchResultVO;
     }
 
     @Override
     public SearchResultVO<List<IncidentManageRPCResp>> getCountByDeptIds(IncidentManageCountRPCReq query) {
+        if (query.getYear() == null) {
+            throw new AccidentException(AccidentResultCodes.YEAR_NULL);
+        }
         List<IncidentManageRPCResp> list = new ArrayList<>();
 
         // 获取该部门及其子部门的所有信息
@@ -80,23 +85,27 @@
             getSingleTimeData(incidentManageCountRPCResp, query);
             list.add(incidentManageCountRPCResp);
         }
-        return new SearchResultVO<>(
+        SearchResultVO searchResultVO = new SearchResultVO<>(
                 false,
                 null,
                 null,
                 null,
-                (long) list.size(),
+                null,
                 list,
                 ResultCodes.OK
         );
+        if (!list.isEmpty()){
+            searchResultVO.setCount(list.size());
+        }
+        return searchResultVO;
     }
 
     private void getSingleTimeData(IncidentManageRPCResp incidentManageRPCResp, IncidentManageCountRPCReq query) {
+
         List<IncidentManageCountRPCResp> resList = new ArrayList<>();
 
-
         List<AccidentReportCountRPC> accidentReportCountRPCList = new ArrayList<>();
-        Long deptId = query.getDeptId();
+        Long deptId = incidentManageRPCResp.getDeptId();
         String startTime;
         String endTime;
         if (query.getMonth() == null) {
@@ -113,13 +122,13 @@
         }
 
         // 拼装第一层时间(去重)
-        Set<String> timeSET = accidentReportCountRPCList.stream().map(AccidentReportCountRPC::getTime).collect(Collectors.toSet());
+        Set<Integer> timeSET = accidentReportCountRPCList.stream().map(AccidentReportCountRPC::getTime).collect(Collectors.toSet());
         // 拼装第一层时间(排序)
-        List<String> timeList = timeSET.stream().sorted().collect(Collectors.toList());
+        List<Integer> timeList = timeSET.stream().sorted().collect(Collectors.toList());
         if (!CollectionUtils.isEmpty(timeSET)) {
-            for (String str : timeSET) {
+            for (Integer integer : timeSET) {
                 IncidentManageCountRPCResp incidentManageCountRPCResp = new IncidentManageCountRPCResp();
-                incidentManageCountRPCResp.setTime(str);
+                incidentManageCountRPCResp.setTime(integer);
                 incidentManageCountRPCResp.setDetail(new ArrayList<>());
                 resList.add(incidentManageCountRPCResp);
             }

--
Gitblit v1.9.2