用 Java 写一个 List 转 Tree 工具类

• 主要代码

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;
    }
}

• 使用方式

1. 定义一个结果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;
    
    // ...
}

2. 调用代码

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