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

---
 src/main/java/cc/mrbird/febs/dapp/service/impl/DappSystemServiceImpl.java |  119 +++++++++++++++++++++++++++++++++++++++--------------------
 1 files changed, 78 insertions(+), 41 deletions(-)

diff --git a/src/main/java/cc/mrbird/febs/dapp/service/impl/DappSystemServiceImpl.java b/src/main/java/cc/mrbird/febs/dapp/service/impl/DappSystemServiceImpl.java
index a63588c..338eb57 100644
--- a/src/main/java/cc/mrbird/febs/dapp/service/impl/DappSystemServiceImpl.java
+++ b/src/main/java/cc/mrbird/febs/dapp/service/impl/DappSystemServiceImpl.java
@@ -28,6 +28,7 @@
 import java.math.RoundingMode;
 import java.util.Date;
 import java.util.List;
+import java.util.stream.Collectors;
 
 /**
  * @author
@@ -218,8 +219,12 @@
 
         //如果添加的为右节点,那么进入判断
         if(DbMemberNode.STATE_TWO == dbMemberNode.getLeftRight()){
+            //更新父节点的右节点
+            dbMemberNodeOld.setRightNode(dbMemberNode.getId());
+            dbMemberNodeMapper.updateById(dbMemberNodeOld);
+
             //父节点为右节点,那么进入判断,父节点是否有上级节点
-            int leftRight = ObjectUtil.isEmpty(dbMemberNodeOld.getLeftNode()) ? 1 : 2;
+            int leftRight = ObjectUtil.isEmpty(dbMemberNodeOld.getLeftRight()) ? 1 : dbMemberNodeOld.getLeftRight();
             if(DbMemberNode.STATE_TWO == leftRight){
                 if(ObjectUtil.isEmpty(dbMemberNodeOld.getParentNode())){
                     return;
@@ -239,6 +244,9 @@
                 return;
             }
         }else{
+            //更新父节点的左节点
+            dbMemberNodeOld.setLeftNode(dbMemberNode.getId());
+            dbMemberNodeMapper.updateById(dbMemberNodeOld);
             //如果添加的为左节点,那么没人出局
             return;
         }
@@ -250,10 +258,12 @@
      */
     public void memberNodeNext(DbMemberNode dbMemberNodeRoot) {
         log.info("进入复投");
+
         DataDictionaryCustom inviteNumDic = dataDictionaryCustomMapper.selectDicDataByTypeAndCode(
                 DataDictionaryEnum.INVITE_NUM.getType(),
                 DataDictionaryEnum.INVITE_NUM.getCode()
         );
+
         int inviteNum = Integer.parseInt(inviteNumDic.getValue());
 
         Long memberId = dbMemberNodeRoot.getMemberId();
@@ -268,49 +278,67 @@
         if(inviteNum > dappMemberEntities.size()){
             return;
         }
-        Integer countFund = dbMemberNodeRoot.getCountFund();
-        Integer type = dbMemberNodeRoot.getType();
-        //如果是13星局,则直接复投产生收益
-        if(NodeType.NODE_13.getNodeType() == type){
-            //复投
-            futouSixTeen(memberId,dbMemberNodeRoot.getAmount(),type,countFund+1);
-            //产生收益
-            BigDecimal perkAmount = dbMemberNodeRoot.getAmount().multiply(new BigDecimal("1.4"));
-            perkSevenTeen(memberId,perkAmount,countFund);
+        List<Long> collect = dappMemberEntities.stream().map(DappMemberEntity::getId).collect(Collectors.toList());
+        QueryWrapper<DbMemberNode> inviteNodeQuery = new QueryWrapper<>();
+        inviteNodeQuery.select("member_id");
+        inviteNodeQuery.eq("work_state", DbMemberNode.STATE_ONE);
+        inviteNodeQuery.in("member_id",collect);
+        inviteNodeQuery.groupBy("member_id");
+        List<DbMemberNode> inviteNodes = dbMemberNodeMapper.selectList(inviteNodeQuery);
+        if(CollUtil.isEmpty(inviteNodes)){
             return;
         }
-        //如果是1到12星团
-        if(type >=NodeType.NODE_1.getNodeType() && type <= NodeType.NODE_12.getNodeType()){
-            //需要复投本轮,
-            //如果是第一轮,当下一个星团没有投入时,收益复投下一个新团
-            //如果是第一轮,下一个新团有投入,则产生收益
-            //复投或者收益的金额
-            BigDecimal nodeAmount = dbMemberNodeRoot.getAmount().multiply(new BigDecimal("1.4"));
-            if(1 == countFund){//第一轮
-                //复投本轮,轮数 +1
-                futouSixTeen(memberId,dbMemberNodeRoot.getAmount(),type,countFund+1);
-                //收益复投(当前的下一个星团没有已投入),还是收益提出
-                Integer nextType = type + 1;
-                QueryWrapper<DbMemberNode> nodeQueryWrapper = new QueryWrapper<>();
-                nodeQueryWrapper.eq("member_id",memberId);
-                nodeQueryWrapper.eq("type", nextType);
-                nodeQueryWrapper.eq("work_state", DbMemberNode.STATE_ONE);
-                DbMemberNode dbMemberNode = dbMemberNodeMapper.selectOne(nodeQueryWrapper);
-                if(ObjectUtil.isEmpty(dbMemberNode)){
-                    futouSixTeen(memberId,nodeAmount,nextType,1);
-                }else{
-                    perkSevenTeen(memberId,nodeAmount,countFund);
-                }
-            }else if(countFund <= 9 && countFund >= 2){//2-9轮
-                //复投本轮,轮数 +1
-                futouSixTeen(memberId,dbMemberNodeRoot.getAmount(),type,countFund+1);
-                perkSevenTeen(memberId,nodeAmount,countFund);
-            }else{//10轮,收益本金都返回
-                BigDecimal add = dbMemberNodeRoot.getAmount().add(nodeAmount);
-                perkSevenTeen(memberId,add,countFund);
+        if(inviteNum > inviteNodes.size()){
+            return;
+        }
+        //任何星级的原来节点都要出局
+        dbMemberNodeRoot.setWorkState(DbMemberNode.STATE_TWO);
+        dbMemberNodeMapper.updateById(dbMemberNodeRoot);
+        //任何星级,10轮,收益本金都返回
+        Integer countFund = dbMemberNodeRoot.getCountFund();
+        //收益
+        BigDecimal perkAmount = dbMemberNodeRoot.getAmount().multiply(new BigDecimal("1.4"));
+        if(10 == countFund){//任何星级,10轮,收益本金都返回
+            BigDecimal add = dbMemberNodeRoot.getAmount().add(perkAmount);
+            perkSevenTeen(memberId,add,countFund);
 
-                dbMemberNodeRoot.setWorkState(DbMemberNode.STATE_TWO);
-                dbMemberNodeMapper.updateById(dbMemberNodeRoot);
+            return;
+        }else{
+            Integer type = dbMemberNodeRoot.getType();
+            //如果是13星局,则直接复投产生收益
+            if(NodeType.NODE_13.getNodeType() == type){
+                //复投
+                futouSixTeen(memberId,dbMemberNodeRoot.getAmount(),type,countFund+1);
+                //产生收益
+                perkSevenTeen(memberId,perkAmount,countFund);
+                return;
+            }
+            //如果是1到12星团
+            if(type >=NodeType.NODE_1.getNodeType() && type <= NodeType.NODE_12.getNodeType()){
+                //需要复投本轮,
+                //如果是第一轮,当下一个星团没有投入时,收益复投下一个新团
+                //如果是第一轮,下一个新团有投入,则产生收益
+                //复投或者收益的金额
+                if(1 == countFund){//第一轮
+                    //复投本轮,轮数 +1
+                    futouSixTeen(memberId,dbMemberNodeRoot.getAmount(),type,countFund+1);
+                    //收益复投(当前的下一个星团没有已投入),还是收益提出
+                    Integer nextType = type + 1;
+                    QueryWrapper<DbMemberNode> nodeQueryWrapper = new QueryWrapper<>();
+                    nodeQueryWrapper.eq("member_id",memberId);
+                    nodeQueryWrapper.eq("type", nextType);
+                    nodeQueryWrapper.eq("work_state", DbMemberNode.STATE_ONE);
+                    DbMemberNode dbMemberNode = dbMemberNodeMapper.selectOne(nodeQueryWrapper);
+                    if(ObjectUtil.isEmpty(dbMemberNode)){
+                        futouSixTeen(memberId,perkAmount,nextType,1);
+                    }else{
+                        perkSevenTeen(memberId,perkAmount,countFund);
+                    }
+                }else{//2-9轮
+                    //复投本轮,轮数 +1
+                    futouSixTeen(memberId,dbMemberNodeRoot.getAmount(),type,countFund+1);
+                    perkSevenTeen(memberId,perkAmount,countFund);
+                }
             }
         }
     }
@@ -368,6 +396,15 @@
         }
         dbMemberNode.setWorkState(1);
         dbMemberNodeMapper.insert(dbMemberNode);
+
+        if(ObjectUtil.isNotEmpty(dbMemberNodeOld)){
+            if(dbMemberNode.getLeftRight() == 1){
+                dbMemberNodeOld.setLeftNode(dbMemberNode.getId());
+            }else{
+                dbMemberNodeOld.setRightNode(dbMemberNode.getId());
+            }
+            dbMemberNodeMapper.updateById(dbMemberNodeOld);
+        }
     }
 
 

--
Gitblit v1.9.1