From 99bb4c189d4519acda66ab39124d2b793c021a5e Mon Sep 17 00:00:00 2001
From: shenyijian <1178253012@qq.com>
Date: Fri, 27 Sep 2024 11:46:47 +0800
Subject: [PATCH] 湖北水库:施工项目批量删除

---
 skjcmanager/skjcmanager-service/skjcmanager-sm/src/main/java/cn/gistack/sm/sg/service/impl/SgProjectInfoServiceImpl.java |   63 ++++++++++++++++++++++++++++++-
 1 files changed, 61 insertions(+), 2 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 b8bfbde..030e473 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
@@ -7,6 +7,7 @@
 import cn.gistack.sm.sg.mapper.SgGxStepMapper;
 import cn.gistack.sm.sg.mapper.SgProgressReportMapper;
 import cn.gistack.sm.sg.mapper.SgProjectInfoMapper;
+import cn.gistack.sm.sg.service.SgProgressReportService;
 import cn.gistack.sm.sg.service.SgProjectInfoService;
 import cn.gistack.sm.sjztmd.entity.AttAdBase;
 import cn.gistack.sm.sjztmd.service.IAttAdBaseService;
@@ -14,8 +15,10 @@
 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.jetbrains.annotations.NotNull;
+import org.springblade.core.log.exception.ServiceException;
 import org.springblade.core.secure.utils.AuthUtil;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
@@ -48,11 +51,14 @@
 	@Autowired
 	private SgGxStepMapper sgGxStepMapper;
 
+	@Autowired
+	private SgProgressReportServiceImpl sgProgressReportService;
+
 	@Override
 	public Object queryPageList(IPage<SgProjectInfoDO> page, ProjectSearchDTO searchDTO) {
 		List<String> sgCodeS = getSgCodeByUser();
 		searchDTO.setSgCodeS(sgCodeS);
-//		setDeptId(searchDTO);
+		setAdCode(searchDTO);
 		Boolean isPage = searchDTO.getIsPage();
 		if (isPage==null) isPage = true;
 		if (isPage ){
@@ -623,6 +629,8 @@
 
 	@Override
 	public Map<String, Object> projectCount(ProjectSearchDTO searchDTO) {
+		List<String> sgCodeS = getSgCodeByUser();
+		searchDTO.setSgCodeS(sgCodeS);
 		Map<String, Object> map  = new HashMap<>();
 		List<SgProjectInfoVO> progressList = progressList(searchDTO);
 		BigDecimal reduce = progressList.stream().filter(s->s.getProjectProgressNum()!=null).map(SgProjectInfoDO::getProjectProgressNum).reduce(BigDecimal.ZERO, BigDecimal::add);
@@ -682,6 +690,7 @@
 	public List<SgProjectTreeVO> projectTree(ProjectSearchDTO dto) {
 		List<String> sgCodeS = getSgCodeByUser();
 		dto.setSgCodeS(sgCodeS);
+		setAdCode(dto);
 		List<SgProjectTreeVO> list = 	sgProjectInfoMapper.projectTree(dto);
 		if (CollectionUtils.isNotEmpty(list)){
 			list.forEach(x->{
@@ -766,9 +775,59 @@
 					provinceNode.setChildList(provinceChildren);
 					tree.add(provinceNode);
 				}
-				return tree;
+			List<SgProjectTreeVO> childList = tree.get(0).getChildList();
+			List<SgProjectTreeVO> collect = childList.stream()
+				.sorted(Comparator.comparing(s -> {
+					String scode = s.getRegionName();
+					if (scode.equals("第一标段")) {
+						return 1;
+					} else if (scode.equals("第二标段")) {
+						return 2;
+					} else if (scode.equals("第三标段")) {
+						return 3;
+					} else if (scode.equals("第四标段")) {
+						return 4;
+					} else if (scode.equals("第五标段")) {
+						return 5;
+					} else {
+						return Integer.MAX_VALUE;
+					}
+				}))
+				.collect(Collectors.toList());
+			tree.get(0).setChildList(collect);
+			return tree;
+
 		}
 		return new ArrayList<>();
 	}
 
+	private void setAdCode(ProjectSearchDTO dto) {
+		String deptId = AuthUtil.getDeptId();
+		Long roleId = sgProgressReportService.getRoleId();
+		if(roleId.equals(1811657352658042882L)){
+			dto.setAdCode(deptId);
+		}
+	}
+
+	@Override
+	public IPage<ResVO> resList(IPage<ResVO> page, ProjectSearchDTO dto) {
+		List<String> sgCodeByUser = getSgCodeByUser();
+		dto.setSgCodeS(sgCodeByUser);
+		return page.setRecords(baseMapper.resList(page,dto));
+	}
+
+	@Override
+	@Transactional
+	public void delete(String ids) {
+		List<String> list = Arrays.asList(ids.split(","));
+		list.forEach(id->{
+			List<SgProgressReportDO> reportDOS = new LambdaQueryChainWrapper<>(sgProgressReportMapper).eq(SgProgressReportDO::getProframId, id).list();
+			if (CollectionUtils.isNotEmpty(reportDOS)){
+				SgProjectInfoDO sgProjectInfoDO = sgProjectInfoMapper.selectById(id);
+				throw new ServiceException(sgProjectInfoDO.getProjectName() + "存在施工审批,无法删除!");
+			}
+			sgProjectInfoMapper.deleteById(id);
+		});
+	}
+
 }

--
Gitblit v1.9.3