package com.ruoyi.common.utils;
|
|
import com.ruoyi.common.core.domain.entity.SysDept;
|
|
import java.util.*;
|
import java.util.function.BiConsumer;
|
import java.util.function.Function;
|
import java.util.stream.Collectors;
|
|
/**
|
* 树结构工具类：根据 parentId 构造树、一次性获取所有叶子节点。
|
* 核心方法为泛型实现，id / parentId / children 通过函数注入，可复用于任何节点类型。
|
*/
|
public class TreeUtils {
|
|
private TreeUtils() {
|
}
|
|
/* ================== 通用方法 ================== */
|
|
/**
|
* 根据 parentId 构造树，O(n) 一次遍历。
|
* 根节点为「父Id不在集合中」的节点；若不存在根节点（如全是自引用的环）则原样返回。
|
*
|
* @param nodes 节点列表
|
* @param idGetter 取节点id
|
* @param parentIdGetter 取父节点id
|
* @param childrenSetter 设置子节点列表
|
*/
|
public static <T, K> List<T> buildTree(
|
List<T> nodes,
|
Function<T, K> idGetter,
|
Function<T, K> parentIdGetter,
|
BiConsumer<T, List<T>> childrenSetter) {
|
if (nodes == null || nodes.isEmpty()) {
|
return new ArrayList<>();
|
}
|
// 按父Id分组，一次遍历拿到每个节点的子列表
|
Map<K, List<T>> byParent = nodes.stream().collect(Collectors.groupingBy(parentIdGetter));
|
for (T node : nodes) {
|
childrenSetter.accept(node, byParent.getOrDefault(idGetter.apply(node), new ArrayList<>()));
|
}
|
Set<K> ids = nodes.stream().map(idGetter).filter(Objects::nonNull).collect(Collectors.toSet());
|
List<T> roots = nodes.stream()
|
.filter(n -> !ids.contains(parentIdGetter.apply(n)))
|
.collect(Collectors.toList());
|
return roots.isEmpty() ? nodes : roots;
|
}
|
|
/**
|
* 一次性获取所有叶子节点：即「id 不出现在任何节点的 parentId 中」的节点，O(n) 一次遍历。
|
*
|
* @param nodes 节点列表
|
* @param idGetter 取节点id
|
* @param parentIdGetter 取父节点id
|
*/
|
public static <T, K> List<T> getLeafNodes(
|
List<T> nodes,
|
Function<T, K> idGetter,
|
Function<T, K> parentIdGetter) {
|
if (nodes == null || nodes.isEmpty()) {
|
return new ArrayList<>();
|
}
|
Set<K> parentIds = nodes.stream()
|
.map(parentIdGetter)
|
.filter(Objects::nonNull)
|
.collect(Collectors.toSet());
|
return nodes.stream()
|
.filter(n -> !parentIds.contains(idGetter.apply(n)))
|
.collect(Collectors.toList());
|
}
|
|
/* ================== SysDept 便捷方法 ================== */
|
|
/**
|
* 根据 SysDept.parentId 构造部门树
|
*/
|
public static List<SysDept> buildDeptTree(List<SysDept> depts) {
|
return buildTree(depts, SysDept::getDeptId, SysDept::getParentId, SysDept::setChildren);
|
}
|
|
/**
|
* 一次性获取所有部门叶子节点
|
*/
|
public static List<SysDept> getLeafDepts(List<SysDept> depts) {
|
return getLeafNodes(depts, SysDept::getHisDeptId, SysDept::getHisParentId);
|
}
|
}
|
