实体类

  1. @Data
  2. @TableName("pms_category")
  3. public class CategoryEntity implements Serializable {
  4.     private static final long serialVersionUID = 1L;
  5.     /**
  6.      * 分类id
  7.      */
  8.     @TableId
  9.     private Long catId;
  10.     /**
  11.      * 分类名称
  12.      */
  13.     private String name;
  14.     /**
  15.      * 父分类id
  16.      */
  17.     private Long parentCid;
  18.     /**
  19.      * 层级
  20.      */
  21.     private Integer catLevel;
  22.     /**
  23.      * 是否显示[0-不显示，1显示]
  24.      */
  25.     @TableLogic(value = "1", delval = "0")
  26.     private Integer showStatus;
  27.     /**
  28.      * 排序
  29.      */
  30.     private Integer sort;
  31.     /**
  32.      * 图标地址
  33.      */
  34.     private String icon;
  35.     /**
  36.      * 计量单位
  37.      */
  38.     private String productUnit;
  39.     /**
  40.      * 商品数量
  41.      */
  42.     private Integer productCount;
  44.     @JsonInclude(JsonInclude.Include.NON_EMPTY)
  45.     @TableField(exist = false)
  46.     private List<CategoryEntity> children;
  47. }

实现

  1. public List<CategoryEntity> listWithTree() {
  2.     //1、查出所有分类
  3.     List<CategoryEntity> entities = baseMapper.selectList(null);
  5.     //2、组装成父子的树形结构
  6.     //2.1）、找到所有的一级分类，给children设置子分类
  7.     return entities.stream()
  8.         // 过滤找出一级分类
  9.         .filter(categoryEntity -> categoryEntity.getParentCid() == 0)
  10.         // 处理，给一级菜单递归设置子菜单
  11.         .peek(menu -> menu.setChildren(getChildless(menu, entities)))
  12.         // 按sort属性排序
  13.         .sorted(Comparator.comparingInt(menu -> (menu.getSort() == null ? 0 : menu.getSort())))
  14.         .collect(Collectors.toList());
  15. }
  17. /**
  18.  * 递归查找所有菜单的子菜单
  19.  */
  20. private List<CategoryEntity> getChildless(CategoryEntity root, List<CategoryEntity> all) {
  21.     return all.stream()
  22.         .filter(categoryEntity -> categoryEntity.getParentCid().equals(root.getCatId()))
  23.         .peek(categoryEntity -> {
  24.             // 找到子菜单
  25.             categoryEntity.setChildren(getChildless(categoryEntity, all));
  26.         })
  27.         .sorted(Comparator.comparingInt(menu -> (menu.getSort() == null ? 0 : menu.getSort())))
  28.         .collect(Collectors.toList());
  29. }