From 96f0cfd7592d22eccf48facb1efba79803da2f87 Mon Sep 17 00:00:00 2001
From: shenghao <shenghao.zhang@versa-ai.com>
Date: Thu, 16 Dec 2021 17:55:57 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96=EF=BC=9A=E5=87=8F=E5=B0=91?=
 =?UTF-8?q?=E8=8F=9C=E5=8D=95=E6=A0=91=E7=9A=84build=E6=AC=A1=E6=95=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../java/vip/xiaonuo/core/factory/TreeBuildFactory.java  | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/snowy-base/snowy-core/src/main/java/vip/xiaonuo/core/factory/TreeBuildFactory.java b/snowy-base/snowy-core/src/main/java/vip/xiaonuo/core/factory/TreeBuildFactory.java
index fa275349..33ebceb1 100644
--- a/snowy-base/snowy-core/src/main/java/vip/xiaonuo/core/factory/TreeBuildFactory.java
+++ b/snowy-base/snowy-core/src/main/java/vip/xiaonuo/core/factory/TreeBuildFactory.java
@@ -75,8 +75,8 @@ public class TreeBuildFactory<T extends BaseTreeNode> {
         if (ObjectUtil.isNotEmpty(nodeSubLists)) {
             nodeSubLists.forEach(t -> this.buildChildNodes(totalNodes, t, CollectionUtil.newArrayList()));
         }
-        childNodeLists.addAll(nodeSubLists);
-        node.setChildren(childNodeLists);
+//        childNodeLists.addAll(nodeSubLists);
+        node.setChildren(nodeSubLists);
     }
 
     /**
@@ -104,8 +104,9 @@ public class TreeBuildFactory<T extends BaseTreeNode> {
      * @date 2020/4/5 14:13
      */
     private List<T> executeBuilding(List<T> nodes) {
-        nodes.forEach(t -> this.buildChildNodes(nodes, t, CollectionUtil.newArrayList()));
-        return nodes;
+        List<T> parentNodes = afterBuild(nodes);
+        parentNodes.forEach(t -> this.buildChildNodes(nodes, t, CollectionUtil.newArrayList()));
+        return parentNodes;
     }
 
     /**