三级分类之Stream的使用

最新推荐文章于 2023-03-17 11:38:18 发布

无奈何杨

最新推荐文章于 2023-03-17 11:38:18 发布

阅读量1.1k

点赞数

文章标签： spring boot 后端 java

本文链接：https://blog.csdn.net/weixin_44783934/article/details/121432738

版权

参考

https://www.bilibili.com/video/BV1np4y1C7Yf?p=45

简介

在跟着尚硅谷学习谷粒商城中遇到这个三级分类问题，使用了Stream Api，记录一下

过程

主要内容就是对商品类别表进行三级分类，category表设计如下，主要关注，分类id，父分类id，排序这几个字段

数据内容，这里只贴了30条，总1000条，随后在web上展示

了解了需求，CategoryEntity如下，关注新加入的非数据库字段属性List<CategoryEntity> 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;

}

我们重点关注业务层，dao层使用的是Mybatis-Plus，不多介绍

就是在这里我们使用了Stream处理了从数据库查到的所有分类信息

处理过程：

从数据库查到所有分类信息List<CategoryEntity>
先查到所有的一级分类（parentCid为0的，从上面的数据内容也可看出来）
接着从这些父分类，找到属于他的子分类（通过getChildren函数）
sorted根据sort字段升序排列
collect形成集合

@Service("categoryService")
public class CategoryServiceImpl extends ServiceImpl<CategoryDao, CategoryEntity> implements CategoryService {

    ...

    @Override
    public List<CategoryEntity> listWithTree() {
        // 1、查出所有分类
        List<CategoryEntity> entities = baseMapper.selectList(null);
        // 2、组装树形结构

        // 2.1、一级分类
        List<CategoryEntity> level1Menu = entities.stream().filter((categoryEntity) ->
                categoryEntity.getParentCid() == 0
        ).peek((menu) ->
                menu.setChildren(getChildren(menu, entities))
        ).sorted(
                Comparator.comparingInt(CategoryEntity::getSort)
        ).collect(Collectors.toList());
        return level1Menu;
    }

    /**
     * 递归查找所有菜单的子菜单
     * @param root
     * @param all
     * @return
     */
    private List<CategoryEntity> getChildren(CategoryEntity root, List<CategoryEntity> all) {
        List<CategoryEntity> children = all.stream().filter((categoryEntity ->
                categoryEntity.getParentCid().equals(root.getCatId()))
        ).peek((categoryEntity ->
                categoryEntity.setChildren(getChildren(categoryEntity, all)))
        ).sorted(
                Comparator.comparingInt(CategoryEntity::getSort)
        ).collect(Collectors.toList());
        return children;
    }
    
    ...
        
}