Class TreeUtil

src/main/java/com/quick/boot/modules/common/util/TreeUtil.java:11–139

万能树工具类 - 提供树形结构构建和排序功能

 * 万能树工具类 - 提供树形结构构建和排序功能
 */
public class TreeUtil {

    /**
     * 构建树形结构（核心方法）
     *
     * @param list 原始数据列表
     * @param idGetter 节点ID获取函数
     * @param parentIdGetter 父节点ID获取函数
     * @param childrenSetter 子节点设置函数
     * @param comparator 节点排序比较器（可为null）
     * @param rootValues 根节点标识值集合（可为null）
     * @param <T> 节点数据类型
     * @param <R> ID数据类型
     * @return 构建完成的树形结构列表
     */
    public static <T, R> List<T> buildTree(
            List<T> list,
            Function<T, R> idGetter,
            Function<T, R> parentIdGetter,
            BiConsumer<T, List<T>> childrenSetter,
            Comparator<T> comparator,
            Set<R> rootValues) {

        // 空列表安全处理
        if (Objects.isNull(list) || list.isEmpty()) {
            return Collections.emptyList();
        }

        // 创建ID到节点的映射，用于快速查找节点
        Map<R, T> idMap = list.stream()
                .collect(Collectors.toMap(idGetter, node -> node, (a, b) -> a));

        // 处理根节点标识值（防止空指针）
        Set<R> effectiveRootValues = Objects.nonNull(rootValues) ? rootValues : Collections.emptySet();

        // 构建父ID到子节点列表的映射
        Map<R, List<T>> parentMap = new HashMap<>();
        list.forEach(node -> {
            R parentId = parentIdGetter.apply(node);
            // 使用computeIfAbsent确保每个父ID都有对应的列表
            parentMap.computeIfAbsent(parentId, k -> new ArrayList<>()).add(node);
        });

        // 检测循环引用（防止无限递归）
        detectCycle(parentMap, idGetter, idMap);

        // 存储最终结果（根节点列表）
        List<T> result = new ArrayList<>();

        // 遍历所有节点构建树结构
        list.forEach(node -> {
            R id = idGetter.apply(node);
            // 获取当前节点的子节点列表
            List<T> children = parentMap.getOrDefault(id, Collections.emptyList());

            // 对子节点进行排序（如果提供了比较器）
            if (!children.isEmpty() && Objects.nonNull(comparator)) {
                children.sort(comparator);