From b60666a3fe9822277efd717380e619f08ac3ad9f Mon Sep 17 00:00:00 2001
From: KKSU <15274802129@163.com>
Date: Thu, 13 Jun 2024 11:42:23 +0800
Subject: [PATCH] 星级矩阵

---
 src/main/java/cc/mrbird/febs/dapp/service/impl/DappWalletServiceImpl.java |   75 +++++++++++++++++++++++++++++++++++++
 1 files changed, 74 insertions(+), 1 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 8a19784..c03d789 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,5 +1,6 @@
 package cc.mrbird.febs.dapp.service.impl;
 
+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.LoginUserUtil;
@@ -9,8 +10,11 @@
 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.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;
@@ -26,6 +30,7 @@
 
 import java.math.BigDecimal;
 import java.math.RoundingMode;
+import java.util.ArrayList;
 import java.util.List;
 
 /**
@@ -159,7 +164,7 @@
             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
@@ -444,4 +449,72 @@
         }
         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