From 86eb9c89e5304b356b0f799829180862e9a21934 Mon Sep 17 00:00:00 2001
From: xieb <vip_xiaobin810@163.com>
Date: Mon, 11 Mar 2024 15:31:49 +0800
Subject: [PATCH] 水库白蚁普查增加年度

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

diff --git a/skjcmanager/skjcmanager-service/skjcmanager-sm/src/main/java/cn/gistack/sm/sjztmd/service/impl/TbResGeneralInvestigationServiceImpl.java b/skjcmanager/skjcmanager-service/skjcmanager-sm/src/main/java/cn/gistack/sm/sjztmd/service/impl/TbResGeneralInvestigationServiceImpl.java
index ea8829e..2ee9243 100644
--- a/skjcmanager/skjcmanager-service/skjcmanager-sm/src/main/java/cn/gistack/sm/sjztmd/service/impl/TbResGeneralInvestigationServiceImpl.java
+++ b/skjcmanager/skjcmanager-service/skjcmanager-sm/src/main/java/cn/gistack/sm/sjztmd/service/impl/TbResGeneralInvestigationServiceImpl.java
@@ -1,9 +1,12 @@
 package cn.gistack.sm.sjztmd.service.impl;
 
 import cn.gistack.sm.sjztmd.entity.TbResGeneralInvestigation;
+import cn.gistack.sm.sjztmd.entity.TbResGeneralInvestigationState;
 import cn.gistack.sm.sjztmd.mapper.TbResGeneralInvestigationMapper;
 import cn.gistack.sm.sjztmd.service.ITbResGeneralInvestigationService;
+import cn.gistack.sm.sjztmd.service.ITbResGeneralInvestigationStateService;
 import com.baomidou.dynamic.datasource.annotation.DS;
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
@@ -19,4 +22,29 @@
 @Transactional(rollbackFor = Exception.class)
 public class TbResGeneralInvestigationServiceImpl extends ServiceImpl<TbResGeneralInvestigationMapper, TbResGeneralInvestigation> implements ITbResGeneralInvestigationService {
 
+	private ITbResGeneralInvestigationStateService tbResGeneralInvestigationStateService;
+
+	@Override
+	@Transactional
+	public boolean saveOrUpdateBy(TbResGeneralInvestigation entity) {
+
+		QueryWrapper queryWrapper = new QueryWrapper();
+		queryWrapper.eq("res_guid",entity.getResGuid());
+		queryWrapper.eq("tb_year",entity.getTbYear());
+		queryWrapper.eq("tb_quarter",entity.getTbYear());
+		TbResGeneralInvestigationState state = tbResGeneralInvestigationStateService.getOne(queryWrapper);
+		if (null == state) {
+			state = new TbResGeneralInvestigationState();
+			state.setResGuid(entity.getResGuid());
+			state.setTbQuarter(entity.getTbQuarter());
+			state.setTbYear(entity.getTbYear());
+		}
+
+		// 是否普查完成 1是 0否
+		state.setCheckState(entity.getIsInvCom());
+
+		tbResGeneralInvestigationStateService.saveOrUpdate(state);
+
+		return this.saveOrUpdate(entity);
+	}
 }

--
Gitblit v1.9.3