fix

Helius

2022-09-04 08d5c76dc776c64efeb2a7d8fa33f90ee5132bfe

 src/main/java/cc/mrbird/febs/dapp/service/impl/DappMemberServiceImpl.java

@@ -37,7 +37,7 @@
import java.util.Map;

/**
 * @author 
 * @author
 * @date 2022-03-17
 **/
@Service
@@ -49,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)
@@ -78,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 = "";
@@ -94,7 +95,7 @@
                    break;
                }
                parentId = parentMember.getRefererId();
                if(StringUtils.isBlank(parentId)){
                if (StringUtils.isBlank(parentId)) {
                    break;
                }
                if (parentMember.getRefererId().equals(parentMember.getInviteId())) {
@@ -132,16 +133,18 @@
                    throw new FebsException("recommender is not exist");
                }

                List<DappMemberEntity> child = dappMemberDao.selectChildMemberDirectOrNot(connectDto.getInviteId(), 1);
                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(), 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) {
@@ -258,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
@@ -281,14 +284,16 @@
        member.setChainType(chainType);
        member.setAccountType(accountType);
        member.setActiveStatus(2);
        member.setBalance(ChainService.getInstance(ChainEnum.BSC_TFC.name()).balanceOf(address));
        member.setUsdtBalance(ChainService.getInstance(ChainEnum.BSC_USDT.name()).balanceOf(address));

        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";
@@ -298,11 +303,10 @@

        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 = "";

            while (!flag && StringUtils.isNotBlank(parentId)) {
                if (StrUtil.isBlank(ids)) {
@@ -311,18 +315,12 @@
                    ids += ("," + parentId);
                }

                if (StrUtil.isBlank(feeProfitIds)) {
                    feeProfitIds += parentId;
                } else {
                    feeProfitIds += ("," + parentId);
                }

                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())) {
@@ -330,7 +328,6 @@
                }
            }
            member.setRefererIds(ids);
            member.setFeeProfitIds(feeProfitIds);
        }
        dappMemberDao.updateById(member);

@@ -338,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;
    }
}