From fa5e63f49f1e5c891312d9cd659b91c09107ded5 Mon Sep 17 00:00:00 2001
From: KKSU <15274802129@163.com>
Date: Sat, 06 Jul 2024 13:06:20 +0800
Subject: [PATCH] 逻辑

---
 src/main/java/cc/mrbird/febs/dapp/controller/MemberWalletController.java |   12 +++++++++++-
 1 files changed, 11 insertions(+), 1 deletions(-)

diff --git a/src/main/java/cc/mrbird/febs/dapp/controller/MemberWalletController.java b/src/main/java/cc/mrbird/febs/dapp/controller/MemberWalletController.java
index c5527f4..2aaeb21 100644
--- a/src/main/java/cc/mrbird/febs/dapp/controller/MemberWalletController.java
+++ b/src/main/java/cc/mrbird/febs/dapp/controller/MemberWalletController.java
@@ -3,9 +3,11 @@
 import cc.mrbird.febs.common.controller.BaseController;
 import cc.mrbird.febs.common.entity.FebsResponse;
 import cc.mrbird.febs.common.entity.QueryRequest;
+import cc.mrbird.febs.common.utils.FebsUtil;
 import cc.mrbird.febs.dapp.entity.DappWalletCoinEntity;
 import cc.mrbird.febs.dapp.entity.DappWalletMineEntity;
 import cc.mrbird.febs.dapp.service.DappWalletService;
+import cc.mrbird.febs.system.entity.User;
 import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.validation.annotation.Validated;
@@ -13,7 +15,7 @@
 import org.springframework.web.bind.annotation.RestController;
 
 /**
- * @author wzy
+ * @author 
  * @date 2022-03-22
  **/
 @Slf4j
@@ -27,11 +29,19 @@
 
     @RequestMapping(value  = "/walletCoin")
     public FebsResponse walletCoin(DappWalletCoinEntity walletCoin, QueryRequest request) {
+        User currentUser = FebsUtil.getCurrentUser();
+        if (currentUser.getDeptId() == null) {
+            walletCoin.setCurrentUser(currentUser.getUserId());
+        }
         return new FebsResponse().success().data(getDataTable(dappWalletService.walletCoinInPage(walletCoin, request)));
     }
 
     @RequestMapping(value  = "/walletMine")
     public FebsResponse walletMine(DappWalletMineEntity walletMine, QueryRequest request) {
+        User currentUser = FebsUtil.getCurrentUser();
+        if (currentUser.getDeptId() == null) {
+            walletMine.setCurrentUser(currentUser.getUserId());
+        }
         return new FebsResponse().success().data(getDataTable(dappWalletService.walletMineInPage(walletMine, request)));
     }
 }

--
Gitblit v1.9.1