java steam 自定义判断 java steam map

转载

代码匠人之心 2023-11-02 09:00:25

文章标签 java steam 自定义判断 java stream map 原理 List 伪代码 Java 文章分类 Java 后端开发

1.前言

Java 8 提供了非常好用的 Stream API ,可以很方便的操作集合。今天我们来探讨两个 Stream 中间操作 map(Function super T, ? extends R> mapper) 和 flatMap(Function super T, ? extends Stream extends R>> mapper)

2. map 操作

map 操作是将流中的元素进行再次加工形成一个新流。这在开发中很有用。比如我们有一个学生集合，我们需要从中提取学生的年龄以分析学生的年龄分布曲线。放在 Java 8 之前我们要通过新建一个集合然后通过遍历学生集合来消费元素中的年龄属性。现在我们通过很简单的流式操作就完成了这个需求。

示意图：

java steam 自定义判断 java steam map_List_02

对应的伪代码：

// 伪代码
List ages=studentList.stream().map(Student::getAge).collect(Collectors.toList());

3. flatMap 操作

通过上面的例子，map 操作应该非常好理解。那么 flatMap 是干嘛的呢？这样我们把上面的例子给改一下，如果是以班级为单位，提取所有班级下的所有学生的年龄以分析学生的年龄分布曲线。这时我们使用上面的方法还行得通吗？

List> studentGroup= gradeList.stream().map(Grade::getStudents).collect(Collectors.toList());

通过上面的一顿操作，我们只能得到每个班的学生集合的集合 List>。我们还需要嵌套循环才能获取学生的年龄数据，十分不便。如果我们能返回全部学生的集合 List 就方便多了。没错！flatMap 可以搞定！

// flatMap 提取 List map 提取年龄
List ages = grades.stream().flatMap(grade -> grade.getStudents().stream()).map(Student::getAge).collect(Collectors.toList());

正如上面的伪代码所示，我们使用 flatMap 将所有的学生汇聚到一起。然后再使用 map 操作提取年龄。 flatMap 不同于 map 地方在于 map 只是提取属性放入流中，而 flatMap 先提取属性放入一个比较小的流，然后再将所有的流合并为一个流。有一种 “聚沙成塔” 的感觉。

再画一张图来加深理解：

java steam 自定义判断 java steam map_java stream map 原理_03