javaweb 菜单树

Vo

public class MenuVo {
    private  String  id ; //主键
    private  String parentId; //上层id
    private  String  name; //菜单名字
    private List<MenuVo> vos; //子节点


    public String getId() {
        return id;
    }

    public void setId(String id) {
        this.id = id;
    }

    public String getParentId() {
        return parentId;
    }

    public void setParentId(String parentId) {
        this.parentId = parentId;
    }

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }

    public List<MenuVo> getVos() {
        return vos;
    }

    public void setVos(List<MenuVo> vos) {
        this.vos = vos;
    }
}

实现逻辑

     @Override
    public List<MenuVo> selectTree() {
        // 查询所有数据
        List<MenuVo> menuVos = this.sysMenuDao.selectAll();
        if (CollectionUtils.isEmpty(menuVos)){
            return new ArrayList<MenuVo>();
        }
        //最顶层菜单    MenuConstants.PARENT_ID = "1" 最顶层id为1 
        List<MenuVo> menuBase = menuVos.stream().filter(vo -> MenuConstants.PARENT_ID.equals(vo.getParentId())).collect(Collectors.toList());
        menuBase.stream().filter(vo -> !CollectionUtils.isEmpty(menuBase)).forEach(vo -> {
            List<MenuVo> menus = iterateMenus(menuVos, vo.getId());
            vo.setVos(menus);
        });
        return  menuBase;
    }

    private List<MenuVo> iterateMenus(List<MenuVo> menuVos, String id) {
        List<MenuVo>  result  = new ArrayList<>();
        menuVos.forEach(vo ->{
            String menuId = vo.getId();
            //获取菜单父id
            String parentId = vo.getParentId();
            if (StringUtils.isNotBlank(parentId)){
                 if (parentId.equals(id)) {
                     //递归查询当前子菜单的子菜单
                     List<MenuVo> iterateMenu = iterateMenus(menuVos,menuId);
                     vo.setVos(iterateMenu);
                     result.add(vo);
                 }
            }
        });
        return result;
    }