From a64d6284082665f9a27add91edc0f7dc7e31afb7 Mon Sep 17 00:00:00 2001 From: xiaoyong931011 <15274802129@163.com> Date: Tue, 08 Nov 2022 11:23:00 +0800 Subject: [PATCH] 20221021 --- src/main/java/cc/mrbird/febs/dapp/service/impl/DappWalletServiceImpl.java | 357 +++++++++++++++++++++++------------------------------------ 1 files changed, 141 insertions(+), 216 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 0c487d5..cb9e924 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 @@ -22,6 +22,7 @@ import cn.hutool.core.date.DateField; import cn.hutool.core.date.DateUnit; import cn.hutool.core.date.DateUtil; +import com.alibaba.fastjson.JSONObject; import com.baomidou.mybatisplus.core.conditions.Wrapper; import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper; @@ -55,6 +56,7 @@ private final DappNftActivationDao dappNftActivationDao; private final ChainProducer chainProducer; + private final DappSystemDao dappSystemDao; @Override public WalletInfoVo walletInfo() { @@ -62,83 +64,20 @@ DappMemberEntity memberInfo = dappMemberDao.selectById(member.getId()); WalletInfoVo walletInfo = new WalletInfoVo(); - List<DappMemberEntity> direct = dappMemberDao.selectChildMemberDirectOrNot(member.getInviteId(), 1); + List<DappMemberEntity> direct = dappMemberDao.selectChildMemberDirectOrNot(member.getInviteId(), 1, null); DappWalletCoinEntity walletCoin = dappWalletCoinDao.selectByMemberId(member.getId()); + DappWalletMineEntity walletMine = dappWalletMineDao.selectByMemberId(member.getId()); + DappMineDataEntity mineData = dappSystemDao.selectMineDataForOne(); walletInfo.setDirectCnt(direct.size()); - walletInfo.setInviteId(member.getInviteId()); + walletInfo.setInviteId(member.getActiveStatus() == 1 ? member.getInviteId() : "-"); walletInfo.setBalance(walletCoin.getAvailableAmount()); walletInfo.setHasBuy(memberInfo.getActiveStatus()); - walletInfo.setOutCnt(1); - walletInfo.setProfit(BigDecimal.TEN); + walletInfo.setOutCnt(memberInfo.getOutCnt()); + walletInfo.setProfit(dappFundFlowDao.selectProfitAmountByMemberId(member.getId())); + walletInfo.setTfcBalance(walletMine.getAvailableAmount()); + walletInfo.setSafePool(mineData.getSafePool()); 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 @@ -164,37 +103,6 @@ } @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); @@ -216,9 +124,16 @@ public Long transfer(TransferDto transferDto) { DappMemberEntity member = LoginUserUtil.getAppUser(); - member = dappMemberDao.selectById(member.getId()); - if (member.getActiveStatus() == 1) { - throw new FebsException("Do not repeat purchase"); + String upgrade = redisUtils.getString("APP_UPGRADE"); + if ("upgrade".equals(upgrade)) { + throw new FebsException("功能升级中"); + } + + if (transferDto.getType() != 2) { + member = dappMemberDao.selectById(member.getId()); + if (member.getActiveStatus() == 1) { + throw new FebsException("Do not repeat purchase"); + } } if (transferDto.getBuyType() == 1) { @@ -227,15 +142,20 @@ throw new FebsException("Balance Not Enough"); } - updateWalletCoinWithLock(transferDto.getAmount().negate(), member.getId()); + updateWalletCoinWithLock(transferDto.getAmount(), member.getId(), 2); DappFundFlowEntity fundFlow = new DappFundFlowEntity(member.getId(), transferDto.getAmount().negate(), 1, 2, transferDto.getFee(), transferDto.getTxHash()); dappFundFlowDao.insert(fundFlow); chainProducer.sendAchieveTreeMsg(member.getId()); } else { + int type = 1; + // 1-认购 2-充值tfc + if (transferDto.getType() == 2) { + type = 6; + } if (transferDto.getId() == null) { - DappFundFlowEntity fundFlow = new DappFundFlowEntity(member.getId(), transferDto.getAmount(), 1, 1, transferDto.getFee(), transferDto.getTxHash()); + DappFundFlowEntity fundFlow = new DappFundFlowEntity(member.getId(), transferDto.getAmount(), type, 1, transferDto.getFee(), transferDto.getTxHash()); dappFundFlowDao.insert(fundFlow); return fundFlow.getId(); } @@ -255,139 +175,69 @@ } @Override - public Map<String, BigDecimal> calPrice(PriceDto priceDto) { - return null; - } + public BigDecimal calPrice(PriceDto priceDto) { + String priceStr = redisUtils.getString(AppContants.REDIS_KEY_TFC_NEW_PRICE); - @Override - public ActiveNftListVo boxSurprise() { - DappMemberEntity member = LoginUserUtil.getAppUser(); - - DappWalletCoinEntity walletCoin = dappWalletCoinDao.selectByMemberId(member.getId()); - if (walletCoin.getBoxCnt() < 1) { - throw new FebsException(MessageSourceUtils.getString("box_surprise_001")); + BigDecimal amount = priceDto.getAmount(); + if (priceDto.getAmount() == null) { + amount = BigDecimal.ZERO; } - - 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(); - - Date time = new Date(); - Date expire = DateUtil.offset(time, DateField.HOUR, 2); - DappNftActivation nftActivation = new DappNftActivation(); - nftActivation.setMemberId(member.getId()); - nftActivation.setCount(box.getIndex()); - nftActivation.setOpenTime(time); - nftActivation.setExpireTime(expire); - nftActivation.setStatus(1); - dappNftActivationDao.insert(nftActivation); - - ActiveNftListVo nft = new ActiveNftListVo(); - nft.setCount(box.getIndex()); - nft.setId(nftActivation.getId()); - nft.setRemain(DateUtil.between(time, expire, DateUnit.SECOND)); - return nft; + return amount.multiply(new BigDecimal("0.1")).divide(new BigDecimal(priceStr), 2, RoundingMode.HALF_UP); } @Override @Transactional(rollbackFor = Exception.class) - public synchronized void activeNft(ActiveDto activeDto) { + public void withdraw(WithdrawDto withdrawDto) { DappMemberEntity member = LoginUserUtil.getAppUser(); - DappNftActivation nftActive = dappNftActivationDao.selectById(activeDto.getId()); - if (nftActive == null) { - throw new FebsException(MessageSourceUtils.getString("nft_active_003")); + DappWalletCoinEntity walletCoin = dappWalletCoinDao.selectByMemberId(member.getId()); + if (walletCoin.getAvailableAmount().compareTo(withdrawDto.getAmount()) < 0) { + throw new FebsException("Balance Not Enough"); } - if (nftActive.getCount() < activeDto.getCount()) { - throw new FebsException(MessageSourceUtils.getString("nft_active_004")); + DappWalletMineEntity walletMine = dappWalletMineDao.selectByMemberId(member.getId()); + if (walletMine.getAvailableAmount().compareTo(withdrawDto.getFee()) < 0) { + throw new FebsException("TFC Not Enough"); } - if (DateUtil.between(new Date(), nftActive.getExpireTime(), DateUnit.SECOND) < 0) { - throw new FebsException(MessageSourceUtils.getString("nft_active_005")); - } + updateWalletCoinWithLock(withdrawDto.getAmount(), member.getId(), 2); + updateWalletMineWithLock(withdrawDto.getFee(), member.getId(), 2); - if (nftActive.getStatus() != 1) { - throw new FebsException(MessageSourceUtils.getString("nft_active_001")); - } + DappFundFlowEntity feeFlow = new DappFundFlowEntity(member.getId(), withdrawDto.getFee().negate(), 7, 2, null, null); + dappFundFlowDao.insert(feeFlow); - PriceDto priceDto = new PriceDto(); - priceDto.setAmount(BigDecimal.ZERO); - Map<String, BigDecimal> price = calPrice(priceDto); - DappFundFlowEntity fundFlow = new DappFundFlowEntity(member.getId(), new BigDecimal(activeDto.getCount()), 8, 1, BigDecimal.ZERO, activeDto.getTxHash()); - - fundFlow.setTargetAmount(price.get("x").multiply(new BigDecimal(activeDto.getCount()))); - fundFlow.setNewestPrice(AppContants.NFT_ACTIVE_PRICE.multiply(BigDecimal.valueOf(nftActive.getCount()))); + DappFundFlowEntity fundFlow = new DappFundFlowEntity(member.getId(), withdrawDto.getAmount().negate(), 5, 1, withdrawDto.getFee(), null); dappFundFlowDao.insert(fundFlow); - int count = nftActive.getCount() - activeDto.getCount(); - if (count == 0) { - nftActive.setStatus(3); - } +// String hash = ChainService.getInstance(ChainEnum.BSC_USDT.name()).transfer(member.getAddress(), withdrawDto.getAmount()); +// fundFlow.setToHash(hash); +// dappFundFlowDao.updateById(fundFlow); - nftActive.setCount(count); - nftActive.setHash(activeDto.getTxHash()); - dappNftActivationDao.updateById(nftActive); +// Map<String, Object> map = new HashMap<>(); +// map.put("fee", withdrawDto.getFee()); +// map.put("address", ChainEnum.BSC_TFC_FEE.getAddress()); +// map.put("flow", feeFlow.getId()); +// chainProducer.sendTfcFee(JSONObject.toJSONString(map)); } @Override - public List<ActiveNftListVo> findUnActiveNftList() { - DappMemberEntity member = LoginUserUtil.getAppUser(); - List<ActiveNftListVo> list = new ArrayList<>(); - - UpdateWrapper<DappNftActivation> query = new UpdateWrapper<>(); - query.eq("status", 1); - query.eq("member_id", member.getId()); - query.ge("expire_time", new Date()); - List<DappNftActivation> nftActivations = dappNftActivationDao.selectList(query); - - if (CollUtil.isEmpty(nftActivations)) { - return list; - } - - Date now = new Date(); - nftActivations.forEach(nft -> { - ActiveNftListVo nftVo = new ActiveNftListVo(); - nftVo.setId(nft.getId()); - nftVo.setRemain(DateUtil.between(now, nft.getExpireTime(), DateUnit.SECOND, false)); - nftVo.setCount(nft.getCount()); - list.add(nftVo); - }); - - return list; - } - - @Override - public void withdraw(WithdrawDto withdrawDto) { - - } - - @Override - public void updateWalletCoinWithLock(BigDecimal amount, Long memberId) { + public void updateWalletCoinWithLock(BigDecimal amount, Long memberId, int type) { boolean isSuccess = false; while(!isSuccess) { DappWalletCoinEntity walletCoin = dappWalletCoinDao.selectByMemberId(memberId); - walletCoin.setTotalAmount(walletCoin.getTotalAmount().add(amount)); - walletCoin.setAvailableAmount(walletCoin.getAvailableAmount().add(amount)); + // 加 + if (type == 1) { + walletCoin.setTotalAmount(walletCoin.getTotalAmount().add(amount)); + walletCoin.setAvailableAmount(walletCoin.getAvailableAmount().add(amount)); + // 减 + } else { + if (amount.compareTo(walletCoin.getAvailableAmount()) > 0) { + throw new FebsException("Balance Not Enough"); + } + walletCoin.setTotalAmount(walletCoin.getTotalAmount().subtract(amount)); + walletCoin.setAvailableAmount(walletCoin.getAvailableAmount().subtract(amount)); + } int i = dappWalletCoinDao.updateWithLock(walletCoin); if (i > 0) { @@ -395,4 +245,79 @@ } } } + + @Override + public void updateWalletMineWithLock(BigDecimal amount, Long memberId, int type) { + boolean isSuccess = false; + while(!isSuccess) { + DappWalletMineEntity walletMine = dappWalletMineDao.selectByMemberId(memberId); + + if (type == 1) { + walletMine.setTotalAmount(walletMine.getTotalAmount().add(amount)); + walletMine.setAvailableAmount(walletMine.getAvailableAmount().add(amount)); + } else { + if (amount.compareTo(walletMine.getAvailableAmount()) > 0) { + throw new FebsException("TFC Not Enough"); + } + walletMine.setTotalAmount(walletMine.getTotalAmount().subtract(amount)); + walletMine.setAvailableAmount(walletMine.getAvailableAmount().subtract(amount)); + } + + int i = dappWalletMineDao.updateWithLock(walletMine); + if (i > 0) { + isSuccess = true; + } + } + } + + @Override + public void addFrozenAmountWithLock(BigDecimal amount, Long memberId) { + boolean isSuccess = false; + while(!isSuccess) { + DappWalletCoinEntity walletCoin = dappWalletCoinDao.selectByMemberId(memberId); + + walletCoin.setTotalAmount(walletCoin.getTotalAmount().add(amount)); + walletCoin.setFrozenAmount(walletCoin.getFrozenAmount().add(amount)); + + int i = dappWalletCoinDao.updateWithLock(walletCoin); + if (i > 0) { + isSuccess = true; + + DappFundFlowEntity frozenAmount = new DappFundFlowEntity(memberId, amount, 9, 2, null, null); + dappFundFlowDao.insert(frozenAmount); + } + } + } + + @Override + public void releaseFrozenAmountWithLock(Long memberId) { + boolean isSuccess = false; + while(!isSuccess) { + DappWalletCoinEntity walletCoin = dappWalletCoinDao.selectByMemberId(memberId); + + if (walletCoin.getFrozenAmount().compareTo(BigDecimal.ZERO) <= 0) { + return; + } + BigDecimal frozen = walletCoin.getFrozenAmount(); + + walletCoin.setAvailableAmount(walletCoin.getAvailableAmount().add(frozen)); + walletCoin.setFrozenAmount(walletCoin.getFrozenAmount().subtract(frozen)); + + int i = dappWalletCoinDao.updateWithLock(walletCoin); + if (i > 0) { + isSuccess = true; + + DappFundFlowEntity releaseFrozen = new DappFundFlowEntity(memberId, frozen, 10, 2, null, null); + dappFundFlowDao.insert(releaseFrozen); + + DappFundFlowEntity releaseFrozenNegate = new DappFundFlowEntity(memberId, frozen.negate(), 9, 2, null, null); + dappFundFlowDao.insert(releaseFrozenNegate); + } + } + } + + @Override + public DappWalletCoinEntity findByMemberId(Long memberId) { + return dappWalletCoinDao.selectByMemberId(memberId); + } } -- Gitblit v1.9.1