it-roy-ru.com

Как преобразовать итератор в поток?

Я ищу краткий способ преобразования 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 , но ничего не нашел.

393
gontard

Один из способов - создать Spliterator из Iterator и использовать его в качестве основы для вашего потока:

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

Альтернатива, которая может быть более читабельна, - это использовать Iterable - и создание Iterable из Iterator очень просто с помощью лямбды, потому что Iterable - это функциональный интерфейс:

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

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

Начиная с версии 21, библиотека Guava предоставляет Streams.stream(iterator)

Это делает то, что @ assylias 's ответ показывает .

96
numéro6

Отличное предложение! Вот мой подход к повторному использованию:

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());
81
Matan

Это возможно в 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);
10
PhilipRoman

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

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