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 | 32 +++++++++++++++++++++++++++++++- 1 files changed, 31 insertions(+), 1 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 f97d719..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; } @@ -268,6 +276,19 @@ return; } if(inviteNum > dappMemberEntities.size()){ + return; + } + 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; + } + if(inviteNum > inviteNodes.size()){ return; } //任何星级的原来节点都要出局 @@ -375,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