Java后端返回多级分类的树形结构

Java后端返回多级分类的树形结构

使用场景，不管是在前后端分离或者是不分离的项目中，我们通常会遇到这种多级分类的情况，比如前端需要选择 省——市——县 或者 计算机类——后端——Java工程师，这样的三级分类，也可能是二级分类或者更高级的分类。

下面我例举三级学科分类作为例子：

首先建表：

其中 id 代表当前学科，sub_parent_id 代表当前学科的父类学科，当值为 0 时表示当前是一级学科，即没有父类学科，至于 sub_class 和 sub_number 分别表示学科等级和代码，这两个可有可无，至于定义什么样的属性取决于如何更好的进行前后端交互。 

接下来我将展示用 递归 方法结合 lambda 中的 stream 流实现多级分类

下面是表的部分数据展示，注意：数据的 id 和父类 id 必须有关联；

 接着是创建和学科代码表对应的实体类：

@Data
@EqualsAndHashCode(callSuper = false)
@TableName("subject")
@ApiModel(value="Subject对象", description="")
public class Subject implements Serializable {

    private static final long serialVersionUID=1L;

    private Integer id;

    @ApiModelProperty(value = "学科名称")
    private String subName;

    @ApiModelProperty(value = "父类学科，为0就是一级学科")
    private Integer subParentId;

    @ApiModelProperty(value = "等级分类")
    private Integer subClass;

    @ApiModelProperty(value = "学科代码")
    private String subNumber;

    @TableField(exist = false)
    private List children;
}

注意实体类中最后一个地方：

@TableField(exist = false) 表示不和数据库中的字段对应，这里只是为了存放子分类。

@TableField(exist = false)
private List children;

下面是接口的编写：

    @ApiOperation(value = "学科代码树")
    @GetMapping("getSubject")
    public CommonResult getSubject(){
        //获取所有分类学科
        List subject = subjectService.listWithTree();
        return CommonResult.success(subject);
    }

接下来就是实现类，即最核心的部分，接口实现类代码之递归实现：

    @Override
    public List listWithTree() {
        List entities = list();
        List subjectVOS  = entities.stream().map(subject -> {
            Subject subjectVO = new Subject();
            BeanUtils.copyProperties(subject, subjectVO);
            return subjectVO;
        }).collect(Collectors.toList());

        List list = subjectVOS.stream().filter(subjectVO ->
                subjectVO.getSubParentId()==0
        ).map((menu)->{
            menu.setChildren(getChildrenData(menu,subjectVOS));
            return menu;
        }).collect(Collectors.toList());
        return list;
    }

    //获取孩子（下级目录）的方法，递归实现
    private List getChildrenData(Subject root, List all) {
        List children = all.stream().filter(subjectVO ->
                subjectVO.getSubParentId().equals(root.getId())
        ).map(subjectVO -> {
            subjectVO.setChildren(getChildrenData(subjectVO,all));
            return subjectVO;
        }).collect(Collectors.toList());
        return children;
    }

filter 是 lambda 中用于获取条件内的数据的写法，先获取所有 parentId 为0 ，即一级分类的数据，然后通过递归的方式获取所有子分类

其中 map 是 lambda 中的写法，用于封装数据

最后进行测试，测试整合了 Swagger-ui 界面测试,下面是得到的部分数据，由于截图太长，只挑选部分。

 以上内容均可实战，如果有其他类类似的场景，均可仿造代码写出对应功能的接口。