在使用 Java Stream API 时，collect() 方法是一个强大的工具，可以将流中的元素收集到不同的数据结构中。当需要将流中的元素收集到 EnumSet 时，可能会遇到一些问题，特别是关于 Supplier 的使用。
问题在于 Stream.collect() 方法有一个重载版本，它接受三个参数：一个 Supplier，一个 BiConsumer 和另一个 BiConsumer。 很多人容易混淆 Supplier 的作用，导致编译错误。
理解 Stream.collect() 的三个参数版本
Stream.collect(Supplier<R>, BiConsumer<R, ? super T>, BiConsumer<R, R>) 方法的三个参数分别代表：
-
Supplier
supplier : 一个函数，用于创建一个新的可变结果容器。 每次调用都会创建一个新的容器。 - BiConsumer<R, ? super T> accumulator: 一个函数，用于将流中的元素合并到结果容器中。
- BiConsumer<R, R> combiner: 一个函数，用于合并两个结果容器。 在并行流处理中，多个结果容器会被合并成一个。
关键在于 Supplier 必须是一个函数，它提供一个新的可变对象，用于累积流中的元素，并作为流执行的结果返回。
正确使用 Supplier 创建 EnumSet
以下是一个正确的示例，展示了如何使用 Stream.collect() 将流中的元素收集到 EnumSet 中：
import java.util.EnumSet;
import java.util.HashSet;
import java.util.Set;
import java.util.stream.Collectors;
import java.util.stream.Stream;
enum ScConstraint {
CONSTRAINT_1,
CONSTRAINT_2,
CONSTRAINT_3
}
public class EnumSetCollector {
public static void main(String[] args) {
HashSet<Integer> nlsCandidates = new HashSet<>();
nlsCandidates.add(1);
nlsCandidates.add(2);
EnumSet<ScConstraint>[] rowConstraints = new EnumSet[20];
for (int i = 0; i < 20; i++) {
rowConstraints[i] = EnumSet.of(ScConstraint.CONSTRAINT_1, ScConstraint.CONSTRAINT_2);
}
Set<ScConstraint> s = EnumSet.of(ScConstraint.CONSTRAINT_1);
int k = 0;
EnumSet<ScConstraint> tbd = nlsCandidates.stream()
.flatMap(p -> rowConstraints[10 * k + p].stream())
.filter(cstr -> !s.contains(cstr))
.collect(
() -> EnumSet.noneOf(ScConstraint.class),
Set::add,
Set::addAll
);
System.out.println(tbd); // 输出: [CONSTRAINT_2]
}
}在这个例子中，() -> EnumSet.noneOf(ScConstraint.class) 是 Supplier，它返回一个空的 EnumSet<ScConstraint>。Set::add 将流中的元素添加到 EnumSet 中，Set::addAll 合并两个 EnumSet。
使用 Collectors.toCollection() 简化代码
另一种更简洁的方式是使用 Collectors.toCollection() 方法：
import java.util.EnumSet;
import java.util.HashSet;
import java.util.Set;
import java.util.stream.Collectors;
import java.util.stream.Stream;
enum ScConstraint {
CONSTRAINT_1,
CONSTRAINT_2,
CONSTRAINT_3
}
public class EnumSetCollector {
public static void main(String[] args) {
HashSet<Integer> nlsCandidates = new HashSet<>();
nlsCandidates.add(1);
nlsCandidates.add(2);
EnumSet<ScConstraint>[] rowConstraints = new EnumSet[20];
for (int i = 0; i < 20; i++) {
rowConstraints[i] = EnumSet.of(ScConstraint.CONSTRAINT_1, ScConstraint.CONSTRAINT_2);
}
Set<ScConstraint> s = EnumSet.of(ScConstraint.CONSTRAINT_1);
int k = 0;
EnumSet<ScConstraint> tbd = nlsCandidates.stream()
.flatMap(p -> rowConstraints[10 * k + p].stream())
.filter(cstr -> !s.contains(cstr))
.collect(Collectors.toCollection(() -> EnumSet.noneOf(ScConstraint.class)));
System.out.println(tbd); // 输出: [CONSTRAINT_2]
}
}Collectors.toCollection(() -> EnumSet.noneOf(ScConstraint.class)) 创建一个 Collector，它将流中的元素收集到一个新的 EnumSet 中。 Supplier 仍然是必需的，用于创建空的 EnumSet。
注意事项
- 确保 Supplier 提供的是一个新的实例，而不是一个静态的实例。 如果使用静态实例，可能会导致多个流操作共享同一个 EnumSet，从而产生意想不到的结果。
- 在使用并行流时，combiner 函数是必需的。 它用于合并多个线程产生的中间结果。
总结
正确使用 Stream.collect() 收集 EnumSet 的关键在于理解 Supplier 的作用。 Supplier 必须提供一个新的可变结果容器。 Collectors.toCollection() 提供了一种更简洁的方式来收集 EnumSet。 通过理解这些概念，可以避免常见的编译错误，并编写出更清晰、更高效的代码。
