文章目錄
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 之前 我們要通過新建一個集合然后通過遍歷學生集合來消費元素中的年齡屬性。現在我們通過很簡單的流式操作就完成了這個需求。
示意圖：
對應的偽代碼：
// 偽代碼
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> 中也存在這兩種操作，作用都差不多。
關注公眾號：Felordcn獲取更多資訊