使用 Stream.collect() 正确收集 EnumSet 的方法

在使用 Java Stream API 进行数据处理时，collect() 方法是一个强大的工具，可以将流中的元素收集到各种数据结构中。当需要将流中的元素收集到一个 EnumSet 时，理解 collect() 方法的正确使用方式至关重要。本文将深入探讨如何为 Stream.collect() 提供正确的 Supplier，以避免常见的编译错误。

理解 Stream.collect() 的三参数形式

Stream.collect() 方法有多种重载形式，其中一种接受三个参数：

 R collect(Supplier supplier,
              BiConsumer accumulator,
              BiConsumer combiner);

• Supplier supplier: 这是一个函数，用于提供一个可变的容器对象，用于累积流中的元素。
• BiConsumer accumulator: 这是一个函数，接受容器对象和流中的元素，并将元素添加到容器中。
• BiConsumer combiner: 这是一个函数，用于在并行流处理中合并多个容器对象。

关键在于 Supplier 参数必须是一个函数，它能够提供一个新的 EnumSet 实例。直接传递一个 EnumSet 实例会导致类型不匹配的编译错误。

解决方案一：使用三参数 collect() 方法

为了解决这个问题，我们需要提供一个 Supplier，它能够返回一个空的 EnumSet。可以使用 lambda 表达式来实现：

EnumSet tbd = nlsCandidates.stream()
    .flatMap(p -> rowConstraints[10 * k + p].stream())
    .filter(cstr -> !s.contains(cstr.RC))
    .collect(
        () -> EnumSet.noneOf(ScConstraint.class), // Supplier: 提供一个新的空 EnumSet
        Set::add,                                   // Accumulator: 将元素添加到 EnumSet
        Set::addAll                                  // Combiner: 合并 EnumSet (用于并行流)
    );

在这个例子中，() -> EnumSet.noneOf(ScConstraint.class) 是一个 Supplier，它在每次调用时都会创建一个新的、空的 EnumSet。Set::add 是一个 BiConsumer，它将流中的元素添加到 EnumSet 中。Set::addAll 是一个 BiConsumer，它用于在并行流处理中合并多个 EnumSet。

解决方案二：使用 Collectors.toCollection()

另一种更简洁的方法是使用 Collectors.toCollection() 方法。它接受一个 Supplier 作为参数，并返回一个 Collector，该 Collector 将流中的元素收集到由 Supplier 提供的集合中。

import java.util.stream.Collectors;

EnumSet tbd = nlsCandidates.stream()
    .flatMap(p -> rowConstraints[10 * k + p].stream())
    .filter(cstr -> !s.contains(cstr.RC))
    .collect(Collectors.toCollection(
        () -> EnumSet.noneOf(ScConstraint.class)  // Supplier: 提供一个新的空 EnumSet
    ));

在这个例子中，Collectors.toCollection(() -> EnumSet.noneOf(ScConstraint.class)) 创建了一个 Collector，它将流中的元素收集到一个新的、空的 EnumSet 中。

总结

当使用 Stream.collect() 方法收集 EnumSet 时，务必确保 Supplier 参数是一个函数，它能够提供一个新的 EnumSet 实例。可以使用 lambda 表达式或 Collectors.toCollection() 方法来实现。理解 Supplier 的作用是避免编译错误的关键。选择哪种方法取决于个人偏好和代码的可读性。Collectors.toCollection() 通常更简洁，但在某些情况下，三参数 collect() 方法可能更灵活。