From fc2adc3521931c2f7bc5b2f3a3d94ebf407828d6 Mon Sep 17 00:00:00 2001
From: yxm <zhp1521624>
Date: Mon, 01 Jul 2024 09:40:55 +0800
Subject: [PATCH] 水库堤防白蚁普查资料当前页压缩导出新增条件优化v20240701

---
 skjcmanager/skjcmanager-service/skjcmanager-sm/src/main/java/cn/gistack/sm/sjztmd/service/impl/TbAttResRsbSpillwayServiceImpl.java |   55 +++++++++++++++++++++++++++++++++++++++++++++++--------
 1 files changed, 47 insertions(+), 8 deletions(-)

diff --git a/skjcmanager/skjcmanager-service/skjcmanager-sm/src/main/java/cn/gistack/sm/sjztmd/service/impl/TbAttResRsbSpillwayServiceImpl.java b/skjcmanager/skjcmanager-service/skjcmanager-sm/src/main/java/cn/gistack/sm/sjztmd/service/impl/TbAttResRsbSpillwayServiceImpl.java
index 41091e4..926accf 100644
--- a/skjcmanager/skjcmanager-service/skjcmanager-sm/src/main/java/cn/gistack/sm/sjztmd/service/impl/TbAttResRsbSpillwayServiceImpl.java
+++ b/skjcmanager/skjcmanager-service/skjcmanager-sm/src/main/java/cn/gistack/sm/sjztmd/service/impl/TbAttResRsbSpillwayServiceImpl.java
@@ -2,7 +2,9 @@
 
 import cn.gistack.sm.sjztmd.entity.TbAttResRsbSpillway;
 import cn.gistack.sm.sjztmd.entity.TbAttResRsbSpillway;
+import cn.gistack.sm.sjztmd.entity.TbAttResRsbVerspi;
 import cn.gistack.sm.sjztmd.mapper.TbAttResRsbSpillwayMapper;
+import cn.gistack.sm.sjztmd.service.ITbAttResBaseService;
 import cn.gistack.sm.sjztmd.service.ITbAttResRsbSpillwayService;
 import cn.gistack.system.user.feign.IUserClient;
 import com.baomidou.dynamic.datasource.annotation.DS;
@@ -15,7 +17,9 @@
 import java.text.SimpleDateFormat;
 import java.util.Date;
 import java.util.List;
+import java.util.function.Function;
 import java.util.stream.Collectors;
+import java.util.stream.Stream;
 
 /**
  * @ClassName TbAttResRsbSpillwayServiceImpl
@@ -30,6 +34,9 @@
 
 	@Autowired
 	private IUserClient userClient;
+
+	@Autowired
+	private ITbAttResBaseService tbAttResBaseService;
 
 	/**
 	 * 查询水库填报详情信息
@@ -48,14 +55,25 @@
 	 */
 	@Override
 	public boolean updateByGuid(List<TbAttResRsbSpillway> newList) {
+		boolean edit = false;
+		// 编辑标记
+		if (newList.get(0).getCheckState()==4){
+			edit = true;
+		}
 		// 查询是否存在
 		List<TbAttResRsbSpillway> oldList = baseMapper.getDetailByResGuid(newList.get(0).getResGuid());
 		// 更新日志
 		setUpdateLog(oldList,newList);
 		// 创建或更新数据
 		boolean flag = addOrUpdate(oldList,newList);
+		if (flag){
+//			if (!edit) {
+//				tbAttResBaseService.updateFinalState(newList.get(0).getResGuid(), 0);
+//			}
+			return true;
+		}
 		// 返回
-		return flag;
+		return false;
 	}
 
 	/**
@@ -82,10 +100,12 @@
 		if (addList.size()>0){
 			// 批量新增
 			addFlag = saveBatch(addList);
+			tbAttResBaseService.updateFinalState(addList.get(0).getResGuid(),addList.get(0).getCheckState());
 		}
 		if (updateList.size()>0){
 			// 批量更新
 			addFlag = updateBatchById(updateList);
+			tbAttResBaseService.updateFinalState(updateList.get(0).getResGuid(),updateList.get(0).getCheckState());
 		}
 		if (removeList.size()>0){
 			// 批量删除
@@ -105,16 +125,37 @@
 	 */
 	@Override
 	public boolean auditByGuid(List<TbAttResRsbSpillway> newList) {
+		boolean edit = false;
+		// 编辑标记
+		if (newList.get(0).getCheckState()==4){
+			edit = true;
+		}
 		// 查询是否存在
 		List<TbAttResRsbSpillway> oldList = baseMapper.getDetailByResGuid(newList.get(0).getResGuid());
 		// 更新操作日志
 		setUpdateLog(oldList,newList);
 		// 更新审核日志
 		setUpdateCheckLog(oldList,newList);
+
 		// 更新数据
-		boolean batch = updateBatchById(newList);
-		// 返回
-		return batch;
+		List<TbAttResRsbSpillway> updateList = Stream.concat(oldList.stream(),newList.stream())
+			.collect(Collectors.toMap(TbAttResRsbSpillway::getGuid, Function.identity(),(oldValue, newValue) -> {
+					oldValue.setCheckState(newValue.getCheckState());
+					return oldValue;
+				}
+			))
+			.values()
+			.stream()
+			.collect(Collectors.toList());
+
+		boolean batch = updateBatchById(updateList);
+		if (batch){
+			if (!edit) {
+				tbAttResBaseService.updateFinalState(newList.get(0).getResGuid(), newList.get(0).getCheckState());
+			}
+			return true;
+		}
+		return false;
 	}
 
 	/**
@@ -194,10 +235,8 @@
 			if (newTbAttResRsbSpillway.getCheckState()==4){
 				fillName = "编辑";
 				// 找到相关的 guid 数据
-				for (TbAttResRsbSpillway tbAttResRsbSpillway : oldList) {
-					if (tbAttResRsbSpillway.getGuid().equals(newTbAttResRsbSpillway.getGuid())){
-						newTbAttResRsbSpillway.setCheckState(tbAttResRsbSpillway.getCheckState());
-					}
+				if (oldList.size()>0){
+					newTbAttResRsbSpillway.setCheckState(oldList.get(0).getCheckState());
 				}
 			}
 

--
Gitblit v1.9.3