From c24e182e6ba82ddb167f7c1f640c68aaec0ab0b7 Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Mon, 21 Mar 2022 14:34:23 +0800
Subject: [PATCH] fix
---
src/main/java/cc/mrbird/febs/dapp/service/impl/DappWalletServiceImpl.java | 91 ++++++++++++++++++++++++++++++++++++++++++---
1 files changed, 85 insertions(+), 6 deletions(-)
diff --git a/src/main/java/cc/mrbird/febs/dapp/service/impl/DappWalletServiceImpl.java b/src/main/java/cc/mrbird/febs/dapp/service/impl/DappWalletServiceImpl.java
index 125da6c..c667864 100644
--- a/src/main/java/cc/mrbird/febs/dapp/service/impl/DappWalletServiceImpl.java
+++ b/src/main/java/cc/mrbird/febs/dapp/service/impl/DappWalletServiceImpl.java
@@ -1,17 +1,23 @@
package cc.mrbird.febs.dapp.service.impl;
+import cc.mrbird.febs.common.contants.AppContants;
+import cc.mrbird.febs.common.exception.FebsException;
import cc.mrbird.febs.common.utils.LoginUserUtil;
-import cc.mrbird.febs.dapp.entity.DappMemberEntity;
-import cc.mrbird.febs.dapp.entity.DappWalletCoinEntity;
-import cc.mrbird.febs.dapp.entity.DappWalletMineEntity;
-import cc.mrbird.febs.dapp.mapper.DappMemberDao;
-import cc.mrbird.febs.dapp.mapper.DappWalletCoinDao;
-import cc.mrbird.febs.dapp.mapper.DappWalletMineDao;
+import cc.mrbird.febs.dapp.dto.RecordInPageDto;
+import cc.mrbird.febs.dapp.dto.WalletOperateDto;
+import cc.mrbird.febs.dapp.entity.*;
+import cc.mrbird.febs.dapp.mapper.*;
import cc.mrbird.febs.dapp.service.DappWalletService;
import cc.mrbird.febs.dapp.vo.WalletInfoVo;
+import com.baomidou.mybatisplus.core.metadata.IPage;
+import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
+
+import java.math.BigDecimal;
+import java.util.List;
/**
* @author wzy
@@ -25,6 +31,8 @@
private final DappMemberDao dappMemberDao;
private final DappWalletMineDao dappWalletMineDao;
private final DappWalletCoinDao dappWalletCoinDao;
+ private final DappFundFlowDao dappFundFlowDao;
+ private final DappAccountMoneyChangeDao dappAccountMoneyChangeDao;
@Override
public WalletInfoVo walletInfo() {
@@ -39,4 +47,75 @@
walletInfo.setAvailableWallet(walletCoin.getAvailableAmount());
return walletInfo;
}
+
+ @Override
+ @Transactional(rollbackFor = Exception.class)
+ public void change(WalletOperateDto walletOperateDto) {
+ DappMemberEntity member = LoginUserUtil.getAppUser();
+
+ BigDecimal ethUsdtPrice = new BigDecimal("1234");
+
+ DappWalletMineEntity walletMine = dappWalletMineDao.selectByMemberId(member.getId());
+ if (walletOperateDto.getAmount().compareTo(walletMine.getAvailableAmount()) > 0) {
+ throw new FebsException("可用金额不足");
+ }
+
+ DappFundFlowEntity fund = new DappFundFlowEntity(member.getId(), walletOperateDto.getAmount().negate(), AppContants.MONEY_TYPE_CHANGE, null);
+ dappFundFlowDao.insert(fund);
+
+ BigDecimal preEthAmount = walletMine.getAvailableAmount();
+
+ // TODO 并发加悲观锁
+ // 更新eth金额
+ walletMine.setAvailableAmount(walletMine.getAvailableAmount().subtract(walletOperateDto.getAmount()));
+ walletMine.setTotalAmount(walletMine.getTotalAmount().subtract(walletOperateDto.getAmount()));
+ dappWalletMineDao.updateById(walletMine);
+
+ DappWalletCoinEntity walletCoin = dappWalletCoinDao.selectByMemberId(member.getId());
+ BigDecimal preUsdtAmount = walletCoin.getAvailableAmount();
+
+ // 更新usdt金额
+ BigDecimal usdt = walletOperateDto.getAmount().multiply(ethUsdtPrice);
+ walletCoin.setAvailableAmount(walletCoin.getAvailableAmount().add(usdt));
+ walletCoin.setTotalAmount(walletCoin.getTotalAmount().add(usdt));
+ dappWalletCoinDao.updateById(walletCoin);
+
+ DappAccountMoneyChangeEntity ethChange = new DappAccountMoneyChangeEntity(member.getId(), preEthAmount, walletOperateDto.getAmount(), walletMine.getTotalAmount(), "ETH兑换USDT-ETH", AppContants.MONEY_TYPE_CHANGE);
+ DappAccountMoneyChangeEntity usdtChange = new DappAccountMoneyChangeEntity(member.getId(), preUsdtAmount, usdt, walletCoin.getTotalAmount(), "ETH兑换USDT-USDT", AppContants.MONEY_TYPE_CHANGE);
+ dappAccountMoneyChangeDao.insert(ethChange);
+ dappAccountMoneyChangeDao.insert(usdtChange);
+ }
+
+ @Override
+ @Transactional(rollbackFor = Exception.class)
+ public void withdraw(WalletOperateDto walletOperateDto) {
+ DappMemberEntity member = LoginUserUtil.getAppUser();
+
+ DappWalletCoinEntity walletCoin = dappWalletCoinDao.selectByMemberId(member.getId());
+ if (walletOperateDto.getAmount().compareTo(walletCoin.getAvailableAmount()) > 0) {
+ throw new FebsException("可用金额不足");
+ }
+
+ // TODO 并发加悲观锁
+ BigDecimal preAmount = walletCoin.getAvailableAmount();
+ walletCoin.setAvailableAmount(walletCoin.getAvailableAmount().subtract(walletOperateDto.getAmount()));
+ walletCoin.setFrozenAmount(walletCoin.getFrozenAmount().subtract(walletOperateDto.getAmount()));
+ dappWalletCoinDao.updateById(walletCoin);
+
+ DappFundFlowEntity fund = new DappFundFlowEntity(member.getId(), walletOperateDto.getAmount().negate(), AppContants.MONEY_TYPE_WITHDRAWAL, 1);
+ dappFundFlowDao.insert(fund);
+
+ DappAccountMoneyChangeEntity usdtChange = new DappAccountMoneyChangeEntity(member.getId(), preAmount, walletOperateDto.getAmount(), walletCoin.getAvailableAmount(), "USDT申请提现", AppContants.MONEY_TYPE_WITHDRAWAL);
+ dappAccountMoneyChangeDao.insert(usdtChange);
+ }
+
+ @Override
+ public List<DappFundFlowEntity> recordInPage(RecordInPageDto recordInPageDto) {
+ Page<DappFundFlowEntity> page = new Page<>(recordInPageDto.getPageNum(), recordInPageDto.getPageSize());
+
+ DappFundFlowEntity dappFundFlowEntity = new DappFundFlowEntity();
+ dappFundFlowEntity.setType(recordInPageDto.getType());
+ IPage<DappFundFlowEntity> records = dappFundFlowDao.selectInPage(page, dappFundFlowEntity);
+ return records.getRecords();
+ }
}
--
Gitblit v1.9.1