Java拼装获取树形菜单结构

r囧r小猫 2022-12-28 08:18 159阅读 0赞

## 1.数据库表结构 ##

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3h1ZWd1Y2hlbg_size_16_color_FFFFFF_t_70]  
建表语句：

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 COMMENT='商品三级分类';

## 2. 方法逻辑 ##

实体类：  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3h1ZWd1Y2hlbg_size_16_color_FFFFFF_t_70 1]

方法逻辑：

**注意：存在小问题，Long值用==判断会有bug**

/** * 查出所有分类以及子分类，以树形结构组装起来 */
        @Override
        public List<CategoryEntity> listWithTree() { 
            //1.查找出所有的菜单
            List<CategoryEntity> categoryEntities = baseMapper.selectList(null);
            //2.对菜单进行树状组装
            //2.1 过滤出所有的 1 级菜单(过滤条件：父级菜单 = 0)
            List<CategoryEntity> collect = categoryEntities.stream().filter((categoryEntity) -> { 
                return categoryEntity.getParentCid() == 0;
            })
                    //获取当前菜单所有的子分类，并封装进 children 属性（private List<CategoryEntity> children）
                    .map((menu) -> { 
                         menu.setChildren(getChildrens(menu, categoryEntities));
                         return menu;
                    })
                    //对当前获取的同一层级的菜单进行排序
                    .sorted((menu1, menu2) -> { 
                        return (menu1.getSort() == null ? 0 : menu1.getSort()) - (menu2.getSort() == null ? 0 : menu2.getSort());
                    })
                    //把结果作为一个list返回
                    .collect(Collectors.toList());
    
            return collect;
        }
    
     /** * 递归查找所有菜单的子菜单 * @param categoryEntity 当前菜单 * @param categoryEntityList 获取的所有菜单 * @return 树状菜单 */
        private List<CategoryEntity> getChildrens(CategoryEntity categoryEntity,List<CategoryEntity> categoryEntityList){ 
            //对获取的所有菜单进行过滤（过滤条件：遍历出的菜单的父级id = 当前菜单id），即获取当前菜单的子类的集合
            List<CategoryEntity> collect = categoryEntityList.stream().filter((categoryEntityStr) -> { 
                        return categoryEntityStr.getParentCid()==categoryEntity.getCatId();
                    }
            //递归查找所有菜单的子菜单
            ).map((categoryEntityChild) -> { 
                categoryEntity.setChildren(getChildrens(categoryEntityChild, categoryEntityList));
                return categoryEntityChild; })
             //对当前获取的同一层级的菜单进行排序(需要对排序号进行非空判断，若为空则给默认值0，否则用原值)
            .sorted((menu1, menu2) -> { 
                        return (menu1.getSort() == null ? 0 : menu1.getSort()) - (menu2.getSort() == null ? 0 : menu2.getSort());
            })
             //把结果作为一个list返回
             .collect(Collectors.toList());
    
            return collect;
        }

## 3.结果测试 ##

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3h1ZWd1Y2hlbg_size_16_color_FFFFFF_t_70 2]

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3h1ZWd1Y2hlbg_size_16_color_FFFFFF_t_70]: /images/20221120/c6992c00aea24433a1e748f42e15fa33.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3h1ZWd1Y2hlbg_size_16_color_FFFFFF_t_70 1]: /images/20221120/5b8d8ca9d9734b46ad3987fd39e31eff.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3h1ZWd1Y2hlbg_size_16_color_FFFFFF_t_70 2]: /images/20221120/301b4c03a1674384b626dbe0f9a20ac3.png