Java8利用stream流实现递归遍历树形结构

利用stream流实现递归遍历树形结构

1. 什么是树形结构

下面用一张图片说明：

在这张表中，每条数据分别有自己的id和parentId，这些数据通过父与子不断连接，形成了一个树结构。

2. 如何通过stream流处理树形结构

我们最终需要的结果是一个树形的json串，如下：

话不多说，直接上代码：

这是course_category类，即数据库对应的vo：

这是CourseCategoryTreeDto类，即我们所需要的dto类：

service层代码实现如下：

@Service
@Slf4j
public class CourseCategoryServiceImpl implements CourseCategoryService {

    @Autowired
    private CourseCategoryMapper courseCategoryMapper;

    @Override
    public List<CourseCategoryTreeDto> queryTreeNodes(String id) {
    	// 获取数据库中的数据，这里建议先排序一下，会快很多
        List<CourseCategoryTreeDto> categoryTreeDtos = courseCategoryMapper.selectTreeNodes(id);
        // 调用递归函数获取树形结构，并排除掉表中的根节点：m -> m.getParentid().equals("1")
        List<CourseCategoryTreeDto> courseCategoryTreeDtos =
                categoryTreeDtos.stream()
                        .filter( m -> m.getParentid().equals("1"))
                        .peek(m -> m.setChildrenTreeNodes(getChildrens(m, categoryTreeDtos)))
                        .collect(Collectors.toList());

        return courseCategoryTreeDtos;
    }

    // 递归函数 过滤拿到父节点为当前节点的节点，递归加入到当前节点的子节点中，再收集起来
    private List<CourseCategoryTreeDto> getChildrens(CourseCategoryTreeDto root, List<CourseCategoryTreeDto> all){
        List<CourseCategoryTreeDto> childrens = all.stream()
                .filter( m -> Objects.equals(m.getParentid(), root.getId()))
                // peek方法返回由该流的元素组成的流，并对每个元素执行所提供的 Consumer操作方法
                // peek和foreach在功能上类似，但foreach执行完函数不做其他操作了，peek完还可以继续对stream进行其它操作
            	// 这篇文章解释的比较清晰：https://blog.csdn.net/VoisSurTonChemin/article/details/122378636
                .peek(m -> m.setChildrenTreeNodes(getChildrens(m, all)))
                .collect(Collectors.toList());
        return childrens;
    }

}

当然，不用stream流我们也可以实现，如下：

@Service
@Slf4j
public class CourseCategoryServiceImpl implements CourseCategoryService {

    @Autowired
    private CourseCategoryMapper courseCategoryMapper;

    @Override
    public List<CourseCategoryTreeDto> queryTreeNodes(String id) {

        // 获得根节点下所有子节点 注意 CourseCategoryTreeDto是CourseCategory的子类
        // 这里获取的List<CourseCategoryTreeDto>实际上只是List<CourseCategory>罢了 并没有构建树形结构 需要下面的处理
        List<CourseCategoryTreeDto> categoryTreeDtos = courseCategoryMapper.selectTreeNodes(id);

        // 定义一个list作为最终返回的数据
        List<CourseCategoryTreeDto> courseCategoryTreeDtos = new ArrayList<>();
        // 为了方便找子节点的父节点，我们定义一个map来存
        HashMap<String, CourseCategoryTreeDto> nodeMap = new HashMap<>();

        // 将数据封装到list中，只包括了根节点的直接下属节点，即实现树形结构
        categoryTreeDtos.stream().forEach(item -> {
            // 存入hashmap中，当前节点的id和信息
            nodeMap.put(item.getId(), item);
            String parentId = item.getParentid();
            // 如果当前节点的的父节点是传过来的节点时，即为根节点的子节点时，加入该节点
            // 因为根节点为0，是定义的一个无关的节点，传过去的数据应该从根节点的叶子节点开始
            if(parentId.equals(id)){
                courseCategoryTreeDtos.add(item);
            }

            // 此处是获取该项item的父节点的ChildrenTreeNodes，将该项加入对应父节点的的ChildrenTreeNodes中
            CourseCategoryTreeDto parentNode = nodeMap.get(parentId);
            if(parentNode != null){
                List childrenTreeNodes = parentNode.getChildrenTreeNodes();
                if(childrenTreeNodes == null){
                    parentNode.setChildrenTreeNodes(new ArrayList<CourseCategoryTreeDto>());
                }
                parentNode.getChildrenTreeNodes().add(item);
            }
        });
    }

}

参考文章：

1. Java8新特性-使用Stream流递归实现遍历树形结构
   https://blog.csdn.net/qq_19244927/article/details/106481777
2. 【Java 8 新特性】Java Stream通过peek()为每个元素提供消费函数
   https://blog.csdn.net/qq_31635851/article/details/111150978
3. Java stream 中 peek() 的合理用法
   https://blog.csdn.net/VoisSurTonChemin/article/details/122378636