From 87f979fb201a82ebad5926735ed6dfa75ca004d3 Mon Sep 17 00:00:00 2001
From: liujie <1793218484@qq.com>
Date: 星期一, 21 七月 2025 09:21:42 +0800
Subject: [PATCH] 修改bug

---
 ruoyi-applet/src/main/java/com/ruoyi/web/controller/system/UserAccountController.java |   19 ++++++++++++++++++-
 1 files changed, 18 insertions(+), 1 deletions(-)

diff --git a/ruoyi-applet/src/main/java/com/ruoyi/web/controller/system/UserAccountController.java b/ruoyi-applet/src/main/java/com/ruoyi/web/controller/system/UserAccountController.java
index ba358e1..19ddb64 100644
--- a/ruoyi-applet/src/main/java/com/ruoyi/web/controller/system/UserAccountController.java
+++ b/ruoyi-applet/src/main/java/com/ruoyi/web/controller/system/UserAccountController.java
@@ -3,6 +3,7 @@
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
+import com.ruoyi.common.constant.HttpStatus;
 import com.ruoyi.common.core.domain.R;
 import com.ruoyi.common.core.domain.model.LoginUser;
 import com.ruoyi.common.utils.StringUtils;
@@ -55,6 +56,10 @@
     public R<UserAccountVo> getUserAccount() {
         UserAccountVo userAccountVo = new UserAccountVo();
         LoginUser loginUser = tokenService.getLoginUser();
+        if(loginUser==null){
+            return R.fail(HttpStatus.UNAUTHORIZED,"请重新登录");
+        }
+
         TbUser user = userService.getById(loginUser.getUserId());
         userAccountVo.setBalance(user.getBalance());
 
@@ -79,6 +84,9 @@
     @GetMapping("/getUserAccountDetail")
     public R<Page<TbAccountDetail>> getUserAccountDetail(UserAccountDetailQuery query) {
         LoginUser loginUser = tokenService.getLoginUser();
+        if(loginUser==null){
+            return R.fail(HttpStatus.UNAUTHORIZED,"请重新登录");
+        }
         LambdaQueryWrapper<TbAccountDetail> wrapper = new LambdaQueryWrapper<>();
         if(StringUtils.isNotEmpty(query.getTime())){
             wrapper.likeRight(TbAccountDetail::getCreateTime,query.getTime());
@@ -95,9 +103,12 @@
     @Transactional(propagation = Propagation.REQUIRED, rollbackFor = Exception.class)
     public R<?> withdrawal(@RequestBody @Valid UserWithdrawalDto dto) {
         if(dto.getAmount()<=0){
-            return R.fail("提现金额不能小于0");
+            return R.fail("提现金额不能小于等于0");
         }
         LoginUser loginUser = tokenService.getLoginUser();
+        if(loginUser==null){
+            return R.fail(HttpStatus.UNAUTHORIZED,"请重新登录");
+        }
         TbUser user = userService.getById(loginUser.getUserId());
         // 判断是否有卡
         List<TbBank> list = bankService.list(new LambdaQueryWrapper<TbBank>().eq(TbBank::getUserId, loginUser.getUserId()).eq(TbBank::getIsDelete, 0));
@@ -136,6 +147,9 @@
     @GetMapping("/getUserBank")
     public R<TbBank> getUserBank() {
         LoginUser loginUser = tokenService.getLoginUser();
+        if(loginUser==null){
+            return R.fail(HttpStatus.UNAUTHORIZED,"请重新登录");
+        }
         TbBank tbBank = bankService.getOne(new LambdaQueryWrapper<TbBank>().eq(TbBank::getUserId, loginUser.getUserId()).eq(TbBank::getIsDelete, 0));
         return R.ok(tbBank);
     }
@@ -145,6 +159,9 @@
     @PostMapping("/saveUserBank")
     public R<?> getUserBank(@Valid @RequestBody SaveUserBankDto dto) {
         LoginUser loginUser = tokenService.getLoginUser();
+        if(loginUser==null){
+            return R.fail(HttpStatus.UNAUTHORIZED,"请重新登录");
+        }
         TbBank tbBank = bankService.getOne(new LambdaQueryWrapper<TbBank>().eq(TbBank::getUserId, loginUser.getUserId()).eq(TbBank::getIsDelete, 0));
         if(tbBank==null){
             tbBank = new TbBank();

--
Gitblit v1.7.1