From c881dcbb782d209f43b9a2878de613e0124e0421 Mon Sep 17 00:00:00 2001
From: KKSU <15274802129@163.com>
Date: Sun, 07 Jul 2024 12:53:00 +0800
Subject: [PATCH] 逻辑

---
 src/main/java/cc/mrbird/febs/tree/MatrixTree.java |   33 ++++++++++++++++++++++++++++++---
 1 files changed, 30 insertions(+), 3 deletions(-)

diff --git a/src/main/java/cc/mrbird/febs/tree/MatrixTree.java b/src/main/java/cc/mrbird/febs/tree/MatrixTree.java
index deea01b..dbaa294 100644
--- a/src/main/java/cc/mrbird/febs/tree/MatrixTree.java
+++ b/src/main/java/cc/mrbird/febs/tree/MatrixTree.java
@@ -1,10 +1,10 @@
 package cc.mrbird.febs.tree;
 
-import com.sun.jmx.remote.internal.ArrayQueue;
+import cn.hutool.core.util.RandomUtil;
+import cn.hutool.json.JSONUtil;
 import lombok.extern.slf4j.Slf4j;
 
 import java.util.ArrayDeque;
-import java.util.Queue;
 
 /**
  * @author wzy
@@ -155,7 +155,7 @@
             return null;
         }
 
-        if (node.getMemberId().equals(param) || node.getAddress().equals(param) || node.getInviteId().equals(param)) {
+        if (node.getFundId().equals(param) ||node.getMemberId().equals(param) || node.getAddress().equals(param) || node.getInviteId().equals(param)) {
             return node;
         }
 
@@ -166,4 +166,31 @@
 
         return getNode(node.getRight(), param);
     }
+
+    /**
+     * 移除树中节点
+     *
+     * @param param
+     */
+    public void remove(Object param) {
+    }
+
+    public static void main(String[] args) {
+        MatrixTree matrixTree = new MatrixTree();
+        String refererId = null;
+        for (int i = 0; i < 11; i++) {
+            String inviteId = RandomUtil.randomString(6);
+            if (i == 0) {
+                refererId = inviteId;
+            }
+            MemberNode memberNode = new MemberNode();
+            memberNode.setMemberId(Long.parseLong(i + 1 + ""));
+            memberNode.setInviteId(inviteId);
+            memberNode.setAddress(RandomUtil.randomString(14));
+            memberNode.setRefererId(refererId);
+            matrixTree.addNode(memberNode);
+        }
+        System.out.println(JSONUtil.parseObj(matrixTree));
+        System.out.println(System.currentTimeMillis());
+    }
 }

--
Gitblit v1.9.1