Java 8 Stream Api 中的 map和 flatMap 操作

点击上方蓝色“程序猿DD”，选择“设为星标”

回复“资源”获取独家整理的学习资料！

来源 | 公众号「码农小胖哥」

1.前言

Java 8 提供了非常好用的 Stream API ,可以很方便的操作集合。今天我们探讨两个 Stream 中间操作 map 和 flatMap

2. map 操作

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

示意图：

对应的伪代码：

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

3. flatMap 操作

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

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

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

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

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

再画一张图来加深理解：

4. 总结

map 操作和 flatMap 操作一旦你熟悉了，可以非常简便地解决一些数据流的操作问题。扩展一下知识，其实Java 8 中不光 Stream 中存在这两种操作，其实 Optional<T> 中也存在这两种操作，作用都差不多。

本文通过OpenWrite的Markdown转换工具发布

关注我，回复“加群”加入各种主题讨论群

阿里内部禁用Executors创建线程池，为什么？
Token ，Cookie、Session傻傻分不清楚？
使用 LocalDateTime 而不是 Date
深入解读阿里云Redis开发规范：不要只会set&get
在服务器上排除问题的头 5 分钟

朕已阅