节点list转为树形结构（Java1.8）

转载

mob604756ea26ec 2021-10-04 17:16:00

文章标签 子节点树形结构递归调用数据空指针异常 文章分类 Java 后端开发

 1     // 查询树形结构节点
 2     public List<CategoryEntity> listWithTree() {
 3         // 查出所有节点
 4         List<CategoryEntity> categoryEntities = baseMapper.selectList(null);
 5         // 组装父子树形结构
 6         return categoryEntities.stream()
 7                 // 获取一级节点
 8                 .filter(categoryEntity -> categoryEntity.getParentCid() == 0L)
 9                 // 插入子节点到children属性
10                 .peek(menu -> menu.setChildren(getChildren(menu, categoryEntities)))
11                 // 根据sort属性排序（空值居后）
12                 .sorted(Comparator.nullsLast(Comparator.comparingInt(CategoryEntity::getSort)))
13                 // 将流数据转成List类型
14                 .collect(Collectors.toList());
15     }
16     // 根据全部节点，获取当前节点的子节点
17     private List<CategoryEntity> getChildren(CategoryEntity root, List<CategoryEntity> all) {
18         return all.stream()
19                 // 获取当前节点的子节点
20                 .filter(menu -> root.getCatId().equals(menu.getParentCid()))
21                 // 递归调用，查找子节点插入至children属性
22                 .peek(categoryEntity -> categoryEntity.setChildren(getChildren(categoryEntity, all)))
23                 // 对子节点根据sort属性排序（空值居后）
24                 .sorted(Comparator.nullsLast(Comparator.comparingInt(CategoryEntity::getSort)))
25                 // 将流数据转成List类型
26                 .collect(Collectors.toList());
27     }