如何将迭代器转换为流?

时间:2014-07-01 13:05:16

标签: java iterator java-8

我正在寻找一种简洁的方式将Iterator转换为Stream或更具体地转换为"视图"迭代器作为流。

出于性能原因,我想在新列表中避免使用迭代器的副本:

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

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

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)

我查看了StreamSupportCollections,但我找不到任何内容。

9 个答案:

答案 0 :(得分:450)

一种方法是从迭代器创建一个Spliterator并将其用作流的基础:

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

可能更具可读性的替代方法是使用Iterable - 使用lambdas从Iterator创建Iterable非常容易,因为Iterable是一个功能接口:

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

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

答案 1 :(得分:89)

从版本21开始,Guava库提供了Streams.stream(iterator)

它会执行@ assyliasanswer shows

答案 2 :(得分:81)

很棒的建议! 这是我可重复使用的内容:

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):

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

答案 3 :(得分:10)

这在java 9中可以使用除了流之外的任何东西,但是需要注意一点需要注意。以下天真的例子无法打印迭代器的最后一个元素

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

然而,这确实可行:

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

答案 4 :(得分:4)

使用SpliteratorIterator创建Spliterators包含多个用于创建spliterator的函数,例如,这里使用的是spliteratorUnknownSize,它将迭代器作为参数,然后创建使用StreamSupport

进行流式传输
Spliterator<Model> spliterator = Spliterators.spliteratorUnknownSize(
        iterator, Spliterator.NONNULL);
Stream<Model> stream = StreamSupport.stream(spliterator, false);

答案 5 :(得分:4)

import com.google.common.collect.Streams;

并使用Streams.stream(iterator)

Streams.stream(iterator)
       .map(v-> function(v))
       .collect(Collectors.toList());

答案 6 :(得分:0)

使用Stream::iterate(T, Predicate, UnaryOperator)在Java 9+上执行此操作的另一种方法:

Stream.iterate(iterator, Iterator::hasNext, UnaryOperator.identity())
        .map(Iterator::next)
        .forEach(System.out::println);

答案 7 :(得分:0)

1 assylias 的解决方案封装在方法中:

public static <T> Stream<T> toStream(Iterator<T> iterator) {
    return StreamSupport.stream(((Iterable<T>)() -> iterator).spliterator(), false);
}

2 guava Streams 实现(标有 @Beta):

public static <T> Stream<T> stream(Iterator<T> iterator) {
    return StreamSupport.stream(Spliterators.spliteratorUnknownSize(iterator, 0), false);
}

答案 8 :(得分:-4)

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