From f12f168ae5354e573b82969184b7e22181b96d08 Mon Sep 17 00:00:00 2001
From: 钟日健 <5689795+arsn@user.noreply.gitee.com>
Date: Tue, 22 Feb 2022 19:11:06 +0800
Subject: [PATCH] 缴费确认修改

---
 src/main/java/org/springblade/modules/vip/service/impl/UserVipServiceImpl.java |   34 ++++++++++++++++++++++++++--------
 1 files changed, 26 insertions(+), 8 deletions(-)

diff --git a/src/main/java/org/springblade/modules/vip/service/impl/UserVipServiceImpl.java b/src/main/java/org/springblade/modules/vip/service/impl/UserVipServiceImpl.java
index ba1182c..11b9da1 100644
--- a/src/main/java/org/springblade/modules/vip/service/impl/UserVipServiceImpl.java
+++ b/src/main/java/org/springblade/modules/vip/service/impl/UserVipServiceImpl.java
@@ -3,6 +3,8 @@
 
 import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import org.springblade.core.mp.support.Condition;
+import org.springblade.modules.system.entity.User;
 import org.springblade.modules.training.entity.TrainingRegistration;
 import org.springblade.modules.vip.entity.UserVip;
 import org.springblade.modules.vip.mapper.UserVipMapper;
@@ -11,6 +13,7 @@
 import org.springframework.stereotype.Service;
 
 import java.util.Date;
+import java.util.List;
 
 /**
  *  用户会员服务实现类
@@ -41,14 +44,29 @@
 	 */
 	@Override
 	public boolean insertUserVipInfo(TrainingRegistration trainingRegistration) {
-		//1.创建用户会员对象
-		UserVip userVip = new UserVip();
-		//2.数据封装
-		userVip.setApplyId(trainingRegistration.getId());
-		userVip.setCreateTime(new Date());
-		userVip.setUpdateTime(new Date());
-		userVip.setUserId(Long.parseLong(trainingRegistration.getUserId()));
+		//先查询之前是否有缴费,比如缴费了取消报名这种情况
+		UserVip userVip1 = new UserVip();
+		userVip1.setUserId(Long.parseLong(trainingRegistration.getUserId()));
+		List<UserVip> list = this.list(Condition.getQueryWrapper(userVip1));
+		boolean status = false;
+		if (list.size()>0){
+			//更新数据
+			UserVip userVip = list.get(0);
+			userVip.setApplyId(trainingRegistration.getId());
+			userVip.setUpdateTime(new Date());
+			//更新
+			status = this.updateById(userVip);
+		}else {
+			//1.创建用户会员对象
+			UserVip userVip = new UserVip();
+			//2.数据封装
+			userVip.setApplyId(trainingRegistration.getId());
+			userVip.setCreateTime(new Date());
+			userVip.setUpdateTime(new Date());
+			userVip.setUserId(Long.parseLong(trainingRegistration.getUserId()));
+			status = this.save(userVip);
+		}
 		//3.返回
-		return this.save(userVip);
+		return status;
 	}
 }

--
Gitblit v1.9.3