Chuyển đổi Iterable thành Stream bằng Java 8 JDK

Tôi có một giao diện trả về java.lang.Iterable<T>.

Tôi muốn thao tác kết quả đó bằng cách sử dụng API Java 8 Stream.

Tuy nhiên Iterable không thể "stream".

Bạn có biết cách sử dụng Iterable dưới dạng Stream mà không chuyển đổi nó thành List không?

Có một câu trả lời tốt hơn nhiều so với sử dụng spliteratorUnknownSizetrực tiếp, cả hai đều dễ dàng hơn và có kết quả tốt hơn. Iterablecó một spliterator()phương thức, vì vậy bạn chỉ nên sử dụng phương thức đó để lấy bộ chia của bạn. Trong trường hợp xấu nhất, đó là cùng một mã (cách triển khai mặc định sử dụng spliteratorUnknownSize), nhưng trong trường hợp phổ biến hơn, trong đó bạn Iterableđã là một bộ sưu tập, bạn sẽ có được một trình phân tách tốt hơn và do đó hiệu suất truyền phát tốt hơn (thậm chí có thể song song tốt). Nó cũng ít mã hơn:

StreamSupport.stream(iterable.spliterator(), false)
             .filter(...)
             .moreStreamOps(...);

Như bạn có thể thấy, nhận được một luồng từ Iterable(cũng xem câu hỏi này ) không phải là rất đau đớn.

Nếu bạn có thể sử dụng thư viện Guava, kể từ phiên bản 21, bạn có thể sử dụng

Streams.stream(iterable)

Bạn có thể dễ dàng tạo Streamra một Iterablehoặc Iterator:

public static <T> Stream<T> stream(Iterable<T> iterable) {
    return StreamSupport.stream(
        Spliterators.spliteratorUnknownSize(
            iterable.iterator(),
            Spliterator.ORDERED
        ),
        false
    );
}

Tôi muốn đề xuất sử dụng thư viện JOOL , nó ẩn ma thuật spliterator đằng sau cuộc gọi Seq.seq (có thể lặp lại) và cũng cung cấp một loạt các chức năng hữu ích bổ sung.

Vì vậy, như một câu trả lời khác được đề cập, Guava đã hỗ trợ cho việc này bằng cách sử dụng:

Streams.stream(iterable);

Tôi muốn nhấn mạnh rằng việc thực hiện làm một cái gì đó hơi khác so với các câu trả lời khác được đề xuất. Nếu Iterablelà loại Collectionhọ đúc nó.

public static <T> Stream<T> stream(Iterable<T> iterable) {
  return (iterable instanceof Collection)
    ? ((Collection<T>) iterable).stream()
    : StreamSupport.stream(iterable.spliterator(), false);
}

public static <T> Stream<T> stream(Iterator<T> iterator) {
  return StreamSupport.stream(
    Spliterators.spliteratorUnknownSize(iterator, 0),
    false
  );
}

Tôi đã tạo lớp này:

public class Streams {
    /**
     * Converts Iterable to stream
     */
    public static <T> Stream<T>  streamOf(final Iterable<T> iterable) {
        return toStream(iterable, false);
    }

    /**
     * Converts Iterable to parallel stream
     */
    public static <T> Stream<T> parallelStreamOf(final Iterable<T> iterable) {
        return toStream(iterable, true);
    }

    private static <T> Stream<T> toStream(final Iterable<T> iterable, final boolean isParallel) {
        return StreamSupport.stream(iterable.spliterator(), isParallel);
    }
}

Tôi nghĩ rằng nó hoàn toàn có thể đọc được vì bạn không phải suy nghĩ về bộ chia và booleans (isParallel).

Một cách giải quyết rất đơn giản cho vấn đề này là tạo ra một Streamable<T>giao diện mở rộng Iterable<T>chứa một default <T> stream()phương thức.

interface Streamable<T> extends Iterable<T> {
    default Stream<T> stream() {
        return StreamSupport.stream(spliterator(), false);
    }
}

Bây giờ bất kỳ ai trong số bạn Iterable<T>có thể được truyền phát một cách tầm thường chỉ bằng cách khai báo chúng implements Streamable<T>thay vì Iterable<T>.

Nếu bạn tình cờ sử dụng Vavr (trước đây gọi là Javaslang), việc này có thể dễ dàng như:

Iterable i = //...
Stream.ofAll(i);

Một cách khác để làm điều đó, với Java 8 và không có lib bên ngoài:

Stream.concat(collectionA.stream(), collectionB.stream())
      .collect(Collectors.toList())