全站最新Java练习项目合集!(附免费码课件)10款Java项目满足你各种需求!手把手教你开发超详细教程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; } }