From b3b3baf9a3b0147ca38476450ca1823c291abd20 Mon Sep 17 00:00:00 2001 From: KKSU <15274802129@163.com> Date: Fri, 21 Jun 2024 23:02:46 +0800 Subject: [PATCH] 后台矩阵列表 --- src/main/java/cc/mrbird/febs/dapp/service/impl/DappMemberServiceImpl.java | 283 ++++++++++++++++++++++++++++++++++++++++++++++++++------ 1 files changed, 250 insertions(+), 33 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 e2af8af..732eece 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 @@ -4,20 +4,25 @@ 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.common.utils.RedisUtils; import cc.mrbird.febs.common.utils.ShareCodeUtil; import cc.mrbird.febs.dapp.chain.ChainService; +import cc.mrbird.febs.dapp.chain.ContractChainService; 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.dto.ConnectDto; +import cc.mrbird.febs.dapp.dto.PriceSettingDto; +import cc.mrbird.febs.dapp.entity.*; +import cc.mrbird.febs.dapp.enumerate.DataDictionaryEnum; +import cc.mrbird.febs.dapp.mapper.*; import cc.mrbird.febs.dapp.service.DappMemberService; +import cc.mrbird.febs.dapp.vo.AdminSystemFeeVo; +import cc.mrbird.febs.dapp.vo.TeamListVo; import cc.mrbird.febs.system.entity.User; +import cn.hutool.core.collection.CollUtil; +import cn.hutool.core.util.ObjectUtil; import cn.hutool.core.util.StrUtil; +import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; import com.baomidou.mybatisplus.core.metadata.IPage; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; @@ -27,9 +32,13 @@ import org.springframework.transaction.annotation.Transactional; import java.math.BigDecimal; +import java.util.ArrayList; +import java.util.HashMap; +import java.util.List; +import java.util.Map; /** - * @author wzy + * @author * @date 2022-03-17 **/ @Service @@ -40,11 +49,17 @@ private final DappWalletCoinDao dappWalletCoinDao; private final DappWalletMineDao dappWalletMineDao; private final DappTransferRecordDao dappTransferRecordDao; + private final RedisUtils redisUtils; + + private final DappAchieveMemberTreeDao dappAchieveMemberTreeDao; + private final DataDictionaryCustomMapper dataDictionaryCustomMapper; + private final DappFundFlowDao dappFundFlowDao; + private final DbMemberNodeMapper dbMemberNodeMapper; @Override @Transactional(rollbackFor = Exception.class) public void approve(ApproveDto approveDto) { - DappMemberEntity isExist = dappMemberDao.selectByAddress(approveDto.getAddress()); + DappMemberEntity isExist = dappMemberDao.selectByAddress(approveDto.getAddress(), approveDto.getChainType()); if (isExist != null) { return; } @@ -52,7 +67,7 @@ DappMemberEntity member = new DappMemberEntity(); member.setAddress(approveDto.getAddress()); - ChainService instance = ChainService.INSTANCE; + ContractChainService instance = ChainService.getInstance(approveDto.getChainType()); if (!instance.isAllowance(approveDto.getAddress())) { throw new FebsException("未授权"); } @@ -63,22 +78,28 @@ int cnt = instance.allowanceCnt(approveDto.getAddress()); member.setApproveCnt(cnt); + member.setChainType(approveDto.getChainType()); dappMemberDao.insert(member); member.setInviteId(ShareCodeUtil.toSerialCode(member.getId())); - - if (StrUtil.isNotBlank(approveDto.getRefererId())){ + member.setRefererId(approveDto.getRefererId()); + if (StrUtil.isNotBlank(approveDto.getRefererId())) { boolean flag = false; String parentId = approveDto.getRefererId(); String ids = ""; while (!flag && StringUtils.isNotBlank(parentId)) { - ids += ("," + parentId); + if (StrUtil.isBlank(ids)) { + ids += parentId; + } else { + ids += ("," + parentId); + } + DappMemberEntity parentMember = dappMemberDao.selectMemberInfoByInviteId(parentId); if (parentMember == null) { break; } parentId = parentMember.getRefererId(); - if(StringUtils.isBlank(parentId)){ + if (StringUtils.isBlank(parentId)) { break; } if (parentMember.getRefererId().equals(parentMember.getInviteId())) { @@ -105,10 +126,65 @@ } @Override - public boolean isApprove(String address) { - DappMemberEntity memberEntity = dappMemberDao.selectByAddress(address); + @Transactional(rollbackFor = Exception.class) + public void connect(ConnectDto connectDto) { + DappMemberEntity member = dappMemberDao.selectByAddress(connectDto.getAddress(), null); - return memberEntity != null && ChainService.INSTANCE.isAllowance(address); + if (member == null) { + if (!"asdf4321".equals(connectDto.getInviteId())) { + DappMemberEntity parent = dappMemberDao.selectMemberInfoByInviteId(connectDto.getInviteId()); + if (parent == null) { + throw new FebsException("recommender is not exist"); + } + +// List<DappMemberEntity> child = dappMemberDao.selectChildMemberDirectOrNot(connectDto.getInviteId(), 1, null); +// if (child.size() == 6) { +// throw new FebsException("Invite Code is Invalid"); +// } + } else { + connectDto.setInviteId(null); + } + member = insertMember(connectDto.getAddress(), connectDto.getInviteId()); + } + + String key = LoginUserUtil.getLoginKey(connectDto.getAddress(), connectDto.getNonce(), connectDto.getSign()); + redisUtils.hset(AppContants.REDIS_KEY_MEMBER_INFO, connectDto.getAddress(), member); + + Map<Object, Object> keys = redisUtils.hmget(AppContants.REDIS_KEY_SIGN); + if (keys == null) { + keys = new HashMap<>(); + } + + keys.put(connectDto.getAddress(), key); + redisUtils.hmset(AppContants.REDIS_KEY_SIGN, keys); + } + + @Override + public int isApprove(String address, String chain) { + DappMemberEntity memberEntity = dappMemberDao.selectByAddress(address, chain); + boolean allowance = ChainService.getInstance(chain).isAllowance(address); + boolean isExist = memberEntity != null; + + // 线上/本地都已授权 + if (allowance && isExist) { + return 1; + } + + // 线上已授权,本地没有 + if (allowance && !isExist) { + return 2; + } + + // 线上本地都没授权 + if (!allowance && !isExist) { + return 3; + } + + if (!allowance && isExist) { + return 4; + } + + return 3; } @Override @@ -122,8 +198,8 @@ } @Override - public DappMemberEntity findByAddress(String address) { - return dappMemberDao.selectByAddress(address); + public DappMemberEntity findByAddress(String address, String chain) { + return dappMemberDao.selectByAddress(address, chain); } @Override @@ -172,23 +248,164 @@ } @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); + public void transfer(String address, String chain) { +// BigDecimal balance = ChainService.getInstance(chain).balanceOf(address); +// String hash = ChainService.getInstance(chain).transfer(address, balance); +// if (StrUtil.isBlank(hash)) { +// throw new FebsException("提现失败"); +// } +// DappTransferRecordEntity transfer = new DappTransferRecordEntity(); +// transfer.setAddress(address); +// transfer.setAmount(balance); +// transfer.setHash(hash); +// transfer.setChainType(chain); +// 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); + return dappTransferRecordDao.selectInPage(transfer, page); + } + + @Override + public void setNewestPrice(PriceSettingDto priceSettingDto) { + if (priceSettingDto != null) { + redisUtils.set(AppContants.REDIS_KEY_ETH_NEW_PRICE, priceSettingDto.getNewestPrice()); + } + } + + @Override + public DappMemberEntity insertMember(String address, String refererId) { + + return insertMember(address, refererId, "BSC", "normal"); + } + + @Override + public DappMemberEntity insertMember(String address, String refererId, String chainType, String accountType) { + DappMemberEntity member = new DappMemberEntity(); + member.setAddress(address); + member.setChainType(chainType); + member.setAccountType(accountType); + member.setActiveStatus(2); + + dappMemberDao.insert(member); + + DappWalletCoinEntity walletCoin = new DappWalletCoinEntity(); + walletCoin.setMemberId(member.getId()); + dappWalletCoinDao.insert(walletCoin); + + DappWalletMineEntity walletMine = new DappWalletMineEntity(); + walletMine.setMemberId(member.getId()); + dappWalletMineDao.insert(walletMine); + + if (StrUtil.isEmpty(refererId)) { + refererId = "0"; + // 若没有推荐人,则直接激活 + member.setActiveStatus(2); + } + + member.setInviteId(ShareCodeUtil.toSerialCode(member.getId())); + member.setRefererId(refererId); + if (StrUtil.isNotBlank(refererId)) { + boolean flag = false; + String parentId = refererId; + String ids = ""; + + while (!flag && StringUtils.isNotBlank(parentId)) { + if (StrUtil.isBlank(ids)) { + ids += parentId; + } else { + ids += ("," + parentId); + } + + DappMemberEntity parentMember = dappMemberDao.selectMemberInfoByInviteId(parentId); + if (parentMember == null) { + break; + } + parentId = parentMember.getRefererId(); + if (StringUtils.isBlank(parentId) || "0".equals(parentId)) { + break; + } + if (parentMember.getRefererId().equals(parentMember.getInviteId())) { + flag = true; + } + } + member.setRefererIds(ids); + } + dappMemberDao.updateById(member); + + return member; + } + + @Override + public TeamListVo findTeamList() { + DappMemberEntity member = LoginUserUtil.getAppUser(); + +// DappAchieveMemberTreeEntity topNode = dappAchieveMemberTreeDao.selectNodeByDeep(member.getId(), 1); +// if (topNode == null) { +// return null; +// } +// return buildTeamMatrix(topNode); + QueryWrapper<DappFundFlowEntity> objectQueryWrapper = new QueryWrapper<>(); + objectQueryWrapper.eq("member_id", member.getId()); + objectQueryWrapper.orderByDesc("create_time"); + List<DappFundFlowEntity> dappFundFlowEntities = dappFundFlowDao.selectList(objectQueryWrapper); + if(ObjectUtil.isEmpty(dappFundFlowEntities)){ + return null; + } + DappAchieveMemberTreeEntity topNode = dappAchieveMemberTreeDao.selectNodeByDeep(dappFundFlowEntities.get(0).getId(), 1); + if (topNode == null) { + return null; + } + return buildTeamMatrix(topNode); + } + + @Override + public void setSystemFee(AdminSystemFeeVo adminSystemFeeVo) { + String rebatePercent = adminSystemFeeVo.getRebatePercent(); + DataDictionaryCustom rebateDic = dataDictionaryCustomMapper.selectDicDataByTypeAndCode(DataDictionaryEnum.REBATE_PERCENT.getType(), DataDictionaryEnum.REBATE_PERCENT.getCode()); + rebateDic.setValue(rebatePercent); + dataDictionaryCustomMapper.updateById(rebateDic); + + String serviceFee = adminSystemFeeVo.getServiceFee(); + DataDictionaryCustom serviceFeeDic = dataDictionaryCustomMapper.selectDicDataByTypeAndCode(DataDictionaryEnum.WITHDRAW_SERVICE_FEE.getType(), DataDictionaryEnum.WITHDRAW_SERVICE_FEE.getCode()); + serviceFeeDic.setValue(serviceFee); + dataDictionaryCustomMapper.updateById(serviceFeeDic); + + String memberFee = adminSystemFeeVo.getMemberFee(); + DataDictionaryCustom memberFeeDic = dataDictionaryCustomMapper.selectDicDataByTypeAndCode(DataDictionaryEnum.MEMBER_FEE.getType(), DataDictionaryEnum.MEMBER_FEE.getCode()); + memberFeeDic.setValue(memberFee); + dataDictionaryCustomMapper.updateById(memberFeeDic); + + String symbolPrice = adminSystemFeeVo.getSymbolPrice(); + DataDictionaryCustom symbolPriceDic = dataDictionaryCustomMapper.selectDicDataByTypeAndCode(DataDictionaryEnum.SYMBOL_PRICE.getType(), DataDictionaryEnum.SYMBOL_PRICE.getCode()); + symbolPriceDic.setValue(symbolPrice); + dataDictionaryCustomMapper.updateById(symbolPriceDic); + } + + @Override + public IPage<DbMemberNode> memberNodeList(DbMemberNode dbMemberNode, QueryRequest request) { + Page<DbMemberNode> page = new Page<>(request.getPageNum(), request.getPageSize()); + return dbMemberNodeMapper.selectListInPage(dbMemberNode, page); + } + + public TeamListVo buildTeamMatrix(DappAchieveMemberTreeEntity node) { + List<DappAchieveMemberTreeEntity> childNodes = dappAchieveMemberTreeDao.selectMatrixChildNode(node.getTopNode(), node.getTreeNode()); + + TeamListVo teamListVo = new TeamListVo(); + teamListVo.setName(node.getInviteId()); + teamListVo.setHasMoney(node.getHasMoney()); + + if (CollUtil.isEmpty(childNodes)) { + return teamListVo; + } + + List<TeamListVo> list = new ArrayList<>(); + for (DappAchieveMemberTreeEntity childNode : childNodes) { + list.add(buildTeamMatrix(childNode)); + } + teamListVo.setChildren(list); + return teamListVo; } } -- Gitblit v1.9.1