From f1d079f9c0b3d384090477ca54e0548373d62b57 Mon Sep 17 00:00:00 2001
From: zhongrj <646384940@qq.com>
Date: Thu, 16 Jan 2025 11:56:39 +0800
Subject: [PATCH] 配置修改
---
src/main/java/com/genersoft/iot/vmp/utils/node/ForestNodeManager.java | 68 ++++++++++++++++++++++++++++++++++
1 files changed, 68 insertions(+), 0 deletions(-)
diff --git a/src/main/java/com/genersoft/iot/vmp/utils/node/ForestNodeManager.java b/src/main/java/com/genersoft/iot/vmp/utils/node/ForestNodeManager.java
new file mode 100644
index 0000000..affce9d
--- /dev/null
+++ b/src/main/java/com/genersoft/iot/vmp/utils/node/ForestNodeManager.java
@@ -0,0 +1,68 @@
+package com.genersoft.iot.vmp.utils.node;
+
+import com.google.common.collect.ImmutableMap;
+import com.google.common.collect.Maps;
+
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Map;
+
+/**
+ * 森林管理类
+ *
+ * @author smallchill
+ */
+public class ForestNodeManager<T extends INode<T>> {
+
+ /**
+ * 森林的所有节点
+ */
+ private final ImmutableMap<String, T> nodeMap;
+
+ /**
+ * 森林的父节点ID
+ */
+ private final Map<String, Object> parentIdMap = Maps.newHashMap();
+
+ public ForestNodeManager(List<T> nodes) {
+ nodeMap = Maps.uniqueIndex(nodes, INode::getChannelId);
+ }
+
+ /**
+ * 根据节点ID获取一个节点
+ *
+ * @param id 节点ID
+ * @return 对应的节点对象
+ */
+ public INode<T> getTreeNodeAt(String id) {
+ if (nodeMap.containsKey(id)) {
+ return nodeMap.get(id);
+ }
+ return null;
+ }
+
+ /**
+ * 增加父节点ID
+ *
+ * @param parentId 父节点ID
+ */
+ public void addParentId(String parentId) {
+ parentIdMap.put(parentId, "");
+ }
+
+ /**
+ * 获取树的根节点(一个森林对应多颗树)
+ *
+ * @return 树的根节点集合
+ */
+ public List<T> getRoot() {
+ List<T> roots = new ArrayList<>();
+ nodeMap.forEach((key, node) -> {
+ if (node.getParentId() == null || parentIdMap.containsKey(node.getChannelId())) {
+ roots.add(node);
+ }
+ });
+ return roots;
+ }
+
+}
--
Gitblit v1.9.3