Como converter um iterador em um fluxo?

468

Estou procurando uma maneira concisa de converter um Iteratorpara um Streamou mais especificamente para "visualizar" o iterador como um fluxo.

Por motivos de desempenho, gostaria de evitar uma cópia do iterador em uma nova lista:

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

Com base nas sugestões apresentadas nos comentários, também tentei usar 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);
}

No entanto, recebo um NoSuchElementException(já que não há invocação de 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)

Eu olhei StreamSupporte Collectionsnão encontrei nada.

gontard
fonte
5
possível duplicata de Como criar um fluxo infinito <E> a partir de um Iterator <E>?
Dmitry Ginzburg
3
@DmitryGinzburg eu não quero criar um fluxo "infinito".
Gontard
1
@DmitryGinzburg Stream.generate(iterator::next)funciona?
Gontard 01/07
1
@DmitryGinzburg Isso não funcionará para um iterador finito.
assylias

Respostas:

543

Uma maneira é criar um Spliterator a partir do Iterator e usá-lo como base para o seu fluxo:

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

Uma alternativa que talvez seja mais legível é usar um Iterable - e criar um Iterable a partir de um Iterator é muito fácil com lambdas porque o Iterable é uma interface funcional:

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

Iterable<String> iterable = () -> sourceIterator;
Stream<String> targetStream = StreamSupport.stream(iterable.spliterator(), false);
assilias
fonte
26
O fluxo é lento: o código está apenas vinculando o fluxo ao iterador, mas a iteração real não acontecerá até uma operação do terminal. Se você usar o iterador enquanto isso, não obterá o resultado esperado. Por exemplo, você pode introduzir um sourceIterator.next()antes de usar o fluxo e verá o efeito (o primeiro item não será visto pelo fluxo).
assylias
9
@assylias, sim, é muito bom! Talvez você possa explicar para futuros leitores essa linha bastante mágica Iterable<String> iterable = () -> sourceIterator;. Eu tenho que admitir que levei algum tempo para entender.
Gontard
7
Eu deveria contar o que encontrei. Iterable<T>é um FunctionalInterfaceque possui apenas um método abstrato iterator(). O mesmo () -> sourceIteratorocorre com uma expressão lambda que instancia uma Iterableinstância como uma implementação anônima.
Jin Kwon
13
Novamente, () -> sourceIterator;é uma forma abreviada denew Iterable<>() { @Override public Iterator<String> iterator() { return sourceIterator; } }
Jin Kwon
7
@JinKwon Na verdade, não é uma forma abreviada de uma classe anônima (existem algumas diferenças sutis, como o escopo e como é compilada), mas se comporta de maneira semelhante nesse caso.
assylias
122

Desde a versão 21, a biblioteca Guava fornece Streams.stream(iterator)

Ele faz o que a resposta de @ assylias mostra .

numéro6
fonte
É muito melhor usar isso de forma consistente até o momento em que o JDK suporta um one-liner nativo. Será muito mais simples encontrar (portanto refatorar) isso no futuro do que as soluções JDK puras mostradas em outros lugares.
drekbour
Isso é excelente, mas… como o Java possui iteradores e fluxos nativos… mas não existe uma maneira direta e direta de ir de um para o outro !? Bastante omissão na minha opinião.
Dan Lenski
92

Ótima sugestão! Aqui está a minha opinião reutilizável:

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);
    }
}

E uso (certifique-se de importar estaticamente comoStream):

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

Isso é possível no Java 9.

Stream.generate(() -> null)
    .takeWhile(x -> iterator.hasNext())
    .map(n -> iterator.next())
    .forEach(System.out::println);
PhilipRoman
fonte
1
Simples, eficiente e sem recorrer à subclasse - essa deve ser a resposta aceita!
Martyglaubitz 16/09/19
1
Infelizmente, eles parecem não funcionar com .parallel()fluxos. Elas também parecem um pouco mais lentas do que as repetidas Spliterator, mesmo para uso seqüencial.
Thomas Ahle
Além disso, o primeiro método é exibido se o iterador estiver vazio. O segundo método funciona por enquanto, mas viola o requisito das funções no map e takeWhile para ser sem estado, então eu hesitaria em fazer isso no código de produção.
Hans-Peter Störr
Realmente, essa deve ser uma resposta aceita. Embora parallelpossa ser descolada, a simplicidade é incrível.
Sven
11

A classe Create Spliteratorfrom Iteratorusing Spliteratorscontém mais de uma função para criar spliterator, por exemplo, aqui estou usando o spliteratorUnknownSizeque está obtendo o iterador como parâmetro e, em seguida, crie Stream usandoStreamSupport

Spliterator<Model> spliterator = Spliterators.spliteratorUnknownSize(
        iterator, Spliterator.NONNULL);
Stream<Model> stream = StreamSupport.stream(spliterator, false);
Bassem Reda Zohdy
fonte
1
import com.google.common.collect.Streams;

e use Streams.stream(iterator):

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

Usar Collections.list(iterator).stream()...

Israel CS Rocha
fonte
6
Embora curto, isso é muito fraco.
Olivier Grégoire
2
Isso desdobra todo o iterador em objeto java e depois o converte em fluxo. Eu não sugerir isso
iec2011007
3
Isso parece ser apenas para enumerações, não iteradores.
31416 John Mayer
1
Não é uma resposta terrível em geral, útil em um beliscão, mas a pergunta menciona desempenho e a resposta não tem desempenho.
Sled