From 5d5cd3ad979a0b85ab8c7034dcad1663fdcf71be Mon Sep 17 00:00:00 2001
From: linwe <872216996@qq.com>
Date: Fri, 01 Dec 2023 14:18:53 +0800
Subject: [PATCH] 文章通过范围查询

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

diff --git a/src/main/java/org/springblade/common/utils/NodeTreeUtil.java b/src/main/java/org/springblade/common/utils/NodeTreeUtil.java
index 88d8860..f616629 100644
--- a/src/main/java/org/springblade/common/utils/NodeTreeUtil.java
+++ b/src/main/java/org/springblade/common/utils/NodeTreeUtil.java
@@ -1,7 +1,8 @@
 package org.springblade.common.utils;
 
 import org.springblade.common.node.TreeIntegerNode;
-import org.springblade.common.node.TreeNode;
+import org.springblade.common.node.TreeLongNode;
+import org.springblade.common.node.TreeStringNode;
 import org.springblade.modules.doorplateAddress.vo.DoorplateAddressVOTree;
 import org.springblade.modules.house.vo.HouseTree;
 
@@ -40,6 +41,44 @@
 	 * @param treeMap
 	 * @return
 	 */
+	public static List<TreeLongNode> getLongNodeTree(Map<Long, TreeLongNode> treeMap){
+		List<TreeLongNode> tree = new ArrayList<>();
+		if (treeMap.size() > 1) {
+			treeMap.forEach((id, treeNode) -> {
+				if (treeMap.containsKey(treeNode.getParentId())) {
+					treeMap.get(treeNode.getParentId()).getChildren().add(treeNode);
+				} else {
+					tree.add(treeNode);
+				}
+			});
+		}
+		return tree;
+	}
+
+	/**
+	 * 树转换
+	 * @param treeMap
+	 * @return
+	 */
+	public static List<TreeStringNode> getStringNodeTree(Map<String, TreeStringNode> treeMap){
+		List<TreeStringNode> tree = new ArrayList<>();
+		if (treeMap.size() > 1) {
+			treeMap.forEach((id, treeNode) -> {
+				if (treeMap.containsKey(treeNode.getParentId())) {
+					treeMap.get(treeNode.getParentId()).getChildren().add(treeNode);
+				} else {
+					tree.add(treeNode);
+				}
+			});
+		}
+		return tree;
+	}
+
+	/**
+	 * 树转换
+	 * @param treeMap
+	 * @return
+	 */
 	public static List<DoorplateAddressVOTree> getAddressNodeTree(Map<String, DoorplateAddressVOTree> treeMap){
 		List<DoorplateAddressVOTree> tree = new ArrayList<>();
 		if (treeMap.size() > 1) {

--
Gitblit v1.9.3