From d06466e5120f8a99323ec2f0a5c9c7bbba70c8ba Mon Sep 17 00:00:00 2001
From: phpcjl <phpcjl@gmail.com>
Date: 星期四, 05 十二月 2024 09:35:37 +0800
Subject: [PATCH] 1.修复绑定门店bug

---
 ruoyi-service/ruoyi-account/src/main/java/com/ruoyi/account/controller/AppUserBankController.java |   18 ++++++++++++++----
 1 files changed, 14 insertions(+), 4 deletions(-)

diff --git a/ruoyi-service/ruoyi-account/src/main/java/com/ruoyi/account/controller/AppUserBankController.java b/ruoyi-service/ruoyi-account/src/main/java/com/ruoyi/account/controller/AppUserBankController.java
index 1fb8829..0af6f92 100644
--- a/ruoyi-service/ruoyi-account/src/main/java/com/ruoyi/account/controller/AppUserBankController.java
+++ b/ruoyi-service/ruoyi-account/src/main/java/com/ruoyi/account/controller/AppUserBankController.java
@@ -1,5 +1,6 @@
 package com.ruoyi.account.controller;
 
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.ruoyi.account.api.model.AppUserBank;
 import com.ruoyi.account.service.AppUserBankService;
 import com.ruoyi.common.core.utils.bean.BeanUtils;
@@ -34,8 +35,16 @@
 	@ApiOperation(value = "保存银行卡")
 	public AjaxResult saveAppUserBank(@RequestBody AppUserBank appUserBank){
 		LoginUser loginUserApplet = tokenService.getLoginUserApplet();
-		appUserBank.setAppUserId(loginUserApplet.getUserid());
-		appUserBankService.saveOrUpdate(appUserBank);
+
+		AppUserBank bank = appUserBankService.getOne(new LambdaQueryWrapper<AppUserBank>()
+				.eq(AppUserBank::getAppUserId, loginUserApplet.getUserid()));
+		if (bank == null){
+			appUserBank.setAppUserId(loginUserApplet.getUserid());
+			appUserBankService.saveOrUpdate(appUserBank);
+		}else {
+			appUserBank.setId(bank.getId());
+			appUserBankService.updateById(appUserBank);
+		}
 		return AjaxResult.success();
 	}
 	
@@ -45,8 +54,9 @@
 	@ApiOperation(value = "获取银行卡信息")
 	public AjaxResult<AppUserBank> getAppUserBank(){
 		LoginUser loginUserApplet = tokenService.getLoginUserApplet();
-		AppUserBank bank = appUserBankService.getById(loginUserApplet.getUserid());
-		return AjaxResult.success(bank);
+		AppUserBank appUserBank = appUserBankService.getOne(new LambdaQueryWrapper<AppUserBank>()
+				.eq(AppUserBank::getAppUserId, loginUserApplet.getUserid()));
+		return AjaxResult.success(appUserBank);
 	}
 	
 }

--
Gitblit v1.7.1