Làm thế nào để chuyển đổi một iterator thành một luồng?

Tôi đang tìm kiếm một cách ngắn gọn để chuyển đổi Iteratormột Streamhoặc cụ thể hơn để "xem" trình vòng lặp dưới dạng luồng.

Vì lý do hiệu suất, tôi muốn tránh một bản sao của trình lặp trong danh sách mới:

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

Dựa trên một số gợi ý trong các bình luận, tôi cũng đã thử sử dụng 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);
}

Tuy nhiên, tôi nhận được một NoSuchElementException(vì không có lời mời 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)

Tôi đã xem StreamSupportvà Collectionstôi không tìm thấy gì cả.

Một cách là tạo Spliterator từ Iterator và sử dụng nó làm cơ sở cho luồng của bạn:

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

Một cách khác có thể dễ đọc hơn là sử dụng Iterable - và tạo Iterable từ Iterator rất dễ dàng với lambdas vì Iterable là giao diện chức năng:

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

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

Kể từ phiên bản 21, thư viện Guava cung cấp Streams.stream(iterator)

Nó thực hiện những gì câu trả lời của @ assylias cho thấy .

Đề nghị tuyệt vời! Đây là tái sử dụng của tôi trên nó:

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

Và cách sử dụng (đảm bảo nhập tĩnh asStream):

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

Điều này có thể có trong Java 9.

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

Tạo Spliteratortừ Iteratorviệc sử dụng Spliteratorslớp chứa nhiều hơn một hàm để tạo bộ chia, ví dụ ở đây đang sử dụng spliteratorUnknownSizecái đang lấy iterator làm tham số, sau đó tạo Stream bằng cách sử dụngStreamSupport

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

import com.google.common.collect.Streams;

và sử dụng Streams.stream(iterator):

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

Sử dụng Collections.list(iterator).stream()...