Java Stream 流如何進行合併操作

碼農小胖哥發表於2020-05-12
Java

1. 前言

Java Stream Api 提供了很多有用的 Api 讓我們很方便將集合或者多個同型別的元素轉換為流進行操作。今天我們來看看如何合併 Stream 流。

2. Stream 流的合併

Stream 流合併的前提是元素的型別能夠一致。

2.1 concat

最簡單合併流的方法是通過 Stream.concat() 靜態方法：

Stream<Integer> stream = Stream.of(1, 2, 3);
Stream<Integer> another = Stream.of(4, 5, 6);
Stream<Integer> concat = Stream.concat(stream, another);

List<Integer> collect = concat.collect(Collectors.toList());
List<Integer> expected = Lists.list(1, 2, 3, 4, 5, 6);

Assertions.assertIterableEquals(expected, collect);

這種合併是將兩個流一前一後進行拼接：

2.2 多個流的合併

多個流的合併我們也可以使用上面的方式進行“套娃操作”：

Stream.concat(Stream.concat(stream, another), more);

你可以一層一層繼續套下去，如果需要合併的流多了，看上去不是很清晰。

我之前介紹過一個Stream 的 flatmap 操作 ，它的大致流程可以參考裡面的這一張圖：

因此我們可以通過 flatmap 進行實現合併多個流：

Stream<Integer> stream = Stream.of(1, 2, 3);
Stream<Integer> another = Stream.of(4, 5, 6);
Stream<Integer> third = Stream.of(7, 8, 9);
Stream<Integer> more = Stream.of(0);
Stream<Integer> concat = Stream.of(stream,another,third,more).
    flatMap(integerStream -> integerStream);
List<Integer> collect = concat.collect(Collectors.toList());
List<Integer> expected = Lists.list(1, 2, 3, 4, 5, 6, 7, 8, 9, 0);
Assertions.assertIterableEquals(expected, collect);

這種方式是先將多個流作為元素生成一個型別為 Stream<Stream<T>> 的流，然後進行 flatmap 平鋪操作合併。

2.3 第三方庫

有很多第三方的強化庫 StreamEx 、Jooλ 都可以進行合併操作。另外反應式程式設計庫 Reactor 3 也可以將 Stream 流合併為反應流，在某些場景下可能會有用。這裡演示一下：

List<Integer> block = Flux.fromStream(stream)
                       .mergeWith(Flux.fromStream(another))
                                 .collectList()
                                 .block();

3. 總結

如果你經常使用 Java Stream Api ，合併 Stream 流是經常遇到的操作。今天簡單介紹了合併 Stream 流的方式，希望對你有用。我是 碼農小胖哥 ,多多關注！更多幹貨奉上。

關注公眾號：Felordcn 獲取更多資訊

個人部落格：https://felord.cn

相關文章