From 7215d4e27d9eede3bc16f282c957f2356d909c2d Mon Sep 17 00:00:00 2001
From: shenyijian <1178253012@qq.com>
Date: Fri, 30 Aug 2024 17:51:53 +0800
Subject: [PATCH] 湖北水库:工程项目树修改

---
 skjcmanager/skjcmanager-service/skjcmanager-sm/src/main/java/cn/gistack/sm/sg/service/impl/SgProjectInfoServiceImpl.java |   17 ++++++++++++++---
 1 files changed, 14 insertions(+), 3 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..3cbcc14 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,6 +630,8 @@
 
 			SgProjectTreeVO provinceNode = new SgProjectTreeVO();
 			provinceNode.setProvinceNm(province);
+			provinceNode.setId(UUID.randomUUID().toString().replace("-",""));
+			provinceNode.setRegionName(province);
 			provinceNode.setChildList(new ArrayList<>());
 
 			for (Map.Entry<String, Map<String, List<SgProjectTreeVO>>> cityEntry : cityMap.entrySet()) {
@@ -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()));
@@ -662,6 +671,8 @@
 			result.add(provinceNode);
 		}
 		return result;
+		}
+		return new ArrayList<>();
 	}
 
 }

--
Gitblit v1.9.3