From 44d30cbd79df473dbdf6d28decde2bbc931c0821 Mon Sep 17 00:00:00 2001
From: shenyijian <1178253012@qq.com>
Date: Fri, 30 Aug 2024 18:19:47 +0800
Subject: [PATCH] 湖北水库：工程项目树数量修改
---
skjcmanager/skjcmanager-service/skjcmanager-sm/src/main/java/cn/gistack/sm/sg/service/impl/SgProjectInfoServiceImpl.java | 24 ++++++++++++++++++------
1 files changed, 18 insertions(+), 6 deletions(-)
diff --git a/skjcmanager/skjcmanager-service/skjcmanager-sm/src/main/java/cn/gistack/sm/sg/service/impl/SgProjectInfoServiceImpl.java b/skjcmanager/skjcmanager-service/skjcmanager-sm/src/main/java/cn/gistack/sm/sg/service/impl/SgProjectInfoServiceImpl.java
index df3110b..4f88061 100644
--- a/skjcmanager/skjcmanager-service/skjcmanager-sm/src/main/java/cn/gistack/sm/sg/service/impl/SgProjectInfoServiceImpl.java
+++ b/skjcmanager/skjcmanager-service/skjcmanager-sm/src/main/java/cn/gistack/sm/sg/service/impl/SgProjectInfoServiceImpl.java
@@ -613,8 +613,9 @@
@Override
- public List<SgProjectTreeVO> projectTree() {
- List<SgProjectTreeVO> list = sgProjectInfoMapper.projectTree();
+ public List<SgProjectTreeVO> projectTree(ProjectSearchDTO dto) {
+ List<SgProjectTreeVO> list = sgProjectInfoMapper.projectTree(dto);
+ if (CollectionUtils.isNotEmpty(list)){
// 按省、市、县分组
Map<String, Map<String, Map<String, List<SgProjectTreeVO>>>> groupedProjects = list.stream()
.collect(Collectors.groupingBy(SgProjectTreeVO::getProvinceNm,
@@ -629,8 +630,10 @@
SgProjectTreeVO provinceNode = new SgProjectTreeVO();
provinceNode.setProvinceNm(province);
+ provinceNode.setId(UUID.randomUUID().toString().replace("-",""));
+ provinceNode.setRegionName(province);
provinceNode.setChildList(new ArrayList<>());
-
+ long provinceNum = 0;
for (Map.Entry<String, Map<String, List<SgProjectTreeVO>>> cityEntry : cityMap.entrySet()) {
String city = cityEntry.getKey();
Map<String, List<SgProjectTreeVO>> countyMap = cityEntry.getValue();
@@ -638,14 +641,20 @@
SgProjectTreeVO cityNode = new SgProjectTreeVO();
List<String> cityIds = new ArrayList<>();
cityNode.setCityNm(city);
+ cityNode.setId(UUID.randomUUID().toString().replace("-",""));
+ cityNode.setRegionName(city);
cityNode.setChildList(new ArrayList<>());
for (Map.Entry<String, List<SgProjectTreeVO>> countyEntry : countyMap.entrySet()) {
String county = countyEntry.getKey();
List<SgProjectTreeVO> countyProjects = countyEntry.getValue();
-
+ for (SgProjectTreeVO countyProject : countyProjects) {
+ countyProject.setRegionName(countyProject.getResNm());
+ }
SgProjectTreeVO countyNode = new SgProjectTreeVO();
countyNode.setCountyNm(county);
+ countyNode.setRegionName(county);
+ countyNode.setId(UUID.randomUUID().toString().replace("-",""));
countyNode.setChildList(new ArrayList<>(countyProjects));
countyNode.setChildNum((long) countyProjects.size());
countyNode.setProjectIds(countyProjects.stream().map(SgProjectTreeVO::getId).collect(Collectors.toList()));
@@ -653,15 +662,18 @@
cityNode.getChildList().add(countyNode);
}
- cityNode.setChildNum((long) cityNode.getChildList().size());
+ cityNode.setChildNum((long) cityIds.size());
cityNode.setProjectIds(cityIds);
provinceNode.getChildList().add(cityNode);
+ provinceNum+= (long) cityIds.size();
}
- provinceNode.setChildNum((long) provinceNode.getChildList().size());
+ provinceNode.setChildNum(provinceNum);
result.add(provinceNode);
}
return result;
+ }
+ return new ArrayList<>();
}
}
--
Gitblit v1.9.3