From 293533a11236c2027241827ddff246db3e37f4c8 Mon Sep 17 00:00:00 2001
From: wzy <wzy19931122ai@163.com>
Date: Fri, 26 Aug 2022 00:06:55 +0800
Subject: [PATCH] finish ahieveTree consumer
---
src/main/java/cc/mrbird/febs/dapp/service/impl/DappSystemServiceImpl.java | 76 ++++++++++++++++++++++++++++++++++++++
1 files changed, 76 insertions(+), 0 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 c6db697..033e34e 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
@@ -16,6 +16,8 @@
import cc.mrbird.febs.dapp.vo.SlipSettingVo;
import cc.mrbird.febs.job.SystemTradeJob;
import cc.mrbird.febs.rabbit.producer.ChainProducer;
+import cc.mrbird.febs.tree.MatrixTree;
+import cc.mrbird.febs.tree.MemberNode;
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.date.DateTime;
import cn.hutool.core.date.DateUnit;
@@ -61,6 +63,9 @@
private final DappSymbolDailyPriceMapper dappSymbolDailyPriceMapper;
private final DappMemberBoxRecordMapper dappMemberBoxRecordMapper;
private final DappWalletCoinDao dappWalletCoinDao;
+
+ private final DappAchieveTreeDao dappAchieveTreeDao;
+ private final DappAchieveMemberTreeDao dappAchieveMemberTreeDao;
@Override
@@ -727,7 +732,78 @@
}
@Override
+ @Transactional(rollbackFor = Exception.class)
public synchronized void achieveTree(Long memberId) {
+ DappMemberEntity member = dappMemberDao.selectById(memberId);
+ // 在大树中，插入当前节点
+ DappAchieveTreeEntity achieveTree = new DappAchieveTreeEntity();
+ achieveTree.setMidNode(memberId);
+ achieveTree.setValidState(1);
+ dappAchieveTreeDao.insert(achieveTree);
+
+ // 在内存树(大树)中插入当前节点，并返回父节点
+ MemberNode node = new MemberNode(member.getId(), member.getAddress(), member.getInviteId(), member.getRefererId());
+ MatrixTree tree = MatrixTree.getInstance();
+ MemberNode parentNode = tree.addNode(node);
+
+ // 创建该节点的矩阵
+ DappAchieveMemberTreeEntity achieveMemberTree = new DappAchieveMemberTreeEntity();
+ achieveMemberTree.setTreeNode(memberId);
+ achieveMemberTree.setTopNode(memberId);
+ achieveMemberTree.setDeep(1);
+ achieveMemberTree.setHasMoney(1);
+ achieveMemberTree.setParentNode(parentNode.getMemberId());
+ dappAchieveMemberTreeDao.insert(achieveMemberTree);
+
+ // 修改父节点在数据库中的左/右节点数据
+ DappAchieveTreeEntity treeMidNode = dappAchieveTreeDao.selectByMidNode(parentNode.getMemberId());
+ boolean isLeft = false;
+ if (parentNode.getLeft() != null && memberId.equals(parentNode.getLeft().getMemberId())) {
+ treeMidNode.setLeftNode(memberId);
+ isLeft = true;
+ } else {
+ treeMidNode.setRightNode(memberId);
+ }
+ dappAchieveTreeDao.updateById(treeMidNode);
+
+ // 更新矩阵中的数据
+ List<DappAchieveMemberTreeEntity> matrixNodes = dappAchieveMemberTreeDao.selectNotBottomNodeInMatrix(parentNode.getMemberId());
+ for (DappAchieveMemberTreeEntity matrixNode : matrixNodes) {
+ if (isLeft) {
+ matrixNode.setLeftNode(memberId);
+ } else {
+ matrixNode.setRightNode(memberId);
+ }
+
+ dappAchieveMemberTreeDao.updateById(matrixNode);
+
+ DappAchieveMemberTreeEntity newMatrixNode = new DappAchieveMemberTreeEntity();
+ newMatrixNode.setTreeNode(memberId);
+ newMatrixNode.setTopNode(matrixNode.getTopNode());
+ newMatrixNode.setParentNode(parentNode.getMemberId());
+ newMatrixNode.setHasMoney(1);
+ newMatrixNode.setDeep(matrixNode.getDeep() + 1);
+ dappAchieveMemberTreeDao.insert(newMatrixNode);
+
+ if (matrixNode.getDeep() == 2) {
+ finishMatrixTree(matrixNode.getTopNode());
+ }
+ }
+ }
+
+ // 完成矩阵树，并重置矩阵且重入
+ public void finishMatrixTree(Long memberId) {
+ List<DappAchieveMemberTreeEntity> matrixTree = dappAchieveMemberTreeDao.selectMatrixTreeByTopNode(memberId);
+ // 如果达到标准，则重置该矩阵树
+ if (matrixTree.size() == 7) {
+ dappAchieveMemberTreeDao.resetMatrixTree(memberId);
+ dappAchieveMemberTreeDao.reentryMoney(memberId);
+
+ DappAchieveMemberTreeEntity bottomNode = dappAchieveMemberTreeDao.selectNodeByDeep(memberId, 3);
+ if (bottomNode != null) {
+ finishMatrixTree(bottomNode.getTopNode());
+ }
+ }
}
}
--
Gitblit v1.9.1