From 9b4433fddf5b401edb0aace8a404ac733b122702 Mon Sep 17 00:00:00 2001
From: 田源 <tianyuan@vci-tech.com>
Date: 星期四, 03 四月 2025 14:35:02 +0800
Subject: [PATCH] 添加非密字段显示
---
Source/BladeX-Tool/blade-core-tool/src/main/java/org/springblade/core/tool/node/ForestNodeManager.java | 85 ++++++++++++++++++++++++++++++++++++++++++
1 files changed, 85 insertions(+), 0 deletions(-)
diff --git a/Source/BladeX-Tool/blade-core-tool/src/main/java/org/springblade/core/tool/node/ForestNodeManager.java b/Source/BladeX-Tool/blade-core-tool/src/main/java/org/springblade/core/tool/node/ForestNodeManager.java
new file mode 100644
index 0000000..f1ff224
--- /dev/null
+++ b/Source/BladeX-Tool/blade-core-tool/src/main/java/org/springblade/core/tool/node/ForestNodeManager.java
@@ -0,0 +1,85 @@
+/*
+ * Copyright (c) 2018-2028, Chill Zhuang All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions are met:
+ *
+ * Redistributions of source code must retain the above copyright notice,
+ * this list of conditions and the following disclaimer.
+ * Redistributions in binary form must reproduce the above copyright
+ * notice, this list of conditions and the following disclaimer in the
+ * documentation and/or other materials provided with the distribution.
+ * Neither the name of the dreamlu.net developer nor the names of its
+ * contributors may be used to endorse or promote products derived from
+ * this software without specific prior written permission.
+ * Author: Chill 搴勯獮 (smallchill@163.com)
+ */
+package org.springblade.core.tool.node;
+
+import com.google.common.collect.ImmutableMap;
+import com.google.common.collect.Maps;
+import org.springblade.core.tool.utils.StringPool;
+
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Map;
+
+/**
+ * 妫灄绠＄悊绫�
+ *
+ * @author smallchill
+ */
+public class ForestNodeManager<T extends INode<T>> {
+
+ /**
+ * 妫灄鐨勬墍鏈夎妭鐐�
+ */
+ private final ImmutableMap<Long, T> nodeMap;
+
+ /**
+ * 妫灄鐨勭埗鑺傜偣ID
+ */
+ private final Map<Long, Object> parentIdMap = Maps.newHashMap();
+
+ public ForestNodeManager(List<T> nodes) {
+ nodeMap = Maps.uniqueIndex(nodes, INode::getId);
+ }
+
+ /**
+ * 鏍规嵁鑺傜偣ID鑾峰彇涓�涓妭鐐�
+ *
+ * @param id 鑺傜偣ID
+ * @return 瀵瑰簲鐨勮妭鐐瑰璞�
+ */
+ public INode<T> getTreeNodeAt(Long id) {
+ if (nodeMap.containsKey(id)) {
+ return nodeMap.get(id);
+ }
+ return null;
+ }
+
+ /**
+ * 澧炲姞鐖惰妭鐐笽D
+ *
+ * @param parentId 鐖惰妭鐐笽D
+ */
+ public void addParentId(Long parentId) {
+ parentIdMap.put(parentId, StringPool.EMPTY);
+ }
+
+ /**
+ * 鑾峰彇鏍戠殑鏍硅妭鐐�(涓�涓．鏋楀搴斿棰楁爲)
+ *
+ * @return 鏍戠殑鏍硅妭鐐归泦鍚�
+ */
+ public List<T> getRoot() {
+ List<T> roots = new ArrayList<>();
+ nodeMap.forEach((key, node) -> {
+ if (node.getParentId() == 0 || parentIdMap.containsKey(node.getId())) {
+ roots.add(node);
+ }
+ });
+ return roots;
+ }
+
+}
--
Gitblit v1.9.3