From 22b6ef7fef37496b54579d62b97e6b4e522af904 Mon Sep 17 00:00:00 2001
From: shenyijian <1178253012@qq.com>
Date: Thu, 27 Jun 2024 18:16:41 +0800
Subject: [PATCH] 湖北水库:施工:项目列表(查询所有项目)

---
 skjcmanager/skjcmanager-service/skjcmanager-sm/src/main/java/cn/gistack/sm/sg/service/impl/SgProjectInfoServiceImpl.java |   16 ++++++++++++----
 1 files changed, 12 insertions(+), 4 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 1414008..16e6114 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
@@ -44,10 +44,12 @@
 	public IPage<SgProjectInfoDO> queryPageList(IPage<SgProjectInfoDO> page, ProjectSearchDTO searchDTO) {
 		List<SgProjectInfoDO> sgProjectInfoDOS = baseMapper.queryPageList(page,searchDTO);
 		List<SgProjectInfoDO> projectProgress = getProjectProgress(sgProjectInfoMapper.queryStepByProject(searchDTO));
-		Map<Long, String> progressMap = projectProgress.stream().collect(Collectors.toMap(SgProjectInfoDO::getId, SgProjectInfoDO::getProjectProgress));
-		sgProjectInfoDOS.forEach(s->{
-			s.setProjectProgress(progressMap.get(s.getId()));
-		});
+		if (CollectionUtils.isNotEmpty(projectProgress)){
+			Map<Long, String> progressMap = projectProgress.stream().filter(s->s.getProjectProgress()!=null).collect(Collectors.toMap(SgProjectInfoDO::getId, SgProjectInfoDO::getProjectProgress));
+			sgProjectInfoDOS.forEach(s->{
+				s.setProjectProgress(progressMap.get(s.getId()));
+			});
+		}
 		return page.setRecords(sgProjectInfoDOS);
 	}
 
@@ -166,6 +168,12 @@
 		});
 		return page.setRecords(list);
 	}
+
+	@Override
+	public List<SgProjectInfoDO> listAll(ProjectSearchDTO searchDTO) {
+		return sgProjectInfoMapper.queryPageList(null,searchDTO);
+	}
+
 	/**
 	 * 获取百分比
 	 * @param bigDecimal

--
Gitblit v1.9.3