java 实现关于多级树形分类结构的查询

最新推荐文章于 2024-05-03 09:12:33 发布

原创最新推荐文章于 2024-05-03 09:12:33 发布 · 6.9k 阅读

64 ·

CC 4.0 BY-SA版权

文章标签：

#java #stream

SpringBoot 专栏收录该内容

6 篇文章

订阅专栏

本文介绍如何在Java中实现多级树形分类结构的查询。通过数据库设计、VO类创建、Controller和Service的实现，详细展示了从数据库获取数据并构建树形结构的过程。其中，Service层使用Java8的Stream API进行数据过滤和映射，实现了递归获取子节点的功能。文章最后提供了完整的GitHub源码链接供读者参考。

java 实现关于多级树形分类结构的查询

1.数据库设计

在这里插入图片描述

2.VO类实现

import lombok.Data;

import java.util.List;

@Data
public class SubjectVO {
    private String id;
    private String parentId;
    private String title;
    private List<SubjectVO> children;
}

3.Controller

@GetMapping("getAllSubjects")
    public ResponseEntity getAllSubjects(){
        List<SubjectVO> list = subjectService.listWithTree();
		//ResponseEntity是我自己封住的返回JSON数据的格式，这里可以替换成你们自己的
        return ResponseEntity.ok().data("list",list);
    }

4.Service

接口EduSubjectService

List<SubjectVO> listWithTree();

实现类EduSubjectServiceImpl

@Override
    public List<SubjectVO> listWithTree() {
        List<EduSubject> entities = list(); //这个可以写成你自己后台接口所取到数据的方法，由于我使用的是Mybatis-plus直接调用这个方法取得数据

//把EduSubject的数组entities包装成SubjectVO数组
        List<SubjectVO> subjectVOS = entities.stream().map(subject -> {
            SubjectVO subjectVO = new SubjectVO();
            BeanUtils.copyProperties(subject, subjectVO);
            return subjectVO;
        }).collect(Collectors.toList());


//filter是java8的写法，取得所有parentId为0的数据，也就是一级目录
//map也是java8的写法，用于封装数据，取得他的孩子（也就是下级目录）的数据
        List<SubjectVO> list = subjectVOS.stream().filter(subjectVO ->
                subjectVO.getParentId().equals("0")
        ).map((menu)->{
            menu.setChildren(getChildrenData(menu,subjectVOS));
            return menu;
        }).collect(Collectors.toList());
        return list;
    }

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