Java8 对父子节点结构集合数据的处理

实体结构

public class PmsBdCategory extends ZblCrmBaseEntity {
    private static final long serialVersionUID = -8723749086394105890L;
    /** 
    * 雪花算法分布id
    */
    private Long id;
    /**
     * 上级ID，一级为0
     */
    private Long parentId;
    /**
     * 类别名称
     */
    private String name;
    /**
     * 类别简称
     */
    private String shortName;

递归获取某个父节点下面的所有子节点

根据某一个父节点的ID获取下面所有的子节点

/**
 * 递归查询子节点
 * @param childCategory
 * @param categoryList
 * @param pid
 */
private void getChildNode(List<PmsBdCategory> childCategory, List<PmsBdCategory> categoryList, Long pid) {
    categoryList.stream()
            .filter(item -> Objects.nonNull(pid) && Objects.equals(pid, item.getParentId()))
            .forEach(menu -> {
                getChildNode(childCategory, categoryList, menu.getId());
                childCategory.add(menu);
            });
}

根据子节点查询所有父节点

根据某一个节点的ID获取下面所有的子节点

/**
 * 递归查询父节点
 * @param childCategory
 * @param categoryList
 * @param id
 */
private void getParentNode(List<PmsBdCategory> childCategory, List<PmsBdCategory> categoryList, Long id) {
    categoryList.stream()
            .filter(item -> Objects.nonNull(id) && Objects.equals(id, item.getId()))
            .forEach(menu -> {
                getParentNode(childCategory, categoryList, menu.getParentId());
                childCategory.add(menu);
            });
}