用 Java 写一个 List 转 Tree 工具类

最新推荐文章于 2024-10-30 16:08:43 发布

原创最新推荐文章于 2024-10-30 16:08:43 发布 · 259 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#java #jvm #servlet

java 专栏收录该内容

232 篇文章

订阅专栏

这篇博客分享了如何利用Java编写一个将List数据结构转换为Tree结构的工具类，适合Java初学者和开发者。文章包含核心代码展示和使用方法说明。

尚学堂给同学们带来全新的Java300集课程啦!java零基础小白自学Java必备优质教程_手把手图解学习Java，让学习成为一种享受_哔哩哔哩_bilibili

主要代码

package com.jianshu.utils;

import lombok.Data;
import lombok.extern.slf4j.Slf4j;

import java.util.List;
import java.util.stream.Collectors;

/**
 * @author fubo
 */
@Slf4j
public class TreeUtil {

    /**
     * @param list         全部节点列表
     * @param rootParentId 根节点ID
     * @param <T>          节点数据model
     * @return 树结构数据
     */
    public static <T extends TreeNodeDTO> List<T> listToTree(List<T> list, Long rootParentId) {
        return list.stream()
                .filter(treeNodeDTO -> treeNodeDTO.getParentId().equals(rootParentId))
                .peek(treeNodeDTO -> treeNodeDTO.setChildrenList(listToTree(list, treeNodeDTO.getId())))
                .collect(Collectors.toList());
    }

    /**
     * 树节点对象
     */
    @Data
    public static class TreeNodeDTO {

        /**
         * ID
         */
        private Long id;

        /**
         * 父级ID
         */
        private Long parentId;

        /**
         * 子节点列表
         */
        private List<? extends TreeNodeDTO> childrenList;
    }
}

使用方式

定义一个结果Model类

package com.jianshu.vo;

import com.jianshu.TreeUtil;
import lombok.Data;
import lombok.EqualsAndHashCode;

/**
 * @author fubo
 */
@Data
@EqualsAndHashCode(callSuper = true)
public class DemoTreeNode extends TreeUtil.TreeNodeDTO {

    /**
     *   节点字段1
     */
    private String column1;

    /**
     *   节点字段2
     */
    private String column2;
    
    // ...
}

调用代码

// 准备list数据
List<DemoTreeNode> list; 
// 计算tree数据
TreeUtil.listToTree(list, 0L);