经常要使用的树型递归方法

分类繁多,要读取分类并按树型结构展示,需要使用递归方法。

    /**
     * 查出所有分类及子分类,以树型结构展示接口
     */
    @Override
    public List<CategoryEntity> listWithTree() {
        // 1.查出所有分类
        List<CategoryEntity> entities = baseMapper.selectList(null);
        // 2.组装树型结构
        // 2.1 找到所有的一级分类,使用 arr.stream().filter(obj) 进行过滤
        //     收集之前,使用 arr.stream().map(obj) 方法收集子分类列表
        List<CategoryEntity> parentList = entities.stream().filter((categoryEntity) -> {
            return categoryEntity.getParentCid() == 0;
        }).map((categoryEntity)->{
            // 递归寻找子分类
            categoryEntity.setChildren(getChildren(categoryEntity, entities));
            return categoryEntity;
        }).sorted((root1, root2)->{
            // 菜单排序
            return (root1.getSort()==null?0:root1.getSort()) - (root2.getSort()==null?0:root2.getSort());
        }).collect(Collectors.toList());

        return parentList;
    }

    /**
     * 递归方法获取子分类
     * 传入两个参数,一个是当前父节点currentRoot,一个是所有的节点allRoot
     */
    private List<CategoryEntity> getChildren(CategoryEntity currentRoot, List<CategoryEntity> allRoot){
        // 查找所有记录中父ID等于当前节点ID的记录
        List<CategoryEntity> children = allRoot.stream().filter((categoryEntity) -> {
            return categoryEntity.getParentCid() == currentRoot.getCatId();
        }).map((categoryEntity)->{
            // 递归寻找子分类
            categoryEntity.setChildren(getChildren(categoryEntity, allRoot));
            return categoryEntity;
        }).sorted((root1, root2) -> {
            // 菜单排序
            return (root1.getSort()==null?0:root1.getSort()) - (root2.getSort()==null?0:root2.getSort());
        }).collect(Collectors.toList());
        return children;
    }

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值