From bf502dd67e829d23001abd8c4d85dcb600f96f4d Mon Sep 17 00:00:00 2001
From: zyy3 <zyy3@zy.com>
Date: Sun, 28 Sep 2025 15:32:58 +0800
Subject: [PATCH] 修改账户资金余额

---
 trading-order-service/src/main/java/com/yami/trading/service/impl/CapitaltWalletServiceImpl.java |   47 +++++++++++++++++++++++++++++++++++++++++++----
 1 files changed, 43 insertions(+), 4 deletions(-)

diff --git a/trading-order-service/src/main/java/com/yami/trading/service/impl/CapitaltWalletServiceImpl.java b/trading-order-service/src/main/java/com/yami/trading/service/impl/CapitaltWalletServiceImpl.java
index 7bfe54f..bd5a61a 100644
--- a/trading-order-service/src/main/java/com/yami/trading/service/impl/CapitaltWalletServiceImpl.java
+++ b/trading-order-service/src/main/java/com/yami/trading/service/impl/CapitaltWalletServiceImpl.java
@@ -4,9 +4,11 @@
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.yami.trading.bean.contract.domain.ContractOrder;
+import com.yami.trading.bean.item.domain.Item;
 import com.yami.trading.bean.model.CapitaltWallet;
 import com.yami.trading.bean.model.MoneyLog;
 import com.yami.trading.bean.model.Wallet;
+import com.yami.trading.bean.model.WalletExtend;
 import com.yami.trading.common.constants.Constants;
 import com.yami.trading.common.domain.Result;
 import com.yami.trading.common.exception.YamiShopBindException;
@@ -16,6 +18,7 @@
 import com.yami.trading.service.MoneyLogService;
 import com.yami.trading.service.WalletService;
 import com.yami.trading.service.contract.ContractOrderService;
+import com.yami.trading.service.item.ItemService;
 import lombok.extern.slf4j.Slf4j;
 import org.checkerframework.checker.units.qual.A;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -43,6 +46,9 @@
 
     @Autowired
     MoneyLogService moneyLogService;
+
+    @Autowired
+    ItemService itemService;
 
     @Override
     public CapitaltWallet getUserIdWallet(String userId) {
@@ -122,14 +128,17 @@
         if (accountType == 1 && moneyRevise.compareTo(BigDecimal.ZERO) <= 0) { //充值
             throw new YamiShopBindException("请输入大于0的数量");
         }
-        if (accountType == 2 && moneyRevise.compareTo(BigDecimal.ZERO) >= 0) {
-            throw new YamiShopBindException("请输入小于0的数量");
+        if (accountType == 2) {
+            moneyRevise = moneyRevise.negate();
         }
         if ("usdt".equals(coinType)) {
             double amount1 = moneyRevise.doubleValue();
             CapitaltWallet capitaltWallet = getOne(new LambdaQueryWrapper<>(CapitaltWallet.class)
                     .eq(CapitaltWallet::getUserId, userId).last(" limit 1 "));
             double amount_before = capitaltWallet.getMoney().doubleValue();
+            if (accountType == 2 && capitaltWallet.getMoney().subtract(moneyRevise).compareTo(BigDecimal.ZERO) < 0) {
+                throw new YamiShopBindException("账号资金不足");
+            }
             update(capitaltWallet, amount1);
 
             // 保存资金日志
@@ -138,14 +147,44 @@
             moneyLog.setAmountBefore(new BigDecimal(amount_before));
             moneyLog.setAmount(new BigDecimal(amount1));
             moneyLog.setAmountAfter(BigDecimal.valueOf(Arith.add(amount_before, amount1)));
-            moneyLog.setLog("后台手动充值");
+            moneyLog.setLog("后台手动充值/扣款");
             moneyLog.setUserId(userId);
             moneyLog.setWalletType(Constants.WALLET);
             moneyLog.setContentType(Constants.MONEYLOG_CONTENT_RECHARGE);
             moneyLog.setCreateTime(new Date());
             moneyLogService.save(moneyLog);
         } else {
-            throw new YamiShopBindException("只支持usdt");
+            Item item = itemService.findBySymbol(coinType);
+            if (item == null) {
+                throw new YamiShopBindException("暂不支持的币种");
+            }
+            WalletExtend walletExtend = new WalletExtend();
+            walletExtend = walletService.saveExtendByPara(userId, coinType);
+
+            double volume = moneyRevise.doubleValue();
+
+            double amount_before = walletExtend.getAmount();
+
+            if (accountType == 2 && Arith.sub(amount_before, volume) < 0) {
+                throw new YamiShopBindException("账号资金不足");
+            }
+
+            // walletExtend = walletService.saveWalletExtendByParaAndUpdate(String.valueOf(recharge.getPartyId()), recharge.getSymbol(), volume);
+            walletService.updateExtend(walletExtend.getPartyId().toString(), walletExtend.getWallettype(), volume);
+
+            // 保存资金日志
+            MoneyLog moneyLog = new MoneyLog();
+            moneyLog.setCategory(Constants.MONEYLOG_CATEGORY_COIN);
+            moneyLog.setAmountBefore(new BigDecimal(amount_before));
+            moneyLog.setAmount(new BigDecimal(volume));
+            moneyLog.setAmountAfter(new BigDecimal(Arith.add(amount_before, volume)));
+
+            moneyLog.setLog("后台手动充值/扣款");
+            moneyLog.setUserId(userId);
+            moneyLog.setWalletType(coinType);
+            moneyLog.setContentType(Constants.MONEYLOG_CONTENT_RECHARGE);
+            moneyLog.setCreateTime(new Date());
+            moneyLogService.save(moneyLog);
         }
         return Result.succeed();
     }

--
Gitblit v1.9.3