递归查找树形状结(利用steam流的方式)leval值标明

最新推荐文章于 2024-02-27 20:22:41 发布

Harbor Lau

最新推荐文章于 2024-02-27 20:22:41 发布

阅读量412

点赞数

文章标签： java servlet mybatis

本文链接：https://blog.csdn.net/weixin_61503139/article/details/126962507

版权

1.bean(根据需求创建)

@Data
public class MenuTreeDto  {
    
    private String b00;
    private String parentId;
    private String b0101;
    private int level;
    private List<MenuTreeDto> children;

}

2.控制层

@RequestMapping("/list/tree")
public Result list(){

    List<MenuTreeDto> entities = dictService.listWithTree();


    return Result.ok(entities);
}

3.业务层

@Override
public List<MenuTreeDto> listWithTree() {
    //1、查出所有分类
    List<MenuTreeDto> entities = dictService.findAll();

    //2、组装成父子的树形结构

    //2.1）、找到所有的一级分类
    List<MenuTreeDto> level1Menus = entities.stream().filter(MenuTreeDto ->
            MenuTreeDto.getParentId().equals("aaa")
    ).map((menu)->{
        menu.setLevel(1);
        menu.setChildren(getChildrens(menu,entities));
        return menu;
    }).
    collect(Collectors.toList());

    return level1Menus;
}

private List<MenuTreeDto> getChildrens(MenuTreeDto root,List<MenuTreeDto> all){

    List<MenuTreeDto> children = all.stream().filter(MenuTreeDto -> {
        return MenuTreeDto.getParentId().equals(root.getB00());
    }).map(MenuTreeDto -> {
        //1、找到子菜单
        MenuTreeDto.setLevel(root.getLevel()+1);
        MenuTreeDto.setChildren(getChildrens(MenuTreeDto,all));
        return MenuTreeDto;
    })
    .collect(Collectors.toList());

    return children;
}

4.postman测试