当我们打开京东商城时，左侧的菜单依次分为三级展示，这是如何实现的呢？
本篇暂不讲述前端，只讲述如何使用java8 的lamada表达式实现树形数据的拼装
1.创建菜单表
CREATE TABLE `pms_category` ( `cat_id` bigint(20) NOT NULL AUTO_INCREMENT COMMENT '分类id', `name` char(50) CHARACTER SET utf8 COLLATE utf8_general_ci NULL DEFAULT NULL COMMENT '分类名称', `parent_cid` bigint(20) NULL DEFAULT NULL COMMENT '父分类id', `cat_level` int(11) NULL DEFAULT NULL COMMENT '层级', `show_status` tinyint(4) NULL DEFAULT NULL COMMENT '是否显示[0-不显示，1显示]', `sort` int(11) NULL DEFAULT NULL COMMENT '排序', `icon` char(255) CHARACTER SET utf8 COLLATE utf8_general_ci NULL DEFAULT NULL COMMENT '图标地址', `product_unit` char(50) CHARACTER SET utf8 COLLATE utf8_general_ci NULL DEFAULT NULL COMMENT '计量单位', `product_count` int(11) NULL DEFAULT NULL COMMENT '商品数量', PRIMARY KEY (`cat_id`) USING BTREE ) ENGINE = InnoDB AUTO_INCREMENT = 1433 CHARACTER SET = utf8 COLLATE = utf8_general_ci COMMENT = '商品三级分类' ROW_FORMAT = Dynamic;
2.创建实体类
其中children用来存放子类
@Data
@TableName("pms_category")
public class CategoryEntity implements Serializable {
private static final long serialVersionUID = 1L;
/**
* 分类id
*/
@TableId
private Long catId;
/**
* 分类名称
*/
private String name;
/**
* 父分类id
*/
private Long parentCid;
/**
* 层级
*/
private Integer catLevel;
/**
* 是否显示[0-不显示，1显示]
*/
private Integer showStatus;
/**
* 排序
*/
private Integer sort;
/**
* 图标地址
*/
private String icon;
/**
* 计量单位
*/
private String productUnit;
/**
* 商品数量
*/
private Integer productCount;
/**
* 子分类
*/
@TableField(exist = false)
private List<CategoryEntity> children;
}
3.实现一级分类
3.1）首先要查询出所有的分类信息entities
3.2）然后对entities过滤查找一级分类，其中一级分的parentId == 0
3.3）设置当前一级分类的子分类
3.4）对一级分类进行排序
代码如下：
public List<CategoryEntity> listWithTree() {
//1.查询所有分类
List<CategoryEntity> entities = baseMapper.selectList(null);
//2.组装
//2.1) 所到所有的一级类别 ,parentId = 0
List<CategoryEntity> levelOneMenus = entities.stream()
.filter(item -> item.getParentCid() == 0)
.map(item -> {
item.setChildren(this.getChildrens(item, entities));
return item;
})
.sorted((item1, item2) -> {
return (item1.getSort() == null ? 0 : item1.getSort()) - (item2.getSort() == null ? 0 : item2.getSort());
})
.collect(Collectors.toList());
return levelOneMenus;
}
4.使用递归对子分类再进行树形组装
4.1）首先判断当前的id是否有等于集合中的元素的parentId
4.2）再对当前子分类使用递归设置子分类
4.3）对当前子分类排序
4.4）返回子分类集合
详细代码实现如下：
private List<CategoryEntity> getChildrens(CategoryEntity root, List<CategoryEntity> all){
List<CategoryEntity> treeMenus = all.stream()
//如果菜单中的父菜单Id == 当前菜单的id，则说明是子菜单
.filter(item -> Objects.equals(item.getParentCid(), root.getCatId()))
.map(item -> {
//递归添加子菜单
List<CategoryEntity> childrens = getChildrens(item, all);
item.setChildren(childrens);
return item;
})
//排序
.sorted((item1, item2) -> {
return (item1.getSort() == null ? 0 : item1.getSort()) - (item2.getSort() == null ? 0 : item2.getSort());
})
.collect(Collectors.toList());
return treeMenus;
}
