商品类目树状图 -- 通过递归生成

素颜马尾好姑娘i 2022-11-20 10:57 190阅读 0赞

1、数据库相关结构

> CREATE TABLE \`pms\_category\` (  
>   \`cat\_id\` bigint(20) NOT NULL AUTO\_INCREMENT COMMENT '分类id',  
>   \`name\` char(50) DEFAULT NULL COMMENT '分类名称',  
>   \`parent\_cid\` bigint(20) DEFAULT NULL COMMENT '父分类id',  
>   \`cat\_level\` int(11) DEFAULT NULL COMMENT '层级',  
>   \`show\_status\` tinyint(4) DEFAULT NULL COMMENT '是否显示\[0-不显示，1显示\]',  
>   \`sort\` int(11) DEFAULT NULL COMMENT '排序',  
>   \`icon\` char(255) DEFAULT NULL COMMENT '图标地址',  
>   \`product\_unit\` char(50) DEFAULT NULL COMMENT '计量单位',  
>   \`product\_count\` int(11) DEFAULT NULL COMMENT '商品数量',  
>   PRIMARY KEY (\`cat\_id\`)  
> ) ENGINE=InnoDB AUTO\_INCREMENT=1433 DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4\_0900\_ai\_ci COMMENT='商品三级分类';

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM1Mjc1MjMz_size_16_color_FFFFFF_t_70][]

2、代码递归逻辑如下所示：

public List<CategoryEntity> listWithTree() {
            // 1、查出所有分类
            List<CategoryEntity> categoryEntityList = baseMapper.selectList(null);
    
            // 2、组装成父子的属性结构
            // 2.1 找到所有的一级分类
            categoryEntityList.stream().filter(categoryEntity -> 
                    categoryEntity.getParentCid() == 0
            ).map((menu) -> {
                menu.setChildren(getChildrens(menu, categoryEntityList));
                return menu;
            }).sorted((menu1, menus2) -> {
                return menu1.getSort() - menus2.getSort();
            }).collect(Collectors.toList());
    
            return categoryEntityList;
        }
    
        /**
         * 递归查询所有的科目子分类
         * @param root
         * @param all
         * @return
         */
        private List<CategoryEntity> getChildrens(CategoryEntity root, List<CategoryEntity> all) {
    
            List<CategoryEntity> children = all.stream().filter(categoryEntity -> {
                return categoryEntity.getParentCid() == root.getCatId();
            }).map(categoryEntity -> {
                // 1、找到子菜单
                categoryEntity.setChildren(getChildrens(categoryEntity, all));
                return categoryEntity;
            }).sorted((menu1, menu2) -> {
                // 2、菜单的排序
                return menu1.getSort() - menu2.getSort();
            }).collect(Collectors.toList());
            return children;
        }

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM1Mjc1MjMz_size_16_color_FFFFFF_t_70]: /images/20221120/100232488e7d4e9488c95d43a1de2a05.png