在 Java 8 中作为流的空安全集合

Question

我正在寻找可以进行收集流但为空安全的方法。如果集合为 null，则返回空流。喜欢这个：

Utils.nullSafeStream(collection).filter(...);

我创建了自己的方法：

public static <T> Stream<T> nullSafeStream(Collection<T> collection) {
    if (collection == null) {
        return Stream.empty();
    }
    return collection.stream();
}

但我很好奇，标准 JDK 中是否有这样的东西？

Answer 1

您可以使用Optional：

Optional.ofNullable(collection).orElse(Collections.emptySet()).stream()...

我任意选择Collections.emptySet()作为默认值，以防collection为 null。这将导致stream()方法调用生成空Stream如果collection为 null。

例：

Collection<Integer> collection = Arrays.asList (1,2,3);
System.out.println (Optional.ofNullable(collection).orElse(Collections.emptySet()).stream().count ());
collection = null;
System.out.println (Optional.ofNullable(collection).orElse(Collections.emptySet()).stream().count ());

输出：

3
0

或者，正如 marstran 建议的那样，您可以使用：

Optional.ofNullable(collection).map(Collection::stream).orElse(Stream.empty ())...

Answer 2

您可以使用 org.apache.commons.collections4.CollectionUtils：：emptyIfNull 函数：

import static org.apache.commons.collections4.CollectionUtils.emptyIfNull;
      
emptyIfNull(list).stream()
                 .filter(...);

Answer 3

不确定是否有帮助，但从 Java 9 开始，您可以编写：

Stream.ofNullable(nullableCollection)
      .flatMap(Collection::stream)

Answer 4

您的collectionAsStream()方法可以简化为比使用Optional更简单的版本：

public static <T> Stream<T> collectionAsStream(Collection<T> collection) {
    return collection == null ? Stream.empty() : collection.stream();
}

请注意，在大多数情况下，最好在构建流管道之前测试空值：

if (collection != null) {
    collection.stream().filter(...)
} // else do nothing

你想要的东西似乎只在你需要返回流（包括平面映射）时有用，或者可能将其与另一个流连接起来。

Answer 5

你可以使用这样的东西：

public static void main(String [] args) {
    List<String> someList = new ArrayList<>();
    asStream(someList).forEach(System.out::println);
}
public static <T> Stream<T> asStream(final Collection<T> collection) {
    return Optional.ofNullable(collection)
            .map(Collection::stream)
            .orElseGet(Stream::empty);
}

Answer 6

如果下载库org.apache.commons.collections4不是一种选择，您可以编写自己的包装器/便利方法。

public static <T> Stream<T> asStream(final Collection<T> collection) {
    return collection == null ? ( Stream<T> ) Collections.emptyList().stream() 
                              : collection.stream();
}

或者用Optional.ofNullable包装集合

public static <T> Stream<T> asStream(final Collection<T> collection) {
    return Optional.ofNullable(collection)
            .orElse( Collections.emptySet()).stream();
}