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