From aa084fcf2f099ebcf1e2d7ae46a03a5f77f76949 Mon Sep 17 00:00:00 2001
From: xiangpei <xiangpei@timesnew.cn>
Date: 星期一, 03 三月 2025 03:00:34 +0800
Subject: [PATCH] 工程库调整
---
business/src/main/java/com/ycl/utils/TreeUtil.java | 43 +++++++++++++++++++++++++++++++++++++++++++
1 files changed, 43 insertions(+), 0 deletions(-)
diff --git a/business/src/main/java/com/ycl/utils/TreeUtil.java b/business/src/main/java/com/ycl/utils/TreeUtil.java
new file mode 100644
index 0000000..4ea7fc9
--- /dev/null
+++ b/business/src/main/java/com/ycl/utils/TreeUtil.java
@@ -0,0 +1,43 @@
+package com.ycl.utils;
+
+import com.ycl.domain.vo.ProjectEngineeringVO;
+
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+
+/**
+ * 鏋勫缓鏍戝伐鍏�
+ *
+ * @author锛歺p
+ * @date锛�2025/3/3 2:37
+ */
+public class TreeUtil {
+
+ /**
+ * 涓哄伐绋嬫瀯閫犳爲
+ *
+ * @param nodes
+ * @return
+ */
+ public static List<ProjectEngineeringVO> treeForProjectEng(List<ProjectEngineeringVO> nodes) {
+ Map<String, ProjectEngineeringVO> nodeMap = new HashMap<>();
+ for (ProjectEngineeringVO node : nodes) {
+ nodeMap.put(node.getId(), node);
+ }
+ List<ProjectEngineeringVO> roots = new ArrayList<>();
+ for (ProjectEngineeringVO node : nodes) {
+ if ("0".equals(node.getParent())) {
+ roots.add(node);
+ } else {
+ ProjectEngineeringVO parentNode = nodeMap.get(node.getParent());
+ if (parentNode != null) {
+ parentNode.getChildren().add(node);
+ }
+ }
+ }
+ return roots;
+ }
+
+}
--
Gitblit v1.8.0