业务场景:实现系统的权限控制、多级的菜单展示,适用于子父级关系的
实现:使用Stream进行实现
调用递归方法进行展示
/**
* 查出所有分类,并且组装成父子结构
*
* 最终实现的效果就是:只发了一个sql语句,查出所有的菜单,再利用递归的方式来组合出父子菜单
* @return
*/
@Override
public List<CategoryEntity> listWithTree() {
//1、查出所有分类
List<CategoryEntity> entities = baseMapper.selectList(null);
//2、组装成父子的树形结构
//2.1)、找到所有的一级分类
List<CategoryEntity> level1Menus = entities.stream().filter(categoryEntity ->
categoryEntity.getParentCid() == 0
).map((menu)->{ //第一个参数是:将找到的当前菜单的所有子菜单,第二个参数是:在entities里面找到menu的子菜单
menu.setChildren(getChildrens(menu,entities)); //当前菜单改了以后,保存当前菜单的子分类
return menu; //把当前菜单返回回去
//如果把父菜单找到了,父菜单还需要进行排序
//如何排序?第一个参数是:传入前面的菜单,第二个参数是:传入后面的菜单,然后两个进行对比
}).sorted((menu1,menu2)->{ //将当前映射好的这些菜单进行排序
//每一个菜单都有Sort()这个方法,这个方法就是它的排序字段,让第一个的排序顺序减去第二个参数的排序顺序,给他们一个最终的升降序的值
//返回对比的结果
return (menu1.getSort()==null?0:menu1.getSort()) - (menu2.getSort()==null?0:menu2.getSort());
}).collect(Collectors.toList()); //将所有的数据都收集到level1Menus里面
return level1Menus;
}```
### 递归方法
/**
*递归查找所有菜单的子菜单
* @param root 当前菜单
* @param all 从哪里获取它的子菜单 (所有菜单)
* 从all里面找到这个root菜单的子菜单
* @return
*/
private List<CategoryEntity> getChildrens(CategoryEntity root,List<CategoryEntity> all){
//整个这个的菜单,就是我们要用的子菜单
List<CategoryEntity> children = all.stream().filter(categoryEntity -> {
//当前菜单的父ID如果等于指定的菜单的id,那么说明当前菜单就是它的子菜单
return categoryEntity.getParentCid() == root.getCatId();
//每一个子菜单可能还会有子菜单,包括这些子菜单还是需要排序的
}).map(categoryEntity -> { //将菜单里面每一个东西都重新映射一下
//第一个参数:当前的这个菜单,第二个参数:在all里面找到categoryEntity的子菜单
categoryEntity.setChildren(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 children;
}
总结于:谷粒商城中三级菜单展示