From 3cd5a88ce846894d82f83daf967b424d0bf841db Mon Sep 17 00:00:00 2001
From: zyy <zyy@email.com>
Date: Mon, 20 Oct 2025 16:05:50 +0800
Subject: [PATCH] C2C

---
 trading-order-service/src/main/java/com/yami/trading/service/impl/CapitaltWalletServiceImpl.java |   15 ++++++++++++---
 1 files changed, 12 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..fdccbd0 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
@@ -39,8 +39,16 @@
 
     @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 +76,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 {

--
Gitblit v1.9.3