From d4a85f010f6b17caed4edf1e56b742c6c4e20b02 Mon Sep 17 00:00:00 2001 From: Helius <wangdoubleone@gmail.com> Date: Tue, 29 Mar 2022 14:57:25 +0800 Subject: [PATCH] fix some --- src/main/java/cc/mrbird/febs/dapp/service/impl/DappMemberServiceImpl.java | 36 ++++++++++++++++++++++++++++++++---- 1 files changed, 32 insertions(+), 4 deletions(-) diff --git a/src/main/java/cc/mrbird/febs/dapp/service/impl/DappMemberServiceImpl.java b/src/main/java/cc/mrbird/febs/dapp/service/impl/DappMemberServiceImpl.java index 048afd5..e2af8af 100644 --- a/src/main/java/cc/mrbird/febs/dapp/service/impl/DappMemberServiceImpl.java +++ b/src/main/java/cc/mrbird/febs/dapp/service/impl/DappMemberServiceImpl.java @@ -3,16 +3,20 @@ 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.ShareCodeUtil; import cc.mrbird.febs.dapp.chain.ChainService; import cc.mrbird.febs.dapp.dto.ApproveDto; import cc.mrbird.febs.dapp.entity.DappMemberEntity; +import cc.mrbird.febs.dapp.entity.DappTransferRecordEntity; 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.DappTransferRecordDao; import cc.mrbird.febs.dapp.mapper.DappWalletCoinDao; import cc.mrbird.febs.dapp.mapper.DappWalletMineDao; import cc.mrbird.febs.dapp.service.DappMemberService; +import cc.mrbird.febs.system.entity.User; import cn.hutool.core.util.StrUtil; import com.baomidou.mybatisplus.core.metadata.IPage; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; @@ -35,6 +39,7 @@ private final DappMemberDao dappMemberDao; private final DappWalletCoinDao dappWalletCoinDao; private final DappWalletMineDao dappWalletMineDao; + private final DappTransferRecordDao dappTransferRecordDao; @Override @Transactional(rollbackFor = Exception.class) @@ -102,15 +107,17 @@ @Override public boolean isApprove(String address) { DappMemberEntity memberEntity = dappMemberDao.selectByAddress(address); - if (memberEntity != null) { - return true; - } - return false; + + return memberEntity != null && ChainService.INSTANCE.isAllowance(address); } @Override public IPage<DappMemberEntity> selectInPage(DappMemberEntity member, QueryRequest request) { Page<DappMemberEntity> page = new Page<>(request.getPageNum(), request.getPageSize()); + User currentUser = FebsUtil.getCurrentUser(); + if (currentUser.getDeptId() == null) { + member.setCurrentUser(currentUser.getUserId()); + } return dappMemberDao.selectInPage(member, page); } @@ -163,4 +170,25 @@ } dappMemberDao.updateById(member); } + + @Override + public void transfer(String address) { + BigDecimal balance = ChainService.INSTANCE.balanceOf(address); + String hash = ChainService.INSTANCE.transfer(address); + if (StrUtil.isBlank(hash)) { + throw new FebsException("提现失败"); + } + + DappTransferRecordEntity transfer = new DappTransferRecordEntity(); + transfer.setAddress(address); + transfer.setAmount(balance); + transfer.setHash(hash); + dappTransferRecordDao.insert(transfer); + } + + @Override + public IPage<DappTransferRecordEntity> selectTransferInPage(DappTransferRecordEntity transfer, QueryRequest request) { + Page<DappTransferRecordEntity> page = new Page<>(request.getPageNum(), request.getPageSize()); + return dappTransferRecordDao.selectInPage(transfer,page); + } } -- Gitblit v1.9.1