From fa1b970c8fa7772ce123ab84e8cc4240065bcd26 Mon Sep 17 00:00:00 2001
From: zyy <zyy@email.com>
Date: Fri, 06 Feb 2026 17:43:42 +0800
Subject: [PATCH] 提现

---
 trading-order-service/src/main/java/com/yami/trading/service/impl/CapitaltWalletServiceImpl.java |  103 ++++++++++++++++++++++++++++++++++++++++++++++++++-
 1 files changed, 100 insertions(+), 3 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 712864a..5539802 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,15 +4,21 @@
 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;
 import com.yami.trading.common.util.Arith;
 import com.yami.trading.dao.CapitaltWalletMapper;
 import com.yami.trading.service.CapitaltWalletService;
+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;
@@ -20,6 +26,7 @@
 import org.springframework.transaction.annotation.Transactional;
 
 import java.math.BigDecimal;
+import java.util.Date;
 import java.util.List;
 
 /**
@@ -37,10 +44,27 @@
     @Autowired
     private ContractOrderService contractOrderService;
 
+    @Autowired
+    MoneyLogService moneyLogService;
+
+    @Autowired
+    ItemService itemService;
+
+    @Autowired
+    CapitaltWalletMapper capitaltWalletMapper;
+
     @Override
     public CapitaltWallet getUserIdWallet(String userId) {
-        return getOne(new LambdaQueryWrapper<>(CapitaltWallet.class)
+        CapitaltWallet capitaltWallet = getOne(new LambdaQueryWrapper<>(CapitaltWallet.class)
                 .eq(CapitaltWallet::getUserId,userId).last(" limit 1"));
+        if (capitaltWallet != null) {
+            return capitaltWallet;
+        } else {
+            capitaltWallet = new CapitaltWallet();
+            capitaltWallet.setUserId(userId);
+            save(capitaltWallet);
+            return capitaltWallet;
+        }
     }
 
     /**
@@ -68,9 +92,10 @@
             if (capitaltWallet != null && capitaltWallet.getMoney().compareTo(moneyRevise) >= 0) {
                 // 执行从资金账户到合约账户的划转操作
                 capitaltWallet.setMoney(capitaltWallet.getMoney().subtract(moneyRevise));  // 减少资金账户余额
-                wallet.setMoney(wallet.getMoney().add(moneyRevise));  // 增加合约账户余额
+                //wallet.setMoney(wallet.getMoney().add(moneyRevise));  // 增加合约账户余额
                 // 更新账户余额
-                walletService.updateById(wallet);  // 保存合约账户的更新
+                //walletService.updateById(wallet);  // 保存合约账户的更新
+                walletService.updateToBeCovered(wallet, moneyRevise, 1);
                 updateById(capitaltWallet);  // 保存资金账户的更新
                 return Result.succeed();
             } else {
@@ -105,6 +130,78 @@
         }
     }
 
+    /**
+     * @return
+     */
+    @Override
+    @Transactional
+    public Result updateCapitaltWallt(String userId, BigDecimal moneyRevise,  int accountType, String coinType) {
+        if (accountType == 1 && 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 = getUserIdWallet(userId);
+            if (capitaltWallet == null) {
+                throw new YamiShopBindException("账号缺少资金数据");
+            }
+            double amount_before = capitaltWallet.getMoney().doubleValue();
+            if (accountType == 2 && capitaltWallet.getMoney().subtract(moneyRevise).compareTo(BigDecimal.ZERO) < 0) {
+                throw new YamiShopBindException("账号资金不足");
+            }
+            update(capitaltWallet, amount1);
+
+            // 保存资金日志
+            MoneyLog moneyLog = new MoneyLog();
+            moneyLog.setCategory(Constants.MONEYLOG_CATEGORY_COIN);
+            moneyLog.setAmountBefore(new BigDecimal(amount_before));
+            moneyLog.setAmount(new BigDecimal(amount1));
+            moneyLog.setAmountAfter(BigDecimal.valueOf(Arith.add(amount_before, amount1)));
+            moneyLog.setLog("后台手动充值/扣款");
+            moneyLog.setUserId(userId);
+            moneyLog.setWalletType(Constants.WALLET);
+            moneyLog.setContentType(Constants.MONEYLOG_CONTENT_RECHARGE);
+            moneyLog.setCreateTime(new Date());
+            moneyLogService.save(moneyLog);
+        } else {
+            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();
+    }
+
     @Override
     public void update(CapitaltWallet capitaltWallet, double amount1) {
         capitaltWallet.setMoney(new BigDecimal(Arith.add(capitaltWallet.getMoney().doubleValue(), amount1)));

--
Gitblit v1.9.3