用stream流查询三级分类,以树形结构返回
首先在实体类中添加一个可以递归存储的列表
@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;
}
然后在service中编写方法
public List<CategoryEntity> listWithTree() {
List<CategoryEntity> entities = baseMapper.selectList(null);
List<CategoryEntity> level1Menus = entities.stream()
.filter((categoryEntity) -> {
return categoryEntity.getParentCid() == 0;
})
.map(menu -> {
menu.setChildren(getChildren(menu,entities));
return menu;
})
.sorted((m1, m2) -> {
return (m1.getSort()==null?0:m1.getSort()) - (m2.getSort()==null?0:m2.getSort());
})
.collect(Collectors.toList());
return level1Menus;
}
private List<CategoryEntity> getChildren(CategoryEntity root,List<CategoryEntity> all){
List<CategoryEntity> children = all.stream()
.filter(categoryEntity -> {
return root.getCatId() == categoryEntity.getParentCid();
})
.peek(categoryEntity -> categoryEntity.setChildren(getChildren(categoryEntity, all)))
.sorted((m1, m2) -> {
return (m1.getSort()==null?0:m1.getSort()) - (m2.getSort()==null?0:m2.getSort());
})
.collect(Collectors.toList());
return children;
};
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】凌霞软件回馈社区,博客园 & 1Panel & Halo 联合会员上线
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】博客园社区专享云产品让利特惠,阿里云新客6.5折上折
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 在鹅厂做java开发是什么体验
· 百万级群聊的设计实践
· WPF到Web的无缝过渡:英雄联盟客户端的OpenSilver迁移实战
· 永远不要相信用户的输入:从 SQL 注入攻防看输入验证的重要性
· 浏览器原生「磁吸」效果!Anchor Positioning 锚点定位神器解析