LarryDpk
发布于 2019-09-26 / 1090 阅读
0

Java 8之例说Stream的合并

本篇文章主要介绍几种合并stream的方法,主要有原生JDK的方法和使用第三方库StreamEx和Jooλ的方法。

1 原生JDK

原生的JDK 8提供的Stream里的一些静态方法,非常有用,让我们来了解一下吧。

1.1 concat()合并两个Stream

Stream<Integer> stream1 = Stream.of(1, 2, 3);
Stream<Integer> stream2 = Stream.of(4, 5);
Stream<Integer> result = Stream.concat(stream1, stream2);
assertEquals(asList(1, 2, 3, 4, 5),
             result.collect(Collectors.toList()));

这是最简单的合并两个流的方式。

1.2 用Stream.of()合并多个流

之前的concat()方法并不支持合并多个stream,这时需要用到of(),可以实现多个stream的合并。

stream1 = Stream.of(1, 2, 3);
stream2 = Stream.of(4, 5);
Stream<Integer> stream3 = Stream.of(6, 7, 8);
result = Stream.of(stream1, stream2, stream3)
  .flatMap(i -> i);
assertEquals(asList(1, 2, 3, 4, 5, 6, 7, 8),
             result.collect(Collectors.toList()));

2 使用StreamEx

StreamEx是开源的Java库,扩展了Java 8 Streams的更多可能性。

Maven引入:

<dependency>
  <groupId>one.util</groupId>
  <artifactId>streamex</artifactId>
  <version>0.7.0</version>
</dependency>

2.1 使用append()合并

stream1 = Stream.of(1, 2, 3);
stream2 = Stream.of(4, 5);
stream3 = Stream.of(6, 7, 8);
result = StreamEx.of(stream1)
  .append(stream2)
  .append(stream3);
assertEquals(asList(1, 2, 3, 4, 5, 6, 7, 8),
             result.collect(Collectors.toList()));

这样我们就可以动态不断合并流。

2.2 使用prepend()合并

Stream<String> streamA = Stream.of("[");
Stream<String> streamB = Stream.of("google", "apple");
Stream<String> streamC = Stream.of("]");
Stream<String> resultStream = StreamEx.of(streamB)
  .prepend(streamA)
  .append(streamC);

assertEquals(asList("[", "google", "apple", "]"),
             resultStream.collect(Collectors.toList()));

使用prepend()能明显地知道前后顺序。

3 使用Jooλ

Mavan引入:

<dependency>
  <groupId>org.jooq</groupId>
  <artifactId>jool-java-8</artifactId>
  <version>0.9.13</version>
</dependency>

Jooλ和StreamEx很类似,也有对应的append()和prepend()方法。

stream1 = Stream.of(1, 2, 3);
stream2 = Stream.of(4, 5);
result = Seq.ofType(stream1, Integer.class).append(stream2);
assertEquals(asList(1, 2, 3, 4, 5),
             result.collect(Collectors.toList()));

/*--------------*/

streamA = Stream.of("[");
streamB = Stream.of("google", "apple");
streamC = Stream.of("]");
resultStream = Seq.ofType(streamB, String.class)
  .prepend(streamA)
  .append(streamC);
assertEquals(asList("[", "google", "apple", "]"),
             resultStream.collect(Collectors.toList()));

4 总结

如果用得不多,JDK足以应付;如果经常使用,场景也比较多,用StreamEx和Jooλ能提高可读性。