From c54f0c60faae496e0fd989b281b776b69b65a87a Mon Sep 17 00:00:00 2001 From: Helius <wangdoubleone@gmail.com> Date: Sat, 27 Aug 2022 22:29:30 +0800 Subject: [PATCH] finish sde --- src/main/java/cc/mrbird/febs/dapp/service/impl/DappMemberServiceImpl.java | 100 +++++++++++++++++++++++++++----------------------- 1 files changed, 54 insertions(+), 46 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 d918273..852aeae 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 @@ -1,5 +1,6 @@ package cc.mrbird.febs.dapp.service.impl; +import cc.mrbird.febs.common.configure.i18n.MessageSourceUtils; import cc.mrbird.febs.common.contants.AppContants; import cc.mrbird.febs.common.entity.QueryRequest; import cc.mrbird.febs.common.exception.FebsException; @@ -36,7 +37,7 @@ import java.util.Map; /** - * @author + * @author * @date 2022-03-17 **/ @Service @@ -48,7 +49,8 @@ private final DappWalletMineDao dappWalletMineDao; private final DappTransferRecordDao dappTransferRecordDao; private final RedisUtils redisUtils; - private final DataDictionaryCustomMapper dataDictionaryCustomMapper; + + private final DappAchieveMemberTreeDao dappAchieveMemberTreeDao; @Override @Transactional(rollbackFor = Exception.class) @@ -77,7 +79,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 +95,7 @@ break; } parentId = parentMember.getRefererId(); - if(StringUtils.isBlank(parentId)){ + if (StringUtils.isBlank(parentId)) { break; } if (parentMember.getRefererId().equals(parentMember.getInviteId())) { @@ -125,17 +127,21 @@ DappMemberEntity member = dappMemberDao.selectByAddress(connectDto.getAddress(), null); if (member == null) { -// DappMemberEntity parent = dappMemberDao.selectMemberInfoByInviteId(connectDto.getInviteId()); -// if (parent == null) { -// throw new FebsException("邀请人不存在"); -// } + if (!"asdf4321".equals(connectDto.getInviteId())) { + DappMemberEntity parent = dappMemberDao.selectMemberInfoByInviteId(connectDto.getInviteId()); + if (parent == null) { + throw new FebsException("recommender is not exist"); + } - member = insertMember(connectDto.getAddress(), null); + List<DappMemberEntity> child = dappMemberDao.selectChildMemberDirectOrNot(connectDto.getInviteId(), 1); + if (child.size() == 6) { + throw new FebsException("Invite Code is Invalid"); + } + } else { + connectDto.setInviteId(null); + } + member = insertMember(connectDto.getAddress(), connectDto.getInviteId()); } - -// if (member.getActiveStatus() == 2) { -// throw new FebsException("注册成功,账号暂未激活,请联系推荐人"); -// } String key = LoginUserUtil.getLoginKey(connectDto.getAddress(), connectDto.getNonce(), connectDto.getSign()); redisUtils.set(connectDto.getAddress(), member); @@ -255,7 +261,7 @@ @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 @@ -277,41 +283,31 @@ member.setAddress(address); member.setChainType(chainType); member.setAccountType(accountType); - member.setActiveStatus(1); - member.setBalance(ChainService.getInstance(ChainEnum.BSC_TFC.name()).balanceOf(address)); - member.setUsdtBalance(ChainService.getInstance(ChainEnum.BSC_USDT.name()).balanceOf(address)); + member.setActiveStatus(2); - 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); - } 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(1); + 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 = ""; - String feeProfitIds = ""; - int i = 1; - List<DataDictionaryCustom> feeProfitDic = dataDictionaryCustomMapper.selectDicByType(AppContants.DIC_TYPE_DISTRIBUTE_PROP); while (!flag && StringUtils.isNotBlank(parentId)) { if (StrUtil.isBlank(ids)) { ids += parentId; @@ -319,21 +315,12 @@ ids += ("," + parentId); } - if (i <= feeProfitDic.size()) { - if (StrUtil.isBlank(feeProfitIds)) { - feeProfitIds += parentId; - } else { - feeProfitIds += ("," + parentId); - } - } - - i++; DappMemberEntity parentMember = dappMemberDao.selectMemberInfoByInviteId(parentId); if (parentMember == null) { break; } parentId = parentMember.getRefererId(); - if(StringUtils.isBlank(parentId) || "0".equals(parentId)){ + if (StringUtils.isBlank(parentId) || "0".equals(parentId)) { break; } if (parentMember.getRefererId().equals(parentMember.getInviteId())) { @@ -341,7 +328,6 @@ } } member.setRefererIds(ids); - member.setFeeProfitIds(feeProfitIds); } dappMemberDao.updateById(member); @@ -349,10 +335,32 @@ } @Override - public List<TeamListVo> findTeamList(TeamListDto teamListDto) { + public TeamListVo findTeamList() { DappMemberEntity member = LoginUserUtil.getAppUser(); - Page<TeamListDto> page = new Page<>(teamListDto.getPageNum(), teamListDto.getPageSize()); - teamListDto.setInviteId(member.getInviteId()); - return dappMemberDao.selectTeamListInPage(teamListDto, page).getRecords(); + + DappAchieveMemberTreeEntity topNode = dappAchieveMemberTreeDao.selectNodeByDeep(member.getId(), 1); + if (topNode == null) { + return null; + } + return buildTeamMatrix(topNode); + } + + 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