查询表中数据，并返回成Tree树形数据，解决办法

匍匐前进の程序猿

已于 2022-07-07 16:17:31 修改

阅读量1.3k

点赞数 1

于 2022-04-07 14:00:59 首次发布

本文链接：https://blog.csdn.net/pi_tiger/article/details/124014211

版权

本文介绍了如何将查询结果转化为树形数据结构并进行展示。通过创建DTO对象，利用Java的Stream API和Jackson库的@JsonInclude注解进行数据格式化。在代码实现中，首先遍历所有数据，然后根据parentId属性构建树形结构。最后，数据以文科课程和理科课程分类的形式展示。

摘要由CSDN通过智能技术生成

前言

在这里插入图片描述
当需要将以上查询到的数据返回为tree，即树形数据时，如何能快速有效的完成？下面将为你解答。

一、实现步骤：

1.创建DTO，用于格式化数据

@JsonInclude(value = JsonInclude.Include.NON_EMPTY)
将该标记放在属性上，如果该属性为NULL则不参与序列化
如果放在类上边,那对这个类的全部属性起作用
Include.Include.ALWAYS 默认
Include.NON_DEFAULT 属性为默认值不序列化
Include.NON_EMPTY 属性为空（“”）或者为 NULL 都不序列化
Include.NON_NULL 属性为NULL 不序列化

	@Data
    @JsonInclude(value = JsonInclude.Include.NON_EMPTY)
    public static class CourceDto {
        private Integer value;
        private String label;
        @JsonIgnore //作用在属性上，生成json时,不生成parentId 属性
        private Integer parentId;
        private List<CourceDto> children = new ArrayList<>();
    }

2.代码实现

代码如下：

	@PostMapping("check-tree-advanced")
    public List<CourceDto> checkRepetitionForTreeAdvanced() throws Exception {
        List<Course> courseList = courseRepo.findAll();

        Map<Integer, CourceDto> map = courseList.stream().collect(Collectors.toMap(Course::getId, e -> {
            CourceDto courceDto = new CourceDto();
            courceDto.setValue(e.getId());
            courceDto.setLabel(e.getName());
            courceDto.setParentId(e.getParentId());
            return courceDto;
        }));
        List<CourceDto> result = new ArrayList<>();
        map.forEach((k, v) -> {
            if (v.getParentId() == null) { // 如果parentId为空，则表述该数据为主目录
                result.add(v);
            } else { // 如果不为空，则找到map中对应的主目录数据，并插入children中
                Integer parentId = v.getParentId();
                CourceDto courceDto = map.get(parentId);
                if (courceDto != null) {
                    courceDto.getChildren().add(v);
                }
            }
        });
        return result;
    }