如何交错(合并)两个Java 8流?


问题内容
 Stream<String> a = Stream.of("one", "three", "five");
 Stream<String> b = Stream.of("two", "four", "six");

对于下面的输出,我需要做什么?

// one
// two
// three
// four
// five
// six

我调查了一下,concat但是正如javadoc所解释的,它只是一个接一个地追加,它不会交错/散布。

Stream<String> out = Stream.concat(a, b);
out.forEach(System.out::println);

创建一个延迟串联的流,其元素是第一个流的所有元素,后跟第二个流的所有元素。

错误地给

 // one
 // three
 // five
 // two
 // four
 // six

如果我收集它们并进行迭代,但是希望获得更多Java8-y,Streamy :-),可以这样做

注意

我不想压缩流

“ zip”操作将从每个集合中获取一个元素并将其组合。

压缩操作的结果将是这样的:(不需要)

 // onetwo
 // threefour
 // fivesix

问题答案:

我会用这样的东西:

public static <T> Stream<T> interleave(Stream<? extends T> a, Stream<? extends T> b) {
    Spliterator<? extends T> spA = a.spliterator(), spB = b.spliterator();
    long s = spA.estimateSize() + spB.estimateSize();
    if(s < 0) s = Long.MAX_VALUE;
    int ch = spA.characteristics() & spB.characteristics()
           & (Spliterator.NONNULL|Spliterator.SIZED);
    ch |= Spliterator.ORDERED;

    return StreamSupport.stream(new Spliterators.AbstractSpliterator<T>(s, ch) {
        Spliterator<? extends T> sp1 = spA, sp2 = spB;

        @Override
        public boolean tryAdvance(Consumer<? super T> action) {
            Spliterator<? extends T> sp = sp1;
            if(sp.tryAdvance(action)) {
                sp1 = sp2;
                sp2 = sp;
                return true;
            }
            return sp2.tryAdvance(action);
        }
    }, false);
}

它尽可能保留输入流的特征,从而允许进行某些优化(例如count()toArray())。此外,ORDERED当输入流可能无序时,它还会添加偶数,以反映交织。

当一个流中的元素多于另一个时,其余元素将出现在末尾。