/**
* 查询树形菜单
* @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;
}
菜单集合遍历