From d4be4cc69f18b01cc39bd3f9dc9497a828848ca8 Mon Sep 17 00:00:00 2001
From: zj <1772600164@qq.com>
Date: Tue, 30 Sep 2025 11:18:44 +0800
Subject: [PATCH] 1

---
 trading-order-admin/src/main/java/com/yami/trading/api/controller/ApiWithdrawController.java |   15 ++++++++++++---
 1 files changed, 12 insertions(+), 3 deletions(-)

diff --git a/trading-order-admin/src/main/java/com/yami/trading/api/controller/ApiWithdrawController.java b/trading-order-admin/src/main/java/com/yami/trading/api/controller/ApiWithdrawController.java
index 9b16eb1..15074a2 100644
--- a/trading-order-admin/src/main/java/com/yami/trading/api/controller/ApiWithdrawController.java
+++ b/trading-order-admin/src/main/java/com/yami/trading/api/controller/ApiWithdrawController.java
@@ -3,6 +3,7 @@
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.yami.trading.bean.model.CapitaltWallet;
 import com.yami.trading.bean.model.User;
+import com.yami.trading.bean.model.Wallet;
 import com.yami.trading.bean.model.Withdraw;
 import com.yami.trading.common.constants.Constants;
 import com.yami.trading.common.domain.Result;
@@ -13,6 +14,7 @@
 import com.yami.trading.security.common.util.SecurityUtils;
 import com.yami.trading.service.CapitaltWalletService;
 import com.yami.trading.service.SessionTokenService;
+import com.yami.trading.service.WalletService;
 import com.yami.trading.service.WithdrawService;
 import com.yami.trading.service.syspara.SysparaService;
 import com.yami.trading.service.user.UserService;
@@ -50,6 +52,8 @@
     protected WalletLogService walletLogService;
     @Autowired
     CapitaltWalletService capitaltWalletService;
+    @Autowired
+    WalletService walletService;
 
     /**
      * 首次进入页面,传递session_token
@@ -111,9 +115,14 @@
 
         // 获取资金账户(capital)
         if(channel.contains("USDT")){
-            CapitaltWallet capitaltWallet = capitaltWalletService.getOne(new LambdaQueryWrapper<>(CapitaltWallet.class)
-                    .eq(CapitaltWallet::getUserId, partyId).last(" limit 1 "));
-            if(capitaltWallet.getMoney().compareTo(new BigDecimal(amount)) < 0){
+//            CapitaltWallet capitaltWallet = capitaltWalletService.getOne(new LambdaQueryWrapper<>(CapitaltWallet.class)
+//                    .eq(CapitaltWallet::getUserId, partyId).last(" limit 1 "));
+//            if(capitaltWallet.getMoney().compareTo(new BigDecimal(amount)) < 0){
+//                throw new YamiShopBindException("Insufficient available balance for withdrawal!");
+//            }
+            Wallet wallet = walletService.getOne(new LambdaQueryWrapper<>(Wallet.class)
+                    .eq(Wallet::getUserId, partyId).last(" limit 1 "));
+            if(wallet.getMoney().compareTo(new BigDecimal(amount)) < 0){
                 throw new YamiShopBindException("Insufficient available balance for withdrawal!");
             }
         }

--
Gitblit v1.9.3