java Stream编写树结构

最新推荐文章于 2024-07-27 17:56:15 发布

weixin_48208008

最新推荐文章于 2024-07-27 17:56:15 发布

阅读量91

点赞数 2

文章标签： java windows python

本文链接：https://blog.csdn.net/weixin_48208008/article/details/138092239

版权

public class JobCategory extends BaseEntity
{
    private static final long serialVersionUID = 1L;

    /** 主键id */
    private Long id;

    /** 父级id */
    @Excel(name = "父级id")
    private Long parentId;

    /** 类别名称 */
    @Excel(name = "类别名称")
    private String categoryName;

    /** 类别描述 */
    @Excel(name = "类别描述")
    private String categoryDesc;

    /** 岗位类别状态（0正常 1停用） */
    @Excel(name = "岗位类别状态", readConverterExp = "0=正常,1=停用")
    private String status;

    /** 删除标志（0代表存在 2代表删除） */
    private String delFlag;

    private List<JobCategory> Children;

 public List<JobCategory> queryTreeList(JobCategory jobCategory) {
        List<JobCategory> list = jobCategoryMapper.selectJobCategoryList(jobCategory);
        //获取父节点
        return list.stream().filter(m -> m.getParentId() == 0).peek(
                (m) -> m.setChildren(getChildrenList(m, list))
        ).collect(Collectors.toList());
    }

    /**
     * 获取子节点列表
     * @param tree tree
     * @param list list
     * @return List
     */
    public static List<JobCategory> getChildrenList(JobCategory tree, List<JobCategory> list){
        return  list.stream().filter(item -> Objects.equals(item.getParentId(), tree.getId())).peek(
                (item) -> item.setChildren(getChildrenList(item, list))
        ).collect(Collectors.toList());
    }