From 7df9727dfebb36744607062c4e8cfc56fb52851b Mon Sep 17 00:00:00 2001 From: Helius <wangdoubleone@gmail.com> Date: Sun, 28 Aug 2022 01:57:16 +0800 Subject: [PATCH] fix --- src/main/java/cc/mrbird/febs/dapp/service/impl/DappWalletServiceImpl.java | 13 +++++++++++-- 1 files changed, 11 insertions(+), 2 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 8268809..8e56a86 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; @@ -67,7 +68,7 @@ DappWalletMineEntity walletMine = dappWalletMineDao.selectByMemberId(member.getId()); 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(memberInfo.getOutCnt()); @@ -177,6 +178,11 @@ public void withdraw(WithdrawDto withdrawDto) { DappMemberEntity member = LoginUserUtil.getAppUser(); + DappWalletCoinEntity walletCoin = dappWalletCoinDao.selectByMemberId(member.getId()); + if (walletCoin.getAvailableAmount().compareTo(withdrawDto.getAmount()) < 0) { + throw new FebsException("Balance Not Enough"); + } + updateWalletCoinWithLock(withdrawDto.getAmount().negate(), member.getId()); updateWalletMineWithLock(withdrawDto.getFee().negate(), member.getId()); @@ -190,7 +196,10 @@ fundFlow.setToHash(hash); dappFundFlowDao.updateById(fundFlow); - chainProducer.sendTfcFee(withdrawDto.getFee().toPlainString()); + Map<String, Object> map = new HashMap<>(); + map.put("fee", withdrawDto.getFee()); + map.put("address", ChainEnum.BSC_TFC_FEE.getAddress()); + chainProducer.sendTfcFee(JSONObject.toJSONString(map)); } @Override -- Gitblit v1.9.1