在使用 Java Stream API 进行数据处理时，collect() 方法是一个强大的工具，可以将流中的元素收集到各种数据结构中。当需要将流中的元素收集到一个 EnumSet 时，理解 collect() 方法的正确使用方式至关重要。本文将深入探讨如何为 Stream.collect() 提供正确的 Supplier，以避免常见的编译错误。
理解 Stream.collect() 的三参数形式
Stream.collect() 方法有多种重载形式，其中一种接受三个参数：
<R> R collect(Supplier<R> supplier,
BiConsumer<R, ? super T> accumulator,
BiConsumer<R, R> combiner);- Supplier<R> supplier: 这是一个函数，用于提供一个可变的容器对象，用于累积流中的元素。
- BiConsumer<R, ? super T> accumulator: 这是一个函数，接受容器对象和流中的元素，并将元素添加到容器中。
- BiConsumer<R, R> combiner: 这是一个函数，用于在并行流处理中合并多个容器对象。
关键在于 Supplier 参数必须是一个函数，它能够提供一个新的 EnumSet 实例。直接传递一个 EnumSet 实例会导致类型不匹配的编译错误。
解决方案一：使用三参数 collect() 方法
为了解决这个问题，我们需要提供一个 Supplier，它能够返回一个空的 EnumSet。可以使用 lambda 表达式来实现：
EnumSet<ScConstraint> 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<ScConstraint> 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() 方法可能更灵活。
