一起迭代两个Java 8流

Question

我想要同时遍历两个Java 8流，以便在每个迭代步骤中有两个参数。像这样的东西，其中somefunction生成类似于Stream<Pair<A,B>>的内容。

Stream<A> as;
Stream<B> bs;
somefunction (as, bs)
  .forEach ((a, b) -> foo (a, b));
// or something like
somefunction (as, bs)
  .forEach ((Pair<A, B> abs) -> foo (abs.left (), abs.right ()));

我想知道Java是否提供类似的功能，尽管Java中没有Pair :-( 如果没有类似的API函数，是否有另一种同时迭代两个流的方法？

- F. Böller

Answer 1

static <A, B> Stream<Pair<A, B>> zip(Stream<A> as, Stream<B> bs)
{
    Iterator<A> i=as.iterator();
    return bs.filter(x->i.hasNext()).map(b->new Pair<>(i.next(), b));
}

这并不提供并行执行，但原始的zip实现也不提供。

正如F. Böller指出的那样，如果bs是无限的而as不是，则它不起作用。 对于适用于所有可能的有限和无限流组合的解决方案，似乎不可避免要使用一个中间的Iterator，它在hasNext方法中检查两个源：

static <A, B> Stream<Pair<A,B>> zip(Stream<A> as, Stream<B> bs) {
    Iterator<A> i1 = as.iterator();
    Iterator<B> i2 = bs.iterator();
    Iterable<Pair<A,B>> i=()->new Iterator<Pair<A,B>>() {
        public boolean hasNext() {
            return i1.hasNext() && i2.hasNext();
        }
        public Pair<A,B> next() {
            return new Pair<A,B>(i1.next(), i2.next());
        }
    };
    return StreamSupport.stream(i.spliterator(), false);
}

如果您想要支持并行压缩，应该考虑 Stream 的来源。例如，您可以像下面这样对两个 ArrayList（或任何RandomAccessList）进行压缩：

ArrayList<Foo> l1=new ArrayList<>();
ArrayList<Bar> l2=new ArrayList<>();
IntStream.range(0, Math.min(l1.size(), l2.size()))
         .mapToObj(i->new Pair(l1.get(i), l2.get(i)))
         . …

¹(除非您直接实现Spliterator)

- Holger

---

刚发现了一个问题，对于无限长度的 B 流这个解决方案是不起作用的。在这种情况下，它会出现无限循环的情况。B 流必须以某种方式受到限制。但是，我没有找到能够通过对您上面的代码进行小改动来解决问题的方案。链接的答案似乎也适用于无限流：链接 - F. Böller

@F. Böller：没错，没有小改变可以解决这个问题。你提供的链接可能可行，但我建议谨慎考虑许可问题，因为代码是从 beta jdk 源中复制粘贴而来，正如回答者本人所说。此外，你应该记住，代码的作者可能有理由将其从 jdk 中移除，因此可能存在其他问题。 - Holger