通过lambda实现菜单树形结构

public List<CategoryEntity> listWithTree() {
    //1、查出所有分类
    List<CategoryEntity> list = baseMapper.selectList(null);
    //2、组装树形结构
    List<CategoryEntity> level1Menus = list.stream()
            .filter(t -> t.getParentCid() == 0)
            .map((menu) -> {
                menu.setChildren(this.getChildren(menu,list));
                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<CategoryEntity> getChildren(CategoryEntity root,List<CategoryEntity> all){
    List<CategoryEntity> childrenList = all.stream()
            .filter(t -> t.getParentCid() == root.getCatId())
            .map(g -> {
                //找子菜单
                g.setChildren(getChildren(g,all));
                return g;
            })
            //菜单排序
            .sorted((menu1,menu2) -> {return (menu1.getSort() == null ? 0 : menu1.getSort()) - (menu2.getSort() == null ? 0 : menu2.getSort()) ;})
            .collect(Collectors.toList());

    return childrenList;
}

注意:主要要有catId也就是主键ID,和parentCId也就是父ID,还要有每条记录的sort顺序。
当前CategoryEntity实体类里有List children;用来存储子菜单的。

如果要用的话,jdk1.8及以上即可。

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值