From af646f794986bdf21553eb920e481fe90ec37319 Mon Sep 17 00:00:00 2001 From: Helius <wangdoubleone@gmail.com> Date: Tue, 02 Aug 2022 01:21:50 +0800 Subject: [PATCH] fix --- src/main/java/cc/mrbird/febs/dapp/service/impl/DappWalletServiceImpl.java | 365 ++++++++++++++++++++++++++++++++++++++++++++++++++- 1 files changed, 352 insertions(+), 13 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..ec08f1c 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,20 +1,41 @@ package cc.mrbird.febs.dapp.service.impl; +import cc.mrbird.febs.common.configure.i18n.MessageSourceUtils; +import cc.mrbird.febs.common.contants.AppContants; +import cc.mrbird.febs.common.entity.QueryRequest; +import cc.mrbird.febs.common.exception.FebsException; +import cc.mrbird.febs.common.utils.FebsUtil; 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.common.utils.RedisUtils; +import cc.mrbird.febs.dapp.chain.ChainEnum; +import cc.mrbird.febs.dapp.chain.ChainService; +import cc.mrbird.febs.dapp.chain.ContractChainService; +import cc.mrbird.febs.dapp.dto.PriceDto; +import cc.mrbird.febs.dapp.dto.RecordInPageDto; +import cc.mrbird.febs.dapp.dto.TransferDto; +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.DappSystemService; import cc.mrbird.febs.dapp.service.DappWalletService; +import cc.mrbird.febs.dapp.utils.BoxUtil; import cc.mrbird.febs.dapp.vo.WalletInfoVo; +import cn.hutool.core.collection.CollUtil; +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.math.RoundingMode; +import java.util.HashMap; +import java.util.List; +import java.util.Map; /** - * @author wzy + * @author * @date 2022-03-18 **/ @Slf4j @@ -25,18 +46,336 @@ private final DappMemberDao dappMemberDao; private final DappWalletMineDao dappWalletMineDao; private final DappWalletCoinDao dappWalletCoinDao; + private final DappFundFlowDao dappFundFlowDao; + private final DappAccountMoneyChangeDao dappAccountMoneyChangeDao; + private final RedisUtils redisUtils; + private final DataDictionaryCustomMapper dataDictionaryCustomMapper; + private final DappSystemService dappSystemService; @Override public WalletInfoVo walletInfo() { DappMemberEntity member = LoginUserUtil.getAppUser(); - DappWalletCoinEntity walletCoin = dappWalletCoinDao.selectByMemberId(member.getId()); - DappWalletMineEntity walletMine = dappWalletMineDao.selectByMemberId(member.getId()); - + Map<String, BigDecimal> map = dappFundFlowDao.selectAmountTotalByType(member.getId()); WalletInfoVo walletInfo = new WalletInfoVo(); - walletInfo.setTotalMine(walletMine.getTotalAmount()); - walletInfo.setAvailableMine(walletMine.getAvailableAmount()); - walletInfo.setAvailableWallet(walletCoin.getAvailableAmount()); + List<DappMemberEntity> direct = dappMemberDao.selectChildMemberDirectOrNot(member.getInviteId(), 1); + List<DappMemberEntity> notDirect = dappMemberDao.selectChildMemberDirectOrNot(member.getInviteId(), 2); + BigDecimal childHoldAmount = dappMemberDao.selectChildHoldAmount(member.getInviteId()); + + DappWalletCoinEntity walletCoin = dappWalletCoinDao.selectByMemberId(member.getId()); + + walletInfo.setTotalChild(notDirect.size()); + walletInfo.setDirectCnt(direct.size()); + walletInfo.setTotalChildCoin(childHoldAmount); + walletInfo.setTeamReward(map.get("teamReward")); + walletInfo.setMiningAmount(map.get("mine")); + walletInfo.setInviteId(member.getInviteId()); + walletInfo.setBoxCnt(walletCoin.getBoxCnt()); return walletInfo; } + + @Override + @Transactional(rollbackFor = Exception.class) + public void change(WalletOperateDto walletOperateDto) { + DappMemberEntity member = LoginUserUtil.getAppUser(); + + BigDecimal ethUsdtPrice = (BigDecimal) redisUtils.get(AppContants.REDIS_KEY_ETH_NEW_PRICE); + + DappWalletMineEntity walletMine = dappWalletMineDao.selectByMemberId(member.getId()); + if (walletOperateDto.getAmount().compareTo(walletMine.getAvailableAmount()) > 0) { + throw new FebsException("可用金额不足"); + } + + DappFundFlowEntity fund = new DappFundFlowEntity(member.getId(), walletOperateDto.getAmount(), AppContants.MONEY_TYPE_CHANGE, null, null); + dappFundFlowDao.insert(fund); + + BigDecimal preEthAmount = walletMine.getAvailableAmount(); + + // TODO 并发加悲观锁 + // 更新eth金额 + walletMine.setAvailableAmount(walletMine.getAvailableAmount().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.getAvailableAmount(), "ETH兑换USDT-ETH, 兑换价格为:" + ethUsdtPrice, AppContants.MONEY_TYPE_CHANGE); + DappAccountMoneyChangeEntity usdtChange = new DappAccountMoneyChangeEntity(member.getId(), preUsdtAmount, usdt, walletCoin.getAvailableAmount(), "ETH兑换USDT-USDT, 兑换价格为:" + ethUsdtPrice, AppContants.MONEY_TYPE_CHANGE); + dappAccountMoneyChangeDao.insert(ethChange); + dappAccountMoneyChangeDao.insert(usdtChange); + } + + @Override + @Transactional(rollbackFor = Exception.class) + public void withdraw(WalletOperateDto walletOperateDto) { + DappMemberEntity member = LoginUserUtil.getAppUser(); + + List<DappFundFlowEntity> fundFlows = dappFundFlowDao.selectListForMemberAndDay(member.getId(), 2); + if (CollUtil.isNotEmpty(fundFlows)) { + throw new FebsException("一天只能提现一次"); + } + + DappWalletCoinEntity walletCoin = dappWalletCoinDao.selectByMemberId(member.getId()); + if (walletOperateDto.getAmount().compareTo(walletCoin.getAvailableAmount()) > 0) { + throw new FebsException("可用金额不足"); + } + + Integer fee = (Integer) redisUtils.get(AppContants.REDIS_KEY_CHANGE_FEE); + + //TODO 并发加悲观锁 + BigDecimal preAmount = walletCoin.getAvailableAmount(); + walletCoin.setAvailableAmount(walletCoin.getAvailableAmount().subtract(walletOperateDto.getAmount())); + walletCoin.setFrozenAmount(walletCoin.getFrozenAmount().add(walletOperateDto.getAmount())); + dappWalletCoinDao.updateById(walletCoin); + + DappFundFlowEntity fund = new DappFundFlowEntity(member.getId(), walletOperateDto.getAmount(), AppContants.MONEY_TYPE_WITHDRAWAL, 1, new BigDecimal(fee)); + 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()); + + DappMemberEntity member = LoginUserUtil.getAppUser(); + DappFundFlowEntity dappFundFlowEntity = new DappFundFlowEntity(); + if (recordInPageDto.getType() != null && recordInPageDto.getType() != 0) { + dappFundFlowEntity.setType(recordInPageDto.getType()); + } + dappFundFlowEntity.setMemberId(member.getId()); + dappFundFlowEntity.setStatus(2); + + IPage<DappFundFlowEntity> records = dappFundFlowDao.selectInPage(page, dappFundFlowEntity); + return records.getRecords(); + } + + @Override + public IPage<DappFundFlowEntity> fundFlowInPage(DappFundFlowEntity dappFundFlowEntity, QueryRequest request) { + Page<DappFundFlowEntity> page = new Page<>(request.getPageNum(), request.getPageSize()); + return dappFundFlowDao.selectInPage(page, dappFundFlowEntity); + } + + @Override + @Transactional(rollbackFor = Exception.class) + public void withdrawAgreeOrNot(Long id, int type) { + DappFundFlowEntity fundFlow = dappFundFlowDao.selectById(id); + if (fundFlow == null) { + throw new FebsException("数据不存在"); + } + + DappWalletCoinEntity walletCoin = dappWalletCoinDao.selectByMemberId(fundFlow.getMemberId()); + if (type == 1) { + walletCoin.setFrozenAmount(walletCoin.getFrozenAmount().subtract(fundFlow.getAmount())); + walletCoin.setTotalAmount(walletCoin.getTotalAmount().subtract(fundFlow.getAmount())); + fundFlow.setStatus(DappFundFlowEntity.WITHDRAW_STATUS_AGREE); + } else if (type == 2) { + BigDecimal preAmount = walletCoin.getAvailableAmount(); + + walletCoin.setFrozenAmount(walletCoin.getFrozenAmount().subtract(fundFlow.getAmount())); + walletCoin.setAvailableAmount(walletCoin.getAvailableAmount().add(fundFlow.getAmount())); + + DappAccountMoneyChangeEntity accountMoneyChange = new DappAccountMoneyChangeEntity(walletCoin.getMemberId(), preAmount, fundFlow.getAmount(), walletCoin.getAvailableAmount(), "提现申请被驳回", 2); + fundFlow.setStatus(DappFundFlowEntity.WITHDRAW_STATUS_DISAGREE); + + dappAccountMoneyChangeDao.insert(accountMoneyChange); + } else { + throw new FebsException("参数错误"); + } + + dappWalletCoinDao.updateById(walletCoin); + dappFundFlowDao.updateById(fundFlow); + } + + @Override + public IPage<DappWalletCoinEntity> walletCoinInPage(DappWalletCoinEntity walletCoin, QueryRequest request) { + Page<DappWalletCoinEntity> page = new Page<>(request.getPageNum(), request.getPageSize()); + return dappWalletCoinDao.selectInPage(walletCoin, page); + } + + @Override + public IPage<DappWalletMineEntity> walletMineInPage(DappWalletMineEntity walletMine, QueryRequest request) { + Page<DappWalletMineEntity> page = new Page<>(request.getPageNum(), request.getPageSize()); + return dappWalletMineDao.selectInPage(walletMine, page); + } + + @Override + public IPage<DappAccountMoneyChangeEntity> accountMoneyChangeInPage(DappAccountMoneyChangeEntity change, QueryRequest request) { + Page<DappAccountMoneyChangeEntity> page = new Page<>(request.getPageNum(), request.getPageSize()); + return dappAccountMoneyChangeDao.selectInPage(change, page); + } + + @Override + public Long transfer(TransferDto transferDto) { + DappMemberEntity member = LoginUserUtil.getAppUser(); + + if (member.getActiveStatus() == 2) { + throw new FebsException("请联系邀请人转币激活"); + } + + // 每日出U剩余量(卖币) + BigDecimal usdtRemain = (BigDecimal) redisUtils.get(AppContants.REDIS_KEY_USDT_OUT_LIMIT_REMAIN); + // 用户24小时可出售量 + BigDecimal saleCoinRemain = (BigDecimal) redisUtils.get(AppContants.REDIS_KEY_COIN_REMAIN + member.getAddress()); + BigDecimal buyCoinRemain = (BigDecimal) redisUtils.get(AppContants.REDIS_KEY_TRANSFER_POOL_VOL_REMAIN); + // 铸池中的币的剩余量 + BigDecimal makeCoinRemain = (BigDecimal) redisUtils.get(AppContants.REDIS_KEY_MAKE_POOL_CNT); + + String hasStart = redisUtils.getString(AppContants.SYSTEM_START_FLAG); + if (transferDto.getId() == null) { + if (DappFundFlowEntity.TYPE_SALE == transferDto.getType()) { + if (!"start".equals(hasStart)) { + throw new FebsException(MessageSourceUtils.getString("transfer_msg_001")); + } + + if (transferDto.getAmount().multiply(transferDto.getPrice()).compareTo(usdtRemain) > 0) { + throw new FebsException(MessageSourceUtils.getString("transfer_msg_002")); + } + + if (transferDto.getAmount().compareTo(saleCoinRemain) > 0) { + throw new FebsException(MessageSourceUtils.getString("transfer_msg_003")); + } + + usdtRemain = usdtRemain.subtract(transferDto.getAmount().multiply(transferDto.getPrice())); + saleCoinRemain = saleCoinRemain.subtract(transferDto.getAmount()); + + // 修改当日U剩余量 + redisUtils.set(AppContants.REDIS_KEY_USDT_OUT_LIMIT_REMAIN, usdtRemain); + // 修改用户24小时可售量 + redisUtils.set(AppContants.REDIS_KEY_COIN_REMAIN + member.getAddress(), saleCoinRemain); + } else if (DappFundFlowEntity.TYPE_BUY == transferDto.getType()) { + if ("start".equals(hasStart)) { + if (transferDto.getAmount().compareTo(buyCoinRemain) > 0) { + throw new FebsException(MessageSourceUtils.getString("transfer_msg_004")); + } + buyCoinRemain = buyCoinRemain.subtract(transferDto.getAmount()); + + // 修改当日可购买量 + redisUtils.set(AppContants.REDIS_KEY_TRANSFER_POOL_VOL_REMAIN, buyCoinRemain); + // 如果系统还没有启动,则判断铸池中的剩余量 + } else { + DataDictionaryCustom dic = dataDictionaryCustomMapper.selectDicDataByTypeAndCode(AppContants.DIC_TYPE_SYSTEM_SETTING, AppContants.DIC_VALUE_MAKER_MIN_LIMIT); + if (transferDto.getAmount().compareTo(new BigDecimal(dic.getValue())) < 0) { + throw new FebsException(MessageSourceUtils.getString("transfer_msg_005")); + } + + if (transferDto.getAmount().compareTo(makeCoinRemain) > 0) { + throw new FebsException(MessageSourceUtils.getString("transfer_msg_006")); + } + makeCoinRemain = makeCoinRemain.subtract(transferDto.getAmount()); + + // 修改铸池量 + redisUtils.set(AppContants.REDIS_KEY_MAKE_POOL_CNT, makeCoinRemain); + } + } + + DappFundFlowEntity fundFlow = new DappFundFlowEntity(member.getId(), transferDto.getAmount(), transferDto.getType(), 1, transferDto.getFee(), transferDto.getTxHash()); + fundFlow.setNewestPrice(transferDto.getPrice()); + dappFundFlowDao.insert(fundFlow); + return fundFlow.getId(); + } + + if ("success".equals(transferDto.getFlag())) { + DappFundFlowEntity flow = dappFundFlowDao.selectById(transferDto.getId()); + + flow.setFromHash(transferDto.getTxHash()); + dappFundFlowDao.updateById(flow); + } else { + DappFundFlowEntity flow = dappFundFlowDao.selectById(transferDto.getId()); + if (flow.getStatus() == 1) { + if (DappFundFlowEntity.TYPE_BUY == flow.getType()) { + if ("start".equals(hasStart)) { + buyCoinRemain = buyCoinRemain.add(flow.getAmount()); + + // 修改当日可购买量 + redisUtils.set(AppContants.REDIS_KEY_TRANSFER_POOL_VOL_REMAIN, buyCoinRemain); + } else { + makeCoinRemain = makeCoinRemain.add(flow.getAmount()); + + // 修改铸池量 + redisUtils.set(AppContants.REDIS_KEY_MAKE_POOL_CNT, makeCoinRemain); + } + } else { + usdtRemain = usdtRemain.add(transferDto.getAmount().multiply(transferDto.getPrice())); + saleCoinRemain = saleCoinRemain.add(transferDto.getAmount()); + + // 修改当日U剩余量 + redisUtils.set(AppContants.REDIS_KEY_USDT_OUT_LIMIT_REMAIN, usdtRemain); + // 修改用户24小时可售量 + redisUtils.set(AppContants.REDIS_KEY_COIN_REMAIN + member.getAddress(), saleCoinRemain); + } + dappFundFlowDao.deleteById(transferDto.getId()); + } + } + return null; + } + + @Override + public Map<String, BigDecimal> calPrice(PriceDto priceDto) { + DappMemberEntity member = LoginUserUtil.getAppUser(); + + if (!dappSystemService.systemHasStart()) { + HashMap<String, BigDecimal> map = new HashMap<>(); + map.put("x", new BigDecimal("0.05")); + map.put("y", new BigDecimal("0.05")); + return map; + } + + ContractChainService tfcInstance = ChainService.getInstance(ChainEnum.BSC_TFC.name()); + // u剩余数量 + BigDecimal sourceU = ChainService.getInstance(ChainEnum.BSC_USDT.name()).balanceOf(ChainEnum.BSC_USDT_SOURCE.getAddress()); + // 源池代币剩余数量 + BigDecimal sourceCoin = tfcInstance.balanceOf(ChainEnum.BSC_USDT_SOURCE.getAddress()); + // 用户卖出数量 + BigDecimal coin = priceDto.getAmount(); + BigDecimal x = sourceU.divide(sourceCoin, tfcInstance.decimals(), RoundingMode.HALF_UP); + BigDecimal y = sourceU.divide(sourceCoin.add(coin), tfcInstance.decimals(), RoundingMode.HALF_UP); + + log.info("购买价格:{}, 出卖价格:{}", x, y); + HashMap<String, BigDecimal> map = new HashMap<>(); + map.put("x", x); + map.put("y", y); + return map; + } + + @Override + public int boxSurprise() { + DappMemberEntity member = LoginUserUtil.getAppUser(); + + DappWalletCoinEntity walletCoin = dappWalletCoinDao.selectByMemberId(member.getId()); + if (walletCoin.getBoxCnt() < 1) { + throw new FebsException("盲盒数量不足"); + } + + walletCoin.setBoxCnt(walletCoin.getBoxCnt() - 1); + dappWalletCoinDao.updateById(walletCoin); + + BoxUtil.Box box = BoxUtil.openBox(); + + DappFundFlowEntity boxFundFlow = new DappFundFlowEntity(member.getId(), new BigDecimal(box.getIndex()), 6, 2, BigDecimal.ZERO); + dappFundFlowDao.insert(boxFundFlow); + + new Thread(() -> { + try { + int count = box.getIndex(); + while (count > 0) { + ChainService.getInstance(ChainEnum.BSC_NFT_SDC.name()).safeMintNFT(member.getAddress()); + Thread.sleep(5000); + count--; + } + } catch (Exception e) { + e.printStackTrace(); + log.error("发放卡牌错误", e); + } + }).start(); + + return box.getIndex(); + } } -- Gitblit v1.9.1