关于java：如何将迭代器转换为流？


How to convert an iterator to a stream?

我正在寻找一种将Iterator转换为Stream的简明方法，或者更具体地说，将迭代器"查看"为流。

出于性能原因，我希望避免在新列表中使用迭代器的副本：

1
2
3
4
Iterator<String> sourceIterator = Arrays.asList("A","B","C").iterator();
Collection<String> copyList = new ArrayList<String>();
sourceIterator.forEachRemaining(copyList::add);
Stream<String> targetStream = copyList.stream();

根据评论中的一些建议，我还尝试使用Stream.generate：

1
2
3
4
5
public static void main(String[] args) throws Exception {
    Iterator<String> sourceIterator = Arrays.asList("A","B","C").iterator();
    Stream<String> targetStream = Stream.generate(sourceIterator::next);
    targetStream.forEach(System.out::println);
}

号

但是，我得到了一个NoSuchElementException(因为没有调用hasNext)

1
2
3
4
5
6
7
Exception in thread"main" java.util.NoSuchElementException
    at java.util.AbstractList$Itr.next(AbstractList.java:364)
    at Main$$Lambda$1/1175962212.get(Unknown Source)
    at java.util.stream.StreamSpliterators$InfiniteSupplyingSpliterator$OfRef.tryAdvance(StreamSpliterators.java:1351)
    at java.util.Spliterator.forEachRemaining(Spliterator.java:326)
    at java.util.stream.ReferencePipeline$Head.forEach(ReferencePipeline.java:580)
    at Main.main(Main.java:20)

我看过StreamSupport和Collections，但什么都没找到。

相关讨论

一种方法是从迭代器创建一个拆分器，并将其用作流的基础：

1
2
3
4
Iterator<String> sourceIterator = Arrays.asList("A","B","C").iterator();
Stream<String> targetStream = StreamSupport.stream(
          Spliterators.spliteratorUnknownSize(sourceIterator, Spliterator.ORDERED),
          false);

另一种可能更易读的方法是使用iterable，并且通过迭代器创建iterable对于lambda来说非常容易，因为iterable是一个函数接口：

1
2
3
4
Iterator<String> sourceIterator = Arrays.asList("A","B","C").iterator();

Iterable<String> iterable = () -> sourceIterator;
Stream<String> targetStream = StreamSupport.stream(iterable.spliterator(), false);

号

相关讨论

从版本21开始，guava库提供Streams.stream(iterator)。

这是@assylias的答案所显示的。

相关讨论

好建议！这是我的可重复使用版本：

1
2
3
4
5
6
7
8
9
10
11
public class StreamUtils {

    public static <T> Stream<T> asStream(Iterator<T> sourceIterator) {
        return asStream(sourceIterator, false);
    }

    public static <T> Stream<T> asStream(Iterator<T> sourceIterator, boolean parallel) {
        Iterable<T> iterable = () -> sourceIterator;
        return StreamSupport.stream(iterable.spliterator(), parallel);
    }
}

和用法(确保静态导入asstream)：

1
2
3
List<String> aPrefixedStrings = asStream(sourceIterator)
                .filter(t -> t.startsWith("A"))
                .collect(toList());

。

在Java 9中，除了流之外，这是可能的，但是有一个警告要注意。下面的简单示例将无法打印迭代器的最后一个元素。

1
2
3
Stream.generate(iterator::next)
    .takeWhile(i -> iterator.hasNext())
    .forEach(System.out::println);

。

但是，这确实可以正常工作：

1
2
3
4
Stream.generate(() -> null)
    .takeWhile(x -> iterator.hasNext())
    .map(n -> iterator.next())
    .forEach(System.out::println);

。

使用Spliterators类从Iterator创建Spliterator包含多个用于创建拆分器的函数，例如这里使用的spliteratorUnknownSize是以迭代器为参数，然后使用StreamSupport创建流。

1
2
3
Spliterator<Model> spliterator = Spliterators.spliteratorUnknownSize(
        iterator, Spliterator.NONNULL);
Stream<Model> stream = StreamSupport.stream(spliterator, false);

使用Collections.list(iterator).stream()...。

相关讨论