From 2bd69d1dca40d2fdf07f72e0117d2f0febde5359 Mon Sep 17 00:00:00 2001
From: zhongrj <646384940@qq.com>
Date: Sat, 23 Dec 2023 15:27:02 +0800
Subject: [PATCH] 新增民警楼盘,功能数据过滤

---
 src/main/java/org/springblade/common/utils/NodeTreeUtil.java |   18 ++++++++++++------
 1 files changed, 12 insertions(+), 6 deletions(-)

diff --git a/src/main/java/org/springblade/common/utils/NodeTreeUtil.java b/src/main/java/org/springblade/common/utils/NodeTreeUtil.java
index a81edf8..7568fd1 100644
--- a/src/main/java/org/springblade/common/utils/NodeTreeUtil.java
+++ b/src/main/java/org/springblade/common/utils/NodeTreeUtil.java
@@ -63,13 +63,19 @@
 	public static List<TreeStringNode> getStringNodeTree(Map<String, TreeStringNode> treeMap){
 		List<TreeStringNode> tree = new ArrayList<>();
 		if (treeMap.size() > 0) {
-			treeMap.forEach((id, treeNode) -> {
-				if (treeMap.containsKey(treeNode.getParentId())) {
-					treeMap.get(treeNode.getParentId()).getChildren().add(treeNode);
-				} else {
+			if (treeMap.size()==1){
+				treeMap.forEach((id, treeNode) -> {
 					tree.add(treeNode);
-				}
-			});
+				});
+			}else {
+				treeMap.forEach((id, treeNode) -> {
+					if (treeMap.containsKey(treeNode.getParentId())) {
+						treeMap.get(treeNode.getParentId()).getChildren().add(treeNode);
+					} else {
+						tree.add(treeNode);
+					}
+				});
+			}
 		}
 		return tree;
 	}

--
Gitblit v1.9.3