From fd13a47afef5654fc831947cfc0507312ac0ed78 Mon Sep 17 00:00:00 2001
From: guoshilong <123456>
Date: Fri, 16 Jun 2023 18:06:05 +0800
Subject: [PATCH] 监测站点修改接口逻辑修改

---
 skjcmanager/skjcmanager-service/skjcmanager-sm/src/main/java/cn/gistack/sm/sjztmd/service/impl/AttStBaseServiceImpl.java |   36 ++++++++++++++++++++++++++++++++++++
 1 files changed, 36 insertions(+), 0 deletions(-)

diff --git a/skjcmanager/skjcmanager-service/skjcmanager-sm/src/main/java/cn/gistack/sm/sjztmd/service/impl/AttStBaseServiceImpl.java b/skjcmanager/skjcmanager-service/skjcmanager-sm/src/main/java/cn/gistack/sm/sjztmd/service/impl/AttStBaseServiceImpl.java
index cc87734..6c2e01b 100644
--- a/skjcmanager/skjcmanager-service/skjcmanager-sm/src/main/java/cn/gistack/sm/sjztmd/service/impl/AttStBaseServiceImpl.java
+++ b/skjcmanager/skjcmanager-service/skjcmanager-sm/src/main/java/cn/gistack/sm/sjztmd/service/impl/AttStBaseServiceImpl.java
@@ -6,10 +6,46 @@
 import com.baomidou.dynamic.datasource.annotation.DS;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import org.springblade.core.mp.base.BaseServiceImpl;
+import org.springblade.core.mp.support.Condition;
 import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
+import org.springblade.core.tool.utils.ObjectUtil;
+
+import java.util.List;
 
 @Service
 @DS("zt")
 public class AttStBaseServiceImpl extends ServiceImpl<AttStBaseMapper, AttStBase> implements IAttStBaseService {
 
+	@Override
+	@Transactional(rollbackFor = Exception.class)
+	public boolean updateById(AttStBase entity) {
+		AttStBase base = new AttStBase();
+		if (ObjectUtil.isNotEmpty(entity.getIsMainRain())&&entity.getIsMainRain().equals(1)){
+			//设置雨情站为主站
+			//获取该水库下的所有雨情站
+			base.setResGuid(entity.getResGuid());
+			base.setIsPptn(1);
+			List<AttStBase> list = list(Condition.getQueryWrapper(base));
+			//设置所有雨量站为辅站
+			list.forEach(e->{
+				e.setIsMainRain(0);
+			});
+			updateBatchById(list);
+			//设置该站为主站
+		}else if (ObjectUtil.isNotEmpty(entity.getIsMainRsvr())&&entity.getIsMainRsvr().equals(1)){
+			//设置水情站为主站
+			//获取该水库下的所有水情站
+			base.setResGuid(entity.getResGuid());
+			base.setIsRsvr(1);
+			List<AttStBase> list = list(Condition.getQueryWrapper(base));
+			//设置所有水情站为辅站
+			list.forEach(e->{
+				e.setIsMainRsvr(0);
+			});
+			updateBatchById(list);
+			//设置该站为主站
+		}
+		return super.updateById(entity);
+	}
 }

--
Gitblit v1.9.3