三级分类/菜单的查询——树形结构

本文以商品的分类来举例，实体类如下：

import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.annotation.TableField;
import com.baomidou.mybatisplus.annotation.TableId;
import java.io.Serializable;
import java.util.List;

import com.baomidou.mybatisplus.annotation.TableName;
import io.swagger.annotations.ApiModel;
import io.swagger.annotations.ApiModelProperty;
import lombok.Data;
import lombok.EqualsAndHashCode;
import lombok.experimental.Accessors;

/** * <p> * 商品三级分类 * </p> * * @author 胡尚 * @since 2022-07-21 */
@Data
@EqualsAndHashCode(callSuper = false)
@Accessors(chain = true)
@ApiModel(value="Category对象", description="商品三级分类")
@TableName("pms_category")
public class PmsCategoryDO implements Serializable {
    

    private static final long serialVersionUID = 1L;

    @ApiModelProperty(value = "分类id")
    @TableId(value = "cat_id", type = IdType.AUTO)
    private Long catId;

    @ApiModelProperty(value = "分类名称")
    private String name;

    @ApiModelProperty(value = "父分类id")
    private Long parentCid;

    @ApiModelProperty(value = "层级")
    private Integer catLevel;

    @ApiModelProperty(value = "是否显示[0-不显示，1显示]")
    private Integer showStatus;

    @ApiModelProperty(value = "排序")
    private Integer sort;

    @ApiModelProperty(value = "图标地址")
    private String icon;

    @ApiModelProperty(value = "计量单位")
    private String productUnit;

    @ApiModelProperty(value = "商品数量")
    private Integer productCount;

    @ApiModelProperty(value = "子分类")
    @TableField(exist = false)
    private List<PmsCategoryDO> children;
}

业务方法，本文的重点如下，其他控制层与持久化层的代码就省略了

import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.hs.gulimall.mapper.PmsCategoryMapper;
import com.hs.gulimall.pojo.PmsCategoryDO;
import com.hs.gulimall.service.PmsCategoryService;
import org.springframework.stereotype.Service;

import java.util.List;
import java.util.Objects;
import java.util.stream.Collectors;

/** * <p> * 商品三级分类 服务实现类 * </p> * * @author 胡尚 * @since 2022-07-21 */
@Service
public class PmsCategoryServiceImpl extends ServiceImpl<PmsCategoryMapper, PmsCategoryDO> implements PmsCategoryService {
    


    @Override
    public List<PmsCategoryDO> listWithTree() {
    
        // 查询所有的分类
        List<PmsCategoryDO> categoryList = baseMapper.selectList(null);

        // 查询所有的一级分类，然后再查询二级三级分类
        List<PmsCategoryDO> categoryTreeList = categoryList.stream()
                .filter(categoryEntity -> categoryEntity.getCatLevel() == 1)
                .map(category -> {
    
                    // 查询出当前商品分类的子分类，调用下方的getChildrens()方法
                    category.setChildren(getChildrens(category, categoryList));
                    return category;
                })
                .sorted((o1, o2) -> {
    
                    // 根据数据库中存储的排序字段 进行排序，但是该字段可能为null，所以先进行一些处理
                    o1.setSort(o1.getSort() == null ? 0 : o1.getSort());
                    o2.setSort(o2.getSort() == null ? 0 : o2.getSort());
                    return o1.getSort() - o2.getSort();
                })
                .collect(Collectors.toList());
        return categoryTreeList;
    }

    /** * 获取当前父商品分类信息的子分类 * @param root 父商品分类 * @param categoryList 所有商品分类 * @return 当前商品分类的所有子分类信息 */
    private List<PmsCategoryDO> getChildrens(PmsCategoryDO root, List<PmsCategoryDO> categoryList){
    

        List<PmsCategoryDO> childrenList = categoryList.stream()
                // 根据父分类id进行比较
                .filter(category -> Objects.equals(root.getCatId(), category.getParentCid()))
                .map(category -> {
    
                    // 可能二级分类 还有第三级分类，所以继续递归
                    category.setChildren(getChildrens(category, categoryList));
                    return category;
                })
                .sorted((o1, o2) -> {
    
                    // 根据数据库中存储的排序字段 进行排序，但是该字段可能为null，所以先进行一些处理
                    o1.setSort(o1.getSort() == null ? 0 : o1.getSort());
                    o2.setSort(o2.getSort() == null ? 0 : o2.getSort());
                    return o1.getSort() - o2.getSort();
                })
                .collect(Collectors.toList());

        return childrenList;
    }
}

最终结果如下图所示：