[java] Как конвертировать итератор в поток?


Answers

Отличное предложение! Вот мое многоразовое решение:

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());
Question

Я ищу сжатый способ преобразования 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)

Я посмотрел на StreamSupport и Collections но ничего не нашел.




Создание Spliterator из Iterator с использованием класса Spliterators содержит более одной функции для создания spliterator, например, здесь используется spliteratorUnknownSize который получает итератор в качестве параметра, а затем создает Stream с помощью StreamSupport

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



Используйте Collections.list(iterator).stream()...




Related