Java Stream API flatMap()方法介绍

骑着猪看大海2024-10-16 9:52

Stream API flatMap()方法介绍

flatMap()简介

flatMap 的主要作用是将一个流中的每个元素转换为另一个流，然后将这些流合并成一个单一的流。经常用作处理嵌套的流。

标准语法

Stream<R> flatMap(Function<? super T, ? extends Stream<? extends R>> mapper)

示例

假设现在有一个年龄列表ageList和学生信息列表studentList，需要从studentList中取出有年龄age在年龄列表中的学生信息

不使用stream方法
使用两个for循环，拿每一个age和每一个学生的年龄作比较，如果符合则添加进结果中

java 复制代码

for (int age : ageList) {
    for (SysStudent student : studentList) {
        if (student.getStudentAge() == age) {
            resultList.add(student);
        }
    }
}

使用stream方法
将ageList中的age扁平化成多个流，分别根据每个age做过滤，如果满足条件则收集起来

java 复制代码

resultList = ageList.stream()
            .flatMap(age -> studentList.stream()
                    .filter(student -> student.getStudentAge() == age))
            .collect(Collectors.toList());

flatMap和map的区别

flatMap：将流中的每个元素转换为另一个流，并将这些流合并成一个单一的流。
map：将流中的每个元素转换为另一个对象，但不涉及流的扁平化。

我自己的理解是这样的

比如本例中，如果使用map，则需要在后续步骤中合并成单一的流

java 复制代码

        List<List<SysStudent>> mappedResult = ageList.stream()
                .map(age -> studentList.stream()
                        .filter(student -> student.getStudentAge() == age)
                        .collect(Collectors.toList()))
                .collect(Collectors.toList());
        resultList = mappedResult.stream()
                .flatMap(List::stream)
                .collect(Collectors.toList());