개발관련/other

iterator를 스트림으로 변환하는 방법

Rateye 2021. 8. 28. 16:42
728x90
반응형
질문 : 반복기를 스트림으로 변환하는 방법은 무엇입니까?

IteratorStream 으로 변환하거나보다 구체적으로 iterator를 스트림으로 "보기"하는 간결한 방법을 찾고 있습니다.

성능상의 이유로 새 목록에서 반복자의 복사본을 피하고 싶습니다.

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 살펴 봤지만 아무것도 찾지 못했습니다.

답변

한 가지 방법은 Iterator 에서 Spliterator 를 만들고이를 스트림의 기반으로 사용하는 것입니다.

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);
출처 : https://stackoverflow.com/questions/24511052/how-to-convert-an-iterator-to-a-stream
728x90
반응형