树形菜单查询

实体类

@Document(collection = "api_menu")
public class ApiMenu implements Serializable {

    private static final long serialVersionUID = -5754220183776638354L;

    @Id
    private Long id;//接口菜单id，自增

    private String title;//菜单名称

    private Long parentId;//父id

    private String method;//请求方法

    private String path;//请求地址

    private Map<String,List<ParmsItem>> reqParmsMap;//请求参数

    private Map<String,String> resParmsMap;//返回参数

    private String menuType;//菜单类型 M:菜单 I:接口

    private Integer apiStatus;//接口状态 1：未完成 2：已完成

    private String desc;//接口描述|菜单描述

    private Integer sort;//显示顺序

    private String addUser;//添加人

    private Date addTime;//添加时间

    private String upUser;//更新人

    private Date upTime;//更新时间

    @Transient
    private List<ApiMenu> chirdren;
}

controller

/**
 * 查询树形菜单
 * @return
 */
@GetMapping("/listWithTree")
public List<ApiMenu> list() {
    List<ApiMenu> listWithTree= mongoDbService.listWithTree();
    return listWithTree;
}

实现层

/**
  * 查询树形菜单
  * @return
  */
 public List<ApiMenu> listWithTree() {
     //1.查出所有分类
     List<ApiMenu> apiMenulist = mongoTemplate.findAll(ApiMenu.class);

     //2.构成父子的树形结构
     List<ApiMenu> level1Menus = apiMenulist.stream().filter(apiMenu ->
             //查询所有的一级分类
             apiMenu.getParentId() == 0
     ).map(menu ->{
         //查询所有的一级分类下的子菜单
         menu.setChirdren(getChildrens(menu,apiMenulist));
         return menu;
     }).sorted((menu1,menu2)->{
         //排序
         return (menu1.getSort()==null ? 0: menu1.getSort()) - (menu2.getSort()==null ? 0: menu2.getSort());
     }).collect(Collectors.toList());

     return level1Menus;
 }

 /**
  * 递归查询所有菜单的子菜单
  *
  * @param root  一级分类菜单
  * @param all   所有菜单
  * @return
  */
 private List<ApiMenu> getChildrens(ApiMenu root, List<ApiMenu> all) {
     List<ApiMenu> collect = all.stream().filter(categoryEntity -> {
         return categoryEntity.getParentId() == root.getId();
     }).map(categoryEntity -> {
         categoryEntity.setChirdren(getChildrens(categoryEntity, all));
         return categoryEntity;
     }).sorted((menu1, menu2) -> {
         return (menu1.getSort() == null ? 0 : menu1.getSort()) - (menu2.getSort() == null ? 0 : menu2.getSort());
     }).collect(Collectors.toList());
     return collect;
 }

标签：菜单,return,String,List,private,查询,树形,getSort
来源： https://blog.csdn.net/qq_38923630/article/details/121151155