From e1cdb7930d2eb4baccff2809a420a1f0f544c934 Mon Sep 17 00:00:00 2001
From: shenyijian <1178253012@qq.com>
Date: Mon, 17 Jun 2024 18:58:41 +0800
Subject: [PATCH] 湖北水库:施工:项目整体进度

---
 skjcmanager/skjcmanager-service/skjcmanager-sm/src/main/java/cn/gistack/sm/sg/service/impl/SgProjectInfoServiceImpl.java |   56 +++++++++++++++++++++++++++++++++++++++++++++++++++++++-
 1 files changed, 55 insertions(+), 1 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 9e0db13..9311e59 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
@@ -1,14 +1,23 @@
 package cn.gistack.sm.sg.service.impl;
 
+import cn.gistack.sm.sg.DO.SgDeviceDO;
+import cn.gistack.sm.sg.DO.SgGxDO;
+import cn.gistack.sm.sg.DO.SgGxStepDO;
 import cn.gistack.sm.sg.DO.SgProjectInfoDO;
 import cn.gistack.sm.sg.DTO.ProjectSearchDTO;
+import cn.gistack.sm.sg.VO.SgProjectInfoVO;
 import cn.gistack.sm.sg.mapper.SgProjectInfoMapper;
 import cn.gistack.sm.sg.service.SgProjectInfoService;
 import com.baomidou.mybatisplus.core.metadata.IPage;
+import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 
+import java.util.ArrayList;
 import java.util.List;
+import java.util.Map;
+import java.util.stream.Collectors;
 
 /**
  * <p>
@@ -20,10 +29,55 @@
  */
 @Service
 public class SgProjectInfoServiceImpl extends ServiceImpl<SgProjectInfoMapper, SgProjectInfoDO> implements SgProjectInfoService {
-
+	@Autowired
+	private SgProjectInfoMapper sgProjectInfoMapper;
 	@Override
 	public IPage<SgProjectInfoDO> queryPageList(IPage<SgProjectInfoDO> page, ProjectSearchDTO searchDTO) {
 		List<SgProjectInfoDO> sgProjectInfoDOS = baseMapper.queryPageList(page,searchDTO);
 		return page.setRecords(sgProjectInfoDOS);
 	}
+
+	@Override
+	public List<SgProjectInfoDO> queryStepByProject(ProjectSearchDTO searchDTO) {
+		List<SgProjectInfoVO> list = sgProjectInfoMapper.queryStepByProject( searchDTO);
+		//项目分组
+		Map<Long, List<SgProjectInfoVO>> projectMap = list.stream().collect(Collectors.groupingBy(SgProjectInfoVO::getId));
+		List<SgProjectInfoDO> result = new ArrayList<>();
+		for (Map.Entry<Long, List<SgProjectInfoVO>> entry : projectMap.entrySet()) {
+			Long porId = entry.getKey();
+			List<SgProjectInfoVO> value = entry.getValue();
+			if (CollectionUtils.isEmpty(value)){
+				continue;
+			}
+			SgProjectInfoDO sgProjectInfoDO = SgProjectInfoDO.builder().id(porId).projectName(value.get(0).getProjectName()).build();
+			List<SgGxDO> sgGxDOS = new ArrayList<>();
+			//工序分组
+			Map<Long, List<SgProjectInfoVO>> GxMap = value.stream().filter(s->s.getGxId()!=null).collect(Collectors.groupingBy(SgProjectInfoVO::getGxId));
+			for (Map.Entry<Long, List<SgProjectInfoVO>> gx : GxMap.entrySet()) {
+				Long gxId = gx.getKey();
+				List<SgProjectInfoVO> gxList = gx.getValue();
+				if (CollectionUtils.isEmpty(gxList)){
+					continue;
+				}
+				SgGxDO gxDO = SgGxDO.builder().id(gxId).gxName(gxList.get(0).getGxName()).build();
+				List<SgDeviceDO> deviceDOS = new ArrayList<>();
+				//设备分组
+				Map<Long, List<SgProjectInfoVO>> dMap = gxList.stream().collect(Collectors.groupingBy(SgProjectInfoVO::getDId));
+				for (Map.Entry<Long, List<SgProjectInfoVO>> device : dMap.entrySet()) {
+					Long dId = device.getKey();
+					List<SgProjectInfoVO> deList = device.getValue();
+					SgDeviceDO sgDeviceDO = new SgDeviceDO();
+					sgDeviceDO.setId(dId);
+					sgDeviceDO.setDeviceName(deList.get(0).getDeviceName());
+					sgDeviceDO.setStepDOList(deList.stream().map(s->SgGxStepDO.builder().gxStepName(s.getGxStepName()).approvalStatus(s.getSTATUS()).build()).collect(Collectors.toList()));
+					deviceDOS.add(sgDeviceDO);
+				}
+				gxDO.setDeviceDOS(deviceDOS);
+				sgGxDOS.add(gxDO);
+			}
+			sgProjectInfoDO.setSgGxDOS(sgGxDOS);
+			result.add(sgProjectInfoDO);
+		}
+		return result;
+	}
 }

--
Gitblit v1.9.3