Java处理数据成为树状结构

如题所示，项目中需要将部分数据处理成为树状结构，实现过程如下：

注：也可以使用sql达到该目的，但此处数据不多，故在代码中处理，主要是sql处理不是很会

	// 获取需要封装的数据
	List<Data> dataAll = service.findAll();
    if (CollUtil.isEmpty(dataAll)) {
      return new ArrayList<>();
    }
    // 获取顶层目录 项目中上级编码为空则为一级目录
    List<Data> parentList = dataAll.stream()
        .filter(item -> ObjectUtil.isEmpty(item.getParentNumber())).collect(Collectors.toList());

    // 获取下级目录，上级编码不为空则非一级目录
    List<Data> childList = dataAll.stream()
        .filter(item -> ObjectUtil.isNotEmpty(item.getParentNumber())).collect(Collectors.toList());

    // 封装树状结构
    return parentList.stream().map(item1 -> {
      Data chapterVo = new Data();
      BeanUtils.copyProperties(item1, chapterVo);
      List<Data> videoVoList = childList.stream()
          .filter(item2 -> item1.getBackgroundManagementCategoryNumber().equals(item2.getParentNumber()))
          .map(item3 -> {
            Data videoVo = new Data();
            BeanUtils.copyProperties(item3, videoVo);
            return videoVo;
          })
		  .collect(Collectors.toList());
      chapterVo.setChildren(videoVoList);
      return chapterVo;
    }).collect(Collectors.toList());
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28