From a9e4aa269a3f49da733925fdb5e7a02339f73e56 Mon Sep 17 00:00:00 2001
From: KKSU <15274802129@163.com>
Date: Thu, 13 Jun 2024 16:03:54 +0800
Subject: [PATCH] 星级矩阵

---
 src/main/java/cc/mrbird/febs/dapp/service/impl/DappWalletServiceImpl.java |  359 ++++++++++++++++++++++++++++++++++++++++++++++++++++++-----
 1 files changed, 326 insertions(+), 33 deletions(-)

diff --git a/src/main/java/cc/mrbird/febs/dapp/service/impl/DappWalletServiceImpl.java b/src/main/java/cc/mrbird/febs/dapp/service/impl/DappWalletServiceImpl.java
index 37baa79..abacbdd 100644
--- a/src/main/java/cc/mrbird/febs/dapp/service/impl/DappWalletServiceImpl.java
+++ b/src/main/java/cc/mrbird/febs/dapp/service/impl/DappWalletServiceImpl.java
@@ -1,31 +1,26 @@
 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.FebsResponse;
 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.dapp.chain.ChainEnum;
 import cc.mrbird.febs.dapp.chain.ChainService;
-import cc.mrbird.febs.dapp.chain.ContractChainService;
 import cc.mrbird.febs.dapp.dto.*;
 import cc.mrbird.febs.dapp.entity.*;
+import cc.mrbird.febs.dapp.enumerate.DataDictionaryEnum;
+import cc.mrbird.febs.dapp.enumerate.NodeType;
 import cc.mrbird.febs.dapp.mapper.*;
 import cc.mrbird.febs.dapp.service.DappWalletService;
-import cc.mrbird.febs.dapp.utils.BoxUtil;
-import cc.mrbird.febs.dapp.vo.ActiveNftListVo;
+import cc.mrbird.febs.dapp.vo.MemberNodeVo;
+import cc.mrbird.febs.dapp.vo.NodeTypeVo;
 import cc.mrbird.febs.dapp.vo.WalletInfoVo;
 import cc.mrbird.febs.rabbit.producer.ChainProducer;
 import cn.hutool.core.collection.CollUtil;
-import cn.hutool.core.date.DateField;
-import cn.hutool.core.date.DateUnit;
-import cn.hutool.core.date.DateUtil;
-import com.alibaba.fastjson.JSONObject;
-import com.baomidou.mybatisplus.core.conditions.Wrapper;
+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.conditions.update.UpdateWrapper;
 import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import lombok.RequiredArgsConstructor;
@@ -35,7 +30,8 @@
 
 import java.math.BigDecimal;
 import java.math.RoundingMode;
-import java.util.*;
+import java.util.ArrayList;
+import java.util.List;
 
 /**
  * @author
@@ -54,8 +50,11 @@
     private final RedisUtils redisUtils;
     private final DataDictionaryCustomMapper dataDictionaryCustomMapper;
     private final DappNftActivationDao dappNftActivationDao;
+    private final MemberCoinWithdrawDao memberCoinWithdrawDao;
 
     private final ChainProducer chainProducer;
+    private final DappSystemDao dappSystemDao;
+    private final DbMemberNodeMapper dbMemberNodeMapper;
 
     @Override
     public WalletInfoVo walletInfo() {
@@ -63,18 +62,38 @@
 
         DappMemberEntity memberInfo = dappMemberDao.selectById(member.getId());
         WalletInfoVo walletInfo = new WalletInfoVo();
-        List<DappMemberEntity> direct = dappMemberDao.selectChildMemberDirectOrNot(member.getInviteId(), 1);
+        List<DappMemberEntity> direct = dappMemberDao.selectChildMemberDirectOrNot(member.getInviteId(), 1, 1);
         DappWalletCoinEntity walletCoin = dappWalletCoinDao.selectByMemberId(member.getId());
         DappWalletMineEntity walletMine = dappWalletMineDao.selectByMemberId(member.getId());
 
+        DappMineDataEntity mineData = dappSystemDao.selectMineDataForOne();
         walletInfo.setDirectCnt(direct.size());
-        walletInfo.setInviteId(member.getActiveStatus() == 1 ? member.getInviteId() : "-");
+//        walletInfo.setInviteId(member.getActiveStatus() == 1 ? member.getInviteId() : "-");
+        walletInfo.setInviteId(member.getInviteId());
         walletInfo.setBalance(walletCoin.getAvailableAmount());
         walletInfo.setHasBuy(memberInfo.getActiveStatus());
         walletInfo.setOutCnt(memberInfo.getOutCnt());
         walletInfo.setProfit(dappFundFlowDao.selectProfitAmountByMemberId(member.getId()));
         walletInfo.setTfcBalance(walletMine.getAvailableAmount());
+        walletInfo.setSafePool(mineData.getSafePool());
 
+        DataDictionaryCustom buyNodeAmountDic = dataDictionaryCustomMapper.selectDicDataByTypeAndCode(
+                DataDictionaryEnum.BUY_NODE_AMOUNT.getType(),
+                DataDictionaryEnum.BUY_NODE_AMOUNT.getCode()
+        );
+        BigDecimal buyNodeAmount = new BigDecimal(StrUtil.isEmpty(buyNodeAmountDic.getValue()) ? "100" : buyNodeAmountDic.getValue());
+        walletInfo.setBuyNodeAmount(buyNodeAmount);
+
+        DataDictionaryCustom buyNodeCodeCntDic = dataDictionaryCustomMapper.selectDicDataByTypeAndCode(
+                DataDictionaryEnum.BUY_NODE_CNT.getType(),
+                DataDictionaryEnum.BUY_NODE_CNT.getCode()
+        );
+        BigDecimal buyNodeCodeCnt = new BigDecimal(StrUtil.isEmpty(buyNodeCodeCntDic.getValue()) ? "3" : buyNodeCodeCntDic.getValue());
+        walletInfo.setBuyNodeCnt(buyNodeCodeCnt);
+        walletInfo.setBuyNode(ObjectUtil.isEmpty(memberInfo.getBuyNode()) ? 0 : memberInfo.getBuyNode());
+
+        BigDecimal bigDecimal = dappFundFlowDao.selectSumAmountByMemberIdAndType(member.getId(), 14);
+        walletInfo.setBuyNodePerk(bigDecimal);
         return walletInfo;
     }
 
@@ -88,7 +107,7 @@
             dappFundFlowEntity.setType(recordInPageDto.getType());
         }
         dappFundFlowEntity.setMemberId(member.getId());
-        dappFundFlowEntity.setStatus(2);
+//        dappFundFlowEntity.setStatus(2);
 
         IPage<DappFundFlowEntity> records = dappFundFlowDao.selectInPage(page, dappFundFlowEntity);
         return records.getRecords();
@@ -122,6 +141,11 @@
     public Long transfer(TransferDto transferDto) {
         DappMemberEntity member = LoginUserUtil.getAppUser();
 
+        String upgrade = redisUtils.getString("APP_UPGRADE");
+        if ("upgrade".equals(upgrade)) {
+            throw new FebsException("功能升级中");
+        }
+
         if (transferDto.getType() != 2) {
             member = dappMemberDao.selectById(member.getId());
             if (member.getActiveStatus() == 1) {
@@ -135,12 +159,12 @@
                 throw new FebsException("Balance Not Enough");
             }
 
-            updateWalletCoinWithLock(transferDto.getAmount().negate(), member.getId());
+            updateWalletCoinWithLock(transferDto.getAmount(), member.getId(), 2);
 
             DappFundFlowEntity fundFlow = new DappFundFlowEntity(member.getId(), transferDto.getAmount().negate(), 1, 2, transferDto.getFee(), transferDto.getTxHash());
             dappFundFlowDao.insert(fundFlow);
 
-            chainProducer.sendAchieveTreeMsg(member.getId());
+//            chainProducer.sendAchieveTreeMsg(member.getId());
         } else {
             int type = 1;
             // 1-认购 2-充值tfc
@@ -169,16 +193,19 @@
 
     @Override
     public BigDecimal calPrice(PriceDto priceDto) {
-        String priceStr = redisUtils.getString(AppContants.REDIS_KEY_TFC_NEW_PRICE);
+//        String priceStr = redisUtils.getString(AppContants.REDIS_KEY_TFC_NEW_PRICE);
+        DataDictionaryCustom symbolPrice = dataDictionaryCustomMapper.selectDicDataByTypeAndCode(DataDictionaryEnum.SYMBOL_PRICE.getType(), DataDictionaryEnum.SYMBOL_PRICE.getCode());
+        DataDictionaryCustom serviceFeeDic = dataDictionaryCustomMapper.selectDicDataByTypeAndCode(DataDictionaryEnum.WITHDRAW_SERVICE_FEE.getType(), DataDictionaryEnum.WITHDRAW_SERVICE_FEE.getCode());
 
         BigDecimal amount = priceDto.getAmount();
         if (priceDto.getAmount() == null) {
             amount = BigDecimal.ZERO;
         }
-        return amount.multiply(new BigDecimal("0.1")).divide(new BigDecimal(priceStr), 2, RoundingMode.HALF_UP);
+        return amount.multiply(new BigDecimal(serviceFeeDic.getValue()).divide(BigDecimal.valueOf(100), 8, RoundingMode.HALF_DOWN)).divide(new BigDecimal(symbolPrice.getValue()), 2, RoundingMode.HALF_UP);
     }
 
     @Override
+    @Transactional(rollbackFor = Exception.class)
     public void withdraw(WithdrawDto withdrawDto) {
         DappMemberEntity member = LoginUserUtil.getAppUser();
 
@@ -189,11 +216,12 @@
 
         DappWalletMineEntity walletMine = dappWalletMineDao.selectByMemberId(member.getId());
         if (walletMine.getAvailableAmount().compareTo(withdrawDto.getFee()) < 0) {
-            throw new FebsException("TFC Not Enough");
+            throw new FebsException("DMD Not Enough");
         }
 
-        updateWalletCoinWithLock(withdrawDto.getAmount().negate(), member.getId());
-        updateWalletMineWithLock(withdrawDto.getFee().negate(), member.getId());
+        updateWalletCoinWithLock(withdrawDto.getAmount(), member.getId(), 2);
+        updateWalletMineWithLock(withdrawDto.getFee(), member.getId(), 2);
+
 
         DappFundFlowEntity feeFlow = new DappFundFlowEntity(member.getId(), withdrawDto.getFee().negate(), 7, 2, null, null);
         dappFundFlowDao.insert(feeFlow);
@@ -201,24 +229,46 @@
         DappFundFlowEntity fundFlow = new DappFundFlowEntity(member.getId(), withdrawDto.getAmount().negate(), 5, 2, withdrawDto.getFee(), null);
         dappFundFlowDao.insert(fundFlow);
 
+        MemberCoinWithdrawEntity memberCoinWithdraw = new MemberCoinWithdrawEntity();
+        memberCoinWithdraw.setMemberId(member.getId());
+        memberCoinWithdraw.setAddress(member.getAddress());
+        memberCoinWithdraw.setAmount(withdrawDto.getAmount());
+        memberCoinWithdraw.setFeeAmount(withdrawDto.getFee());
+        memberCoinWithdraw.setStatus(MemberCoinWithdrawEntity.STATUS_YES);
+        memberCoinWithdraw.setSymbol("USDT");
+        memberCoinWithdraw.setFlowId(fundFlow.getId());
+        memberCoinWithdrawDao.insert(memberCoinWithdraw);
+
         String hash = ChainService.getInstance(ChainEnum.BSC_USDT.name()).transfer(member.getAddress(), withdrawDto.getAmount());
         fundFlow.setToHash(hash);
         dappFundFlowDao.updateById(fundFlow);
 
-        Map<String, Object> map = new HashMap<>();
-        map.put("fee", withdrawDto.getFee());
-        map.put("address", ChainEnum.BSC_TFC_FEE.getAddress());
-        chainProducer.sendTfcFee(JSONObject.toJSONString(map));
+        chainProducer.sendFeeDistributeMsg(fundFlow.getId());
+//        Map<String, Object> map = new HashMap<>();
+//        map.put("fee", withdrawDto.getFee());
+//        map.put("address", ChainEnum.BSC_TFC_FEE.getAddress());
+//        map.put("flow", feeFlow.getId());
+//        chainProducer.sendTfcFee(JSONObject.toJSONString(map));
     }
 
     @Override
-    public void updateWalletCoinWithLock(BigDecimal amount, Long memberId) {
+    public void updateWalletCoinWithLock(BigDecimal amount, Long memberId, int type) {
         boolean isSuccess = false;
         while(!isSuccess) {
             DappWalletCoinEntity walletCoin = dappWalletCoinDao.selectByMemberId(memberId);
 
-            walletCoin.setTotalAmount(walletCoin.getTotalAmount().add(amount));
-            walletCoin.setAvailableAmount(walletCoin.getAvailableAmount().add(amount));
+            // 加
+            if (type == 1) {
+                walletCoin.setTotalAmount(walletCoin.getTotalAmount().add(amount));
+                walletCoin.setAvailableAmount(walletCoin.getAvailableAmount().add(amount));
+                // 减
+            } else {
+                if (amount.compareTo(walletCoin.getAvailableAmount()) > 0) {
+                    throw new FebsException("Balance Not Enough");
+                }
+                walletCoin.setTotalAmount(walletCoin.getTotalAmount().subtract(amount));
+                walletCoin.setAvailableAmount(walletCoin.getAvailableAmount().subtract(amount));
+            }
 
             int i = dappWalletCoinDao.updateWithLock(walletCoin);
             if (i > 0) {
@@ -228,13 +278,21 @@
     }
 
     @Override
-    public void updateWalletMineWithLock(BigDecimal amount, Long memberId) {
+    public void updateWalletMineWithLock(BigDecimal amount, Long memberId, int type) {
         boolean isSuccess = false;
         while(!isSuccess) {
             DappWalletMineEntity walletMine = dappWalletMineDao.selectByMemberId(memberId);
 
-            walletMine.setTotalAmount(walletMine.getTotalAmount().add(amount));
-            walletMine.setAvailableAmount(walletMine.getAvailableAmount().add(amount));
+            if (type == 1) {
+                walletMine.setTotalAmount(walletMine.getTotalAmount().add(amount));
+                walletMine.setAvailableAmount(walletMine.getAvailableAmount().add(amount));
+            } else {
+                if (amount.compareTo(walletMine.getAvailableAmount()) > 0) {
+                    throw new FebsException("DMD Not Enough");
+                }
+                walletMine.setTotalAmount(walletMine.getTotalAmount().subtract(amount));
+                walletMine.setAvailableAmount(walletMine.getAvailableAmount().subtract(amount));
+            }
 
             int i = dappWalletMineDao.updateWithLock(walletMine);
             if (i > 0) {
@@ -242,4 +300,239 @@
             }
         }
     }
+
+    @Override
+    public void addFrozenAmountWithLock(BigDecimal amount, Long memberId) {
+        boolean isSuccess = false;
+        while(!isSuccess) {
+            DappWalletCoinEntity walletCoin = dappWalletCoinDao.selectByMemberId(memberId);
+
+            walletCoin.setTotalAmount(walletCoin.getTotalAmount().add(amount));
+            walletCoin.setFrozenAmount(walletCoin.getFrozenAmount().add(amount));
+
+            int i = dappWalletCoinDao.updateWithLock(walletCoin);
+            if (i > 0) {
+                isSuccess = true;
+
+                DappFundFlowEntity frozenAmount = new DappFundFlowEntity(memberId, amount, 9, 2, null, null);
+                dappFundFlowDao.insert(frozenAmount);
+            }
+        }
+    }
+
+    @Override
+    public void releaseFrozenAmountWithLock(Long memberId) {
+        boolean isSuccess = false;
+        while(!isSuccess) {
+            DappWalletCoinEntity walletCoin = dappWalletCoinDao.selectByMemberId(memberId);
+
+            if (walletCoin.getFrozenAmount().compareTo(BigDecimal.ZERO) <= 0) {
+                return;
+            }
+            BigDecimal frozen = walletCoin.getFrozenAmount();
+
+            walletCoin.setAvailableAmount(walletCoin.getAvailableAmount().add(frozen));
+            walletCoin.setFrozenAmount(walletCoin.getFrozenAmount().subtract(frozen));
+
+            int i = dappWalletCoinDao.updateWithLock(walletCoin);
+            if (i > 0) {
+                isSuccess = true;
+
+                DappFundFlowEntity releaseFrozen = new DappFundFlowEntity(memberId, frozen, 10, 2, null, null);
+                dappFundFlowDao.insert(releaseFrozen);
+
+                DappFundFlowEntity releaseFrozenNegate = new DappFundFlowEntity(memberId, frozen.negate(), 9, 2, null, null);
+                dappFundFlowDao.insert(releaseFrozenNegate);
+            }
+        }
+    }
+
+    @Override
+    public DappWalletCoinEntity findByMemberId(Long memberId) {
+        return dappWalletCoinDao.selectByMemberId(memberId);
+    }
+
+    @Override
+    public Long buyNode(BuyNodeDto transferDto) {
+        DappMemberEntity member = LoginUserUtil.getAppUser();
+
+        String upgrade = redisUtils.getString("APP_UPGRADE");
+        if ("upgrade".equals(upgrade)) {
+            throw new FebsException("功能升级中");
+        }
+
+        member = dappMemberDao.selectById(member.getId());
+//        int buyNode = ObjectUtil.isEmpty(member.getBuyNode()) ? 0 : 1;
+        int buyNode = member.getBuyNode();
+        DataDictionaryCustom buyNodeCodeCntDic = dataDictionaryCustomMapper.selectDicDataByTypeAndCode(
+                DataDictionaryEnum.BUY_NODE_CNT.getType(),
+                DataDictionaryEnum.BUY_NODE_CNT.getCode()
+        );
+        if (buyNode >= Integer.parseInt(buyNodeCodeCntDic.getValue())) {
+            throw new FebsException("Do not repeat purchase");
+        }
+
+        if (ObjectUtil.isEmpty(transferDto.getId())) {
+            DappFundFlowEntity fundFlow = new DappFundFlowEntity(
+                    member.getId(),
+                    transferDto.getAmount().negate(),
+                    13,
+                    1,
+                    BigDecimal.ZERO,
+                    transferDto.getTxHash());
+            dappFundFlowDao.insert(fundFlow);
+            return fundFlow.getId();
+        } else {
+            DappFundFlowEntity flow = dappFundFlowDao.selectById(transferDto.getId());
+            if(ObjectUtil.isEmpty(flow)){
+                return null;
+            }
+            if ("success".equals(transferDto.getFlag())) {
+                flow.setFromHash(transferDto.getTxHash());
+                dappFundFlowDao.updateById(flow);
+
+            } else {
+                if (flow.getStatus() == 1) {
+                    dappFundFlowDao.deleteById(transferDto.getId());
+                }
+            }
+        }
+        return null;
+    }
+
+    @Override
+    public Long buyStar(BuyStarDto buyStarDto) {
+        DappMemberEntity member = LoginUserUtil.getAppUser();
+
+        String upgrade = redisUtils.getString("APP_UPGRADE");
+        if ("upgrade".equals(upgrade)) {
+            throw new FebsException("功能升级中");
+        }
+
+        /**
+         * 买入先验证当前星团是否已经买入
+         *      如果不是购买一星,则验证前一新团是否购买
+         */
+        if(ObjectUtil.isEmpty(buyStarDto.getNodeType())){
+            throw new FebsException("选择购买的星团");
+        }
+        Integer nodeType = buyStarDto.getNodeType();
+        if(NodeType.NODE_13.getNodeType() < nodeType || nodeType < NodeType.NODE_1.getNodeType()){
+            throw new FebsException("选择购买的星团");
+        }
+        QueryWrapper<DbMemberNode> nodeQueryWrapper = new QueryWrapper<>();
+        nodeQueryWrapper.eq("member_id",member.getId());
+        nodeQueryWrapper.eq("type", nodeType);
+        nodeQueryWrapper.eq("work_state", DbMemberNode.STATE_ONE);
+        List<DbMemberNode> dbMemberNodes = dbMemberNodeMapper.selectList(nodeQueryWrapper);
+        if(CollUtil.isNotEmpty(dbMemberNodes)){
+            throw new FebsException("已购买");
+        }
+        if(NodeType.NODE_1.getNodeType() != buyStarDto.getNodeType()){
+            Integer nextNodeType = nodeType - 1;
+            QueryWrapper<DbMemberNode> nextWrapper = new QueryWrapper<>();
+            nextWrapper.eq("member_id",member.getId());
+            nextWrapper.eq("type", nextNodeType);
+            nextWrapper.eq("work_state", DbMemberNode.STATE_ONE);
+            List<DbMemberNode> nextNode = dbMemberNodeMapper.selectList(nextWrapper);
+            if(CollUtil.isEmpty(nextNode)){
+                throw new FebsException("请先购买上一星团");
+            }
+        }
+
+        if (ObjectUtil.isEmpty(buyStarDto.getId())) {//第一次
+            DappFundFlowEntity fundFlow = new DappFundFlowEntity(
+                    member.getId(),
+                    buyStarDto.getAmount().negate(),
+                    18,
+                    1,
+                    BigDecimal.ZERO,
+                    buyStarDto.getTxHash());
+            dappFundFlowDao.insert(fundFlow);
+            return fundFlow.getId();
+        } else {//第二次
+            DappFundFlowEntity flow = dappFundFlowDao.selectById(buyStarDto.getId());
+            if(ObjectUtil.isEmpty(flow)){
+                return null;
+            }
+            if ("success".equals(buyStarDto.getFlag())) {
+                flow.setFromHash(buyStarDto.getTxHash());
+                flow.setToHash(buyStarDto.getNodeType().toString());
+                dappFundFlowDao.updateById(flow);
+            } else {
+                if (flow.getStatus() == 1) {
+                    dappFundFlowDao.deleteById(buyStarDto.getId());
+                }
+            }
+        }
+        return null;
+    }
+
+    @Override
+    public FebsResponse nodeList() {
+        List<NodeType> nodeList = NodeType.NODE_1.getNodeList();
+        List<NodeTypeVo> objects = new ArrayList<>();
+        if(CollUtil.isNotEmpty(nodeList)){
+            for(NodeType nodeType : nodeList){
+                NodeTypeVo nodeTypeVo = new NodeTypeVo();
+                nodeTypeVo.setNodeType(nodeType.getNodeType());
+                nodeTypeVo.setNodeAmount(nodeType.getNodeAmount());
+                nodeTypeVo.setNodePerk(nodeType.getNodePerk());
+                objects.add(nodeTypeVo);
+            }
+        }
+        return new FebsResponse().success().data(objects);
+    }
+
+    @Override
+    public MemberNodeVo memberNode(MemberNodeDto memberNodeDto) {
+        DappMemberEntity member = LoginUserUtil.getAppUser();
+        Integer type = memberNodeDto.getType();
+
+        QueryWrapper<DbMemberNode> objectQueryWrapper = new QueryWrapper<>();
+        objectQueryWrapper.eq("member_id",member.getId());
+        objectQueryWrapper.eq("type", type);
+        objectQueryWrapper.eq("work_state", DbMemberNode.STATE_ONE);
+        DbMemberNode dbMemberNode = dbMemberNodeMapper.selectOne(objectQueryWrapper);
+        if(ObjectUtil.isEmpty(dbMemberNode)){
+            return null;
+        }
+
+        return buildTeamMatrix(dbMemberNode);
+    }
+
+    public MemberNodeVo buildTeamMatrix(DbMemberNode dbMemberNode) {
+        Long id = dbMemberNode.getId();
+
+        MemberNodeVo memberNodeVo = new MemberNodeVo();
+        memberNodeVo.setInviteId(getInviteId(id));
+        memberNodeVo.setId(id);
+        memberNodeVo.setParentNode(ObjectUtil.isEmpty(dbMemberNode.getParentNode()) ? 0L : dbMemberNode.getParentNode());
+        memberNodeVo.setLeftNode(ObjectUtil.isEmpty(dbMemberNode.getLeftNode()) ? 0L : dbMemberNode.getLeftNode());
+        memberNodeVo.setRightNode(ObjectUtil.isEmpty(dbMemberNode.getRightNode()) ? 0L : dbMemberNode.getRightNode());
+        memberNodeVo.setLeftRight(ObjectUtil.isEmpty(dbMemberNode.getLeftRight()) ? 0 : dbMemberNode.getLeftRight());
+
+        QueryWrapper<DbMemberNode> childQuery = new QueryWrapper<>();
+        childQuery.eq("parent_node",id);
+        childQuery.eq("type", dbMemberNode.getType());
+        childQuery.eq("work_state", DbMemberNode.STATE_ONE);
+        List<DbMemberNode> childrenNode = dbMemberNodeMapper.selectList(childQuery);
+        if(CollUtil.isEmpty(childrenNode)){
+            return memberNodeVo;
+        }
+
+        List<MemberNodeVo> list = new ArrayList<>();
+        for (DbMemberNode childNode : childrenNode) {
+            list.add(buildTeamMatrix(childNode));
+        }
+        memberNodeVo.setChildren(list);
+        return memberNodeVo;
+    }
+
+    public String getInviteId(Long memberNodeId){
+        DbMemberNode dbMemberNode = dbMemberNodeMapper.selectById(memberNodeId);
+        Long memberId = dbMemberNode.getMemberId();
+        DappMemberEntity dappMemberEntity = dappMemberDao.selectById(memberId);
+        return dappMemberEntity.getInviteId();
+    }
 }

--
Gitblit v1.9.1