From ebb2b7790cd50297cf3dbbafcdda7f67971ca7a0 Mon Sep 17 00:00:00 2001
From: shenyijian <1178253012@qq.com>
Date: Sat, 14 Sep 2024 20:24:15 +0800
Subject: [PATCH] 湖北水库:新增上报修复

---
 skjcmanager/skjcmanager-service/skjcmanager-sm/src/main/java/cn/gistack/sm/sg/service/impl/SgGxServiceImpl.java |   39 +++++++++++++++++++++++++++++++++++++++
 1 files changed, 39 insertions(+), 0 deletions(-)

diff --git a/skjcmanager/skjcmanager-service/skjcmanager-sm/src/main/java/cn/gistack/sm/sg/service/impl/SgGxServiceImpl.java b/skjcmanager/skjcmanager-service/skjcmanager-sm/src/main/java/cn/gistack/sm/sg/service/impl/SgGxServiceImpl.java
index 52d5fbb..7d1c0cd 100644
--- a/skjcmanager/skjcmanager-service/skjcmanager-sm/src/main/java/cn/gistack/sm/sg/service/impl/SgGxServiceImpl.java
+++ b/skjcmanager/skjcmanager-service/skjcmanager-sm/src/main/java/cn/gistack/sm/sg/service/impl/SgGxServiceImpl.java
@@ -2,16 +2,23 @@
 
 import cn.gistack.sm.sg.DO.SgGxDO;
 import cn.gistack.sm.sg.DO.SgGxStepDO;
+import cn.gistack.sm.sg.DO.SgReportUserDO;
 import cn.gistack.sm.sg.DTO.ProjectSearchDTO;
 import cn.gistack.sm.sg.VO.SgGxVO;
 import cn.gistack.sm.sg.mapper.SgGxMapper;
+import cn.gistack.sm.sg.mapper.SgGxStepMapper;
+import cn.gistack.sm.sg.mapper.SgProgressReportMapper;
 import cn.gistack.sm.sg.service.SgGxService;
+import cn.gistack.sm.sg.service.SgGxStepService;
 import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
+import com.baomidou.mybatisplus.core.toolkit.StringUtils;
+import com.baomidou.mybatisplus.extension.conditions.query.LambdaQueryChainWrapper;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import org.springblade.core.log.exception.ServiceException;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
 
 import java.util.List;
 import java.util.Map;
@@ -29,15 +36,36 @@
 public class SgGxServiceImpl extends ServiceImpl<SgGxMapper, SgGxDO> implements SgGxService {
 	@Autowired
 	private SgGxMapper sgGxMapper;
+
+	@Autowired
+	private SgGxStepMapper sgGxStepMapper;
+
+	@Autowired
+	private SgGxStepService sgGxStepService;
+
+	@Autowired
+	private SgProgressReportMapper sgProgressReportMapper;
 	@Override
 	public IPage<SgGxVO> queryPageList(IPage<SgGxVO> page, ProjectSearchDTO searchDTO) {
+		if (StringUtils.isNotBlank(searchDTO.getGxType())){
+			List<String> childType = sgGxMapper.childType(searchDTO.getGxType());
+			searchDTO.setChildTypeList(childType);
+		}
 		List<SgGxVO> list = sgGxMapper.queryPageList(page,searchDTO);
+
 		if (CollectionUtils.isNotEmpty(list)){
 			List<Long> ids = list.stream().map(SgGxDO::getId).collect(Collectors.toList());
+			List<Long> userIds = list.stream().map(SgGxDO::getCreateUser).collect(Collectors.toList());
+			List<Map<String,Object>> users = sgProgressReportMapper.getUserByIds(userIds);
 			List<SgGxStepDO> stepDOS = sgGxMapper.queryStepByIds(ids);
 			if (CollectionUtils.isNotEmpty(stepDOS)){
 				Map<Long, List<SgGxStepDO>> map = stepDOS.stream().collect(Collectors.groupingBy(SgGxStepDO::getGxInfoId));
 				for (SgGxVO sgGxDO : list) {
+					for (Map<String, Object> user : users) {
+						if (user.get("ID").equals(sgGxDO.getCreateUser())){
+							sgGxDO.setCreateUserName(user.get("REAL_NAME").toString());
+						}
+					}
 					List<SgGxStepDO> sgGxStepDOS = map.get(sgGxDO.getId());
 					if (CollectionUtils.isNotEmpty(sgGxStepDOS)){
 						sgGxDO.setStepDOList(sgGxStepDOS);
@@ -75,4 +103,15 @@
 		byId.setStepDOList(stepDOS);
 		return byId;
 	}
+
+	@Override
+	@Transactional
+	public void delete(Long id) {
+		SgGxDO byId = this.getById(id);
+		List<SgGxStepDO> list = new LambdaQueryChainWrapper<>(sgGxStepMapper).eq(SgGxStepDO::getGxInfoId, id).list();
+		if (CollectionUtils.isNotEmpty(list)){
+			throw  new ServiceException("此工序存在工序步骤,不允许删除");
+		}
+		sgGxMapper.deleteById(id);
+	}
 }

--
Gitblit v1.9.3