新版仿ok交易所-后端
zyy3
2025-09-26 a2ca1f655c816f801f8881136d17d19da915d99b
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
package com.yami.trading.service.impl;
 
import cn.hutool.core.collection.CollectionUtil;
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.model.CapitaltWallet;
import com.yami.trading.bean.model.MoneyLog;
import com.yami.trading.bean.model.Wallet;
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 lombok.extern.slf4j.Slf4j;
import org.checkerframework.checker.units.qual.A;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
 
import java.math.BigDecimal;
import java.util.Date;
import java.util.List;
 
/**
 * @program: trading-order-master
 * @description: 合约账户
 * @create: 2025-01-08 17:59
 **/
@Service
@Slf4j
public class CapitaltWalletServiceImpl extends ServiceImpl<CapitaltWalletMapper, CapitaltWallet> implements CapitaltWalletService {
 
    @Autowired
    WalletService walletService;
 
    @Autowired
    private ContractOrderService contractOrderService;
 
    @Autowired
    MoneyLogService moneyLogService;
 
    @Override
    public CapitaltWallet getUserIdWallet(String userId) {
        return getOne(new LambdaQueryWrapper<>(CapitaltWallet.class)
                .eq(CapitaltWallet::getUserId,userId).last(" limit 1"));
    }
 
    /**
     *capital  资金
     *contract 合约
     * @param userId 用户id
     * @param deductAccount 划转账户
     * @param receiveAccount  接收账户
     * @param moneyRevise   划转金额
     * @return
     */
    @Override
    @Transactional
    public Result updateWallt(String userId, String deductAccount, String receiveAccount, BigDecimal moneyRevise) {
        // 获取合约账户(contract)
        Wallet wallet = walletService.saveWalletByPartyId(userId);
 
        // 获取资金账户(capital)
        CapitaltWallet capitaltWallet = getOne(new LambdaQueryWrapper<>(CapitaltWallet.class)
                .eq(CapitaltWallet::getUserId, userId).last(" limit 1 "));
 
        // 判断划转账户类型
        if (deductAccount.equals("capital") && receiveAccount.equals("contract")) {
            // 从资金账户(capital)到合约账户(contract)的划转
            if (capitaltWallet != null && capitaltWallet.getMoney().compareTo(moneyRevise) >= 0) {
                // 执行从资金账户到合约账户的划转操作
                capitaltWallet.setMoney(capitaltWallet.getMoney().subtract(moneyRevise));  // 减少资金账户余额
                //wallet.setMoney(wallet.getMoney().add(moneyRevise));  // 增加合约账户余额
                // 更新账户余额
                //walletService.updateById(wallet);  // 保存合约账户的更新
                walletService.updateToBeCovered(wallet, moneyRevise, 1);
                updateById(capitaltWallet);  // 保存资金账户的更新
                return Result.succeed();
            } else {
                throw new YamiShopBindException("Insufficient balance in the fund account");
            }
        } else if (deductAccount.equals("contract") && receiveAccount.equals("capital")) {
 
            LambdaQueryWrapper<ContractOrder> wrapper = new LambdaQueryWrapper<ContractOrder>();
            wrapper.eq(ContractOrder::getPartyId, userId);
            wrapper.eq(ContractOrder::getLocationType, 1);
            wrapper.eq(ContractOrder::getState,"submitted");
            List<ContractOrder> list = contractOrderService.list(wrapper);
            if(CollectionUtil.isNotEmpty(list)){
                throw new YamiShopBindException("Currently holding the full position. Transfer is not supported!");
            }
 
            // 从合约账户(contract)到资金账户(capital)的划转
            if (wallet != null && wallet.getMoney().compareTo(moneyRevise) >= 0) {
                // 执行从合约账户到资金账户的划转操作
                wallet.setMoney(wallet.getMoney().subtract(moneyRevise));  // 减少合约账户余额
                capitaltWallet.setMoney(capitaltWallet.getMoney().add(moneyRevise));  // 增加资金账户余额
                // 更新账户余额
                walletService.updateById(wallet);  // 保存合约账户的更新
                updateById(capitaltWallet);  // 保存资金账户的更新
                return Result.succeed();
            } else {
                throw new YamiShopBindException("Insufficient balance in the contract account");
            }
        } else {
            // 如果划转账户和接收账户不符合预期,返回错误信息
            throw new YamiShopBindException("Unsupported account transfer types");
        }
    }
 
    /**
     * @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.compareTo(BigDecimal.ZERO) >= 0) {
            throw new YamiShopBindException("请输入小于0的数量");
        }
        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();
            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 {
            throw new YamiShopBindException("只支持usdt");
        }
        return Result.succeed();
    }
 
    @Override
    public void update(CapitaltWallet capitaltWallet, double amount1) {
        capitaltWallet.setMoney(new BigDecimal(Arith.add(capitaltWallet.getMoney().doubleValue(), amount1)));
        if (!updateById(capitaltWallet)) {
            throw new YamiShopBindException("操作钱包失败!");
        }
    }
}