一个获取多级目录的小案例

电玩女神 2022-11-29 03:15 146阅读 0赞

### 文章目录 ###

*   *  1、需求
     *  2、代码实现

## 1、需求 ##

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3N1Y2hhaGFlcmthbmc_size_16_color_FFFFFF_t_70_pic_center]  
获取京东商城首页的三级菜单数据

## 2、代码实现 ##

使用springboot+mybatis实现  
1、controller

@RequestMapping("/list/tree")
        @ApiOperation(value = "查询三级分类、组装为分类树",notes = "前端通过key==data获取")
        public R list(){
            List<CategoryEntity> entities = categoryService.listWithTree();
            return R.ok().put("data", entities);
        }

2、service

/**
     * 组装分类树
     *
     * @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)
                //映射、得到所有子分类然后存到children集合内
                .map((menu) -> {
                    menu.setChildren(getChildrens(menu, entities));
                    return menu;
                })
                //排序、进行非空校验并升序排列
                .sorted((menu1, menu2) -> {
                    return (menu1.getSort() == null ? 0 : menu1.getSort()) -
                            (menu2.getSort() == null ? 0 : menu2.getSort());
                })
                //收集组装集合
                .collect(Collectors.toList());
        return level1Menus;
    }
    
    /**
     * 为所有的父分类找到他的子分类并映射到child集合里、
     *
     * @param root 父分类
     * @param all  原始集合数据
     * @return
     */
    private List<CategoryEntity> getChildrens(CategoryEntity root, List<CategoryEntity> all) {
        List<CategoryEntity> children = all.stream()
                //root为父分类、得到所有分类的父Id == root分类Id的数据
                .filter(categoryEntity -> {
                    return categoryEntity.getParentCid() == root.getCatId();
                })
                //映射、然后递归调用、
                .map(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;
    }

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3N1Y2hhaGFlcmthbmc_size_16_color_FFFFFF_t_70_pic_center]: /images/20221124/970878798db74d619896713748152dbf.png