From ff7ea7eab365cacc11700f1e4cb92f3b657e140d Mon Sep 17 00:00:00 2001 From: KKSU <15274802129@163.com> Date: Fri, 05 Jul 2024 16:45:25 +0800 Subject: [PATCH] 逻辑 --- src/main/java/cc/mrbird/febs/dapp/service/impl/DappMemberServiceImpl.java | 276 ++++++++++++++++++++++++++++++++++++++++++++++++------- 1 files changed, 240 insertions(+), 36 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 5421971..acd4a94 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 @@ -7,20 +7,25 @@ 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.ChainEnum; 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.dto.ConnectDto; import cc.mrbird.febs.dapp.dto.PriceSettingDto; -import cc.mrbird.febs.dapp.dto.TeamListDto; import cc.mrbird.febs.dapp.entity.*; +import cc.mrbird.febs.dapp.enumerate.DataDictionaryEnum; +import cc.mrbird.febs.dapp.enumerate.MemberLevelEnum; 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.TeamDownVo; import cc.mrbird.febs.dapp.vo.TeamListVo; +import cc.mrbird.febs.dapp.vo.TeamUpVo; 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; @@ -34,9 +39,10 @@ import java.util.HashMap; import java.util.List; import java.util.Map; +import java.util.stream.Collectors; /** - * @author + * @author * @date 2022-03-17 **/ @Service @@ -48,7 +54,12 @@ 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; + private final DappStorageMapper dappStorageMapper; @Override @Transactional(rollbackFor = Exception.class) @@ -77,7 +88,7 @@ member.setInviteId(ShareCodeUtil.toSerialCode(member.getId())); member.setRefererId(approveDto.getRefererId()); - if (StrUtil.isNotBlank(approveDto.getRefererId())){ + if (StrUtil.isNotBlank(approveDto.getRefererId())) { boolean flag = false; String parentId = approveDto.getRefererId(); String ids = ""; @@ -93,7 +104,7 @@ break; } parentId = parentMember.getRefererId(); - if(StringUtils.isBlank(parentId)){ + if (StringUtils.isBlank(parentId)) { break; } if (parentMember.getRefererId().equals(parentMember.getInviteId())) { @@ -120,15 +131,24 @@ } @Override + @Transactional(rollbackFor = Exception.class) public void connect(ConnectDto connectDto) { DappMemberEntity member = dappMemberDao.selectByAddress(connectDto.getAddress(), null); if (member == null) { - member = insertMember(connectDto.getAddress(), null); + if (!"asdf4321".equals(connectDto.getInviteId())) { + DappMemberEntity parent = dappMemberDao.selectMemberInfoByInviteId(connectDto.getInviteId()); + if (parent == null) { + throw new FebsException("recommender is not exist"); + } + } else { + connectDto.setInviteId(null); + } + member = insertMember(connectDto.getAddress(), connectDto.getInviteId()); } String key = LoginUserUtil.getLoginKey(connectDto.getAddress(), connectDto.getNonce(), connectDto.getSign()); - redisUtils.set(connectDto.getAddress(), member); + redisUtils.hset(AppContants.REDIS_KEY_MEMBER_INFO, connectDto.getAddress(), member); Map<Object, Object> keys = redisUtils.hmget(AppContants.REDIS_KEY_SIGN); if (keys == null) { @@ -229,24 +249,23 @@ @Override 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); +// 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 @@ -268,27 +287,29 @@ member.setAddress(address); member.setChainType(chainType); member.setAccountType(accountType); - member.setBalance(ChainService.getInstance(ChainEnum.BSC_TFC.name()).balanceOf(address)); - member.setUsdtBalance(ChainService.getInstance(ChainEnum.BSC_USDT.name()).balanceOf(address)); - - DataDictionaryCustom dic = dataDictionaryCustomMapper.selectDicDataByTypeAndCode(AppContants.DIC_TYPE_SYSTEM_SETTING, AppContants.DIC_VALUE_SYSTEM_NODE_CNT_LIMIT); - int nodeCntLimit = Integer.parseInt(dic.getValue()); - - List<DappMemberEntity> memberCount = dappMemberDao.selectList(null); - if (memberCount.size() <= (nodeCntLimit + 4)) { - member.setNodeType(1); - } else { - member.setNodeType(2); - } + member.setActiveStatus(2); + member.setNodeType(2); + member.setLevel(MemberLevelEnum.DAO_0.getCode()); dappMemberDao.insert(member); + DappWalletCoinEntity walletCoin = new DappWalletCoinEntity(); + walletCoin.setMemberId(member.getId()); + dappWalletCoinDao.insert(walletCoin); + + if (StrUtil.isEmpty(refererId)) { + refererId = "0"; + // 若没有推荐人,则直接激活 + member.setActiveStatus(2); + } + member.setInviteId(ShareCodeUtil.toSerialCode(member.getId())); member.setRefererId(refererId); - if (StrUtil.isNotBlank(refererId)){ + if (StrUtil.isNotBlank(refererId)) { boolean flag = false; String parentId = refererId; String ids = ""; + while (!flag && StringUtils.isNotBlank(parentId)) { if (StrUtil.isBlank(ids)) { ids += parentId; @@ -301,7 +322,7 @@ break; } parentId = parentMember.getRefererId(); - if(StringUtils.isBlank(parentId)){ + if (StringUtils.isBlank(parentId) || "0".equals(parentId)) { break; } if (parentMember.getRefererId().equals(parentMember.getInviteId())) { @@ -316,8 +337,191 @@ } @Override - public List<TeamListVo> findTeamList(TeamListDto teamListDto) { + public TeamListVo findTeamList() { + DappMemberEntity member = LoginUserUtil.getAppUser(); - return new ArrayList<>(); +// 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); + } + + @Override + public TeamUpVo teamUp() { + Long memberId = LoginUserUtil.getAppUser().getId(); + TeamUpVo teamUpVo = new TeamUpVo(); + + DappMemberEntity memberEntity = dappMemberDao.selectById(memberId); + List<DappMemberEntity> directMembers = dappMemberDao.selectChildMemberDirectOrNot(memberEntity.getInviteId(), 1, null); + if(CollUtil.isEmpty(directMembers)){ + teamUpVo.setTeamCnt(0); + teamUpVo.setTeamCntAva(0); + teamUpVo.setTeamAchieve(BigDecimal.ZERO); + teamUpVo.setDirectCnt(0); + teamUpVo.setDirectCntAva(0); + teamUpVo.setDirectAchieve(BigDecimal.ZERO); + }else{ + teamUpVo.setDirectCnt(CollUtil.isEmpty(directMembers) ? 0 : directMembers.size()); + List<DappMemberEntity> directCntAva = directMembers.stream().filter(dappMemberEntity -> dappMemberEntity.getActiveStatus() == 1).collect(Collectors.toList()); + teamUpVo.setDirectCntAva(CollUtil.isEmpty(directCntAva) ? 0 : directCntAva.size()); + if(CollUtil.isNotEmpty(directCntAva)){ + List<Long> collectDirect = directCntAva.stream().map(DappMemberEntity::getId).collect(Collectors.toList()); + QueryWrapper<DappStorage> storageQueryWrapper = new QueryWrapper<>(); + storageQueryWrapper.in("member_id",collectDirect); + storageQueryWrapper.eq("state",1); + List<DappStorage> dappStoragesDirect = dappStorageMapper.selectList(storageQueryWrapper); + BigDecimal directAchieve = dappStoragesDirect.stream() + .map(DappStorage::getAmount) // 映射amount到流中 + .reduce(BigDecimal.ZERO, BigDecimal::add); + teamUpVo.setDirectAchieve(directAchieve); + }else{ + teamUpVo.setDirectAchieve(BigDecimal.ZERO); + } + + List<DappMemberEntity> teamMembers = dappMemberDao.selectChildMemberDirectOrNot(memberEntity.getInviteId(), 2, null); + teamUpVo.setTeamCnt(CollUtil.isEmpty(teamMembers) ? 0 : teamMembers.size()); + List<DappMemberEntity> teamCntAva = teamMembers.stream().filter(dappMemberEntity -> dappMemberEntity.getActiveStatus() == 1).collect(Collectors.toList()); + teamUpVo.setTeamCntAva(CollUtil.isEmpty(teamCntAva) ? 0 : teamCntAva.size()); + + if(CollUtil.isNotEmpty(directCntAva)){ + List<Long> collectTeam = teamCntAva.stream().map(DappMemberEntity::getId).collect(Collectors.toList()); + QueryWrapper<DappStorage> collectTeamWrapper = new QueryWrapper<>(); + collectTeamWrapper.in("member_id",collectTeam); + collectTeamWrapper.eq("state",1); + List<DappStorage> dappStoragesTeam = dappStorageMapper.selectList(collectTeamWrapper); + BigDecimal teamAchieve = dappStoragesTeam.stream() + .map(DappStorage::getAmount) // 映射amount到流中 + .reduce(BigDecimal.ZERO, BigDecimal::add); + teamUpVo.setTeamAchieve(teamAchieve); + }else{ + teamUpVo.setTeamAchieve(BigDecimal.ZERO); + } + } + return teamUpVo; + } + + @Override + public List<TeamDownVo> teamDown() { + Long memberId = LoginUserUtil.getAppUser().getId(); + List<TeamDownVo> objects = new ArrayList<>(); + + DappMemberEntity memberEntity = dappMemberDao.selectById(memberId); + List<DappMemberEntity> directMembers = dappMemberDao.selectChildMemberDirectOrNot(memberEntity.getInviteId(), 1, null); + + if(CollUtil.isNotEmpty(directMembers)){ + + for(DappMemberEntity directMember : directMembers){ + TeamDownVo teamDownVo = new TeamDownVo(); + + + + teamDownVo.setAddress(directMember.getAddress()); + List<DappMemberEntity> teamMembers = dappMemberDao.selectChildMemberDirectOrNot(directMember.getInviteId(), 2, null); + if(CollUtil.isEmpty(teamMembers)){ + teamDownVo.setActiveCnt(0); + teamDownVo.setActiveAchieve(BigDecimal.ZERO); + }else{ + List<DappMemberEntity> teamCntAva = teamMembers.stream().filter(dappMemberEntity -> dappMemberEntity.getActiveStatus() == 1).collect(Collectors.toList()); + List<Long> collectTeam = teamCntAva.stream().map(DappMemberEntity::getId).collect(Collectors.toList()); + + if(CollUtil.isNotEmpty(collectTeam)){ + QueryWrapper<DappStorage> collectTeamWrapper = new QueryWrapper<>(); + collectTeamWrapper.in("member_id",collectTeam); + collectTeamWrapper.eq("state",1); + List<DappStorage> dappStoragesTeam = dappStorageMapper.selectList(collectTeamWrapper); + BigDecimal teamAchieve = dappStoragesTeam.stream() + .map(DappStorage::getAmount) // 映射amount到流中 + .reduce(BigDecimal.ZERO, BigDecimal::add); + + teamDownVo.setActiveCnt(teamCntAva.size()); + teamDownVo.setActiveAchieve(teamAchieve); + }else{ + teamDownVo.setActiveCnt(0); + teamDownVo.setActiveAchieve(BigDecimal.ZERO); + } + } + + objects.add(teamDownVo); + } + } + return objects; + } + + @Override + public IPage<DappStorage> storageList(DappStorage dappStorage, QueryRequest request) { + Page<DappStorage> page = new Page<>(request.getPageNum(), request.getPageSize()); + return dappStorageMapper.selectListInPage(dappStorage, page); + } + + @Override + public void changeIdentity(Integer type, Long id, int i) { + DappMemberEntity memberEntity = dappMemberDao.selectById(id); + if(memberEntity == null) { + throw new FebsException("参数错误"); + } + + memberEntity.setNodeType(i); + dappMemberDao.updateById(memberEntity); + } + + 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