第一段引用上面的摘要：
本文介绍了在使用Java Stream API的collect()方法收集EnumSet时，如何正确提供Supplier。常见的错误是直接提供一个EnumSet实例，而不是一个生成EnumSet实例的Supplier。本文将详细讲解collect()方法的用法，并提供两种正确的实现方式，帮助开发者避免常见的编译错误。
理解Stream.collect()方法
Stream.collect()方法是Java Stream API中一个强大的终端操作，它允许我们将流中的元素收集到各种数据结构中，例如List、Set、Map等。它有多种重载形式，其中一种形式接受三个参数：
-
Supplier
supplier: 提供一个用于累积结果的容器。这个容器必须是可变的，并且必须是一个新的实例，而不是共享的。 - BiConsumer<R, ? super T> accumulator: 定义如何将流中的元素添加到容器中。
- BiConsumer<R, R> combiner: 定义如何合并两个容器。这个参数在并行流中非常重要，用于合并不同线程的结果。
当我们需要收集到EnumSet时，常见的错误是直接将EnumSet.noneOf(ScConstraint.class)作为supplier参数传递，导致编译错误。这是因为supplier需要是一个Supplier接口的实现，而不是一个EnumSet实例。
正确的实现方式
方式一：使用Lambda表达式提供Supplier
正确的做法是使用Lambda表达式来创建一个Supplier，它会在每次调用时返回一个新的空的EnumSet。
import java.util.EnumSet;
import java.util.HashSet;
import java.util.Set;
import java.util.stream.Stream;
public class EnumSetCollector {
public static void main(String[] args) {
HashSet<Integer> nlsCandidates = new HashSet<>();
nlsCandidates.add(1);
nlsCandidates.add(2);
nlsCandidates.add(3);
EnumSet<ScConstraint>[] rowConstraints = new EnumSet[30];
for (int i = 0; i < 30; i++) {
rowConstraints[i] = EnumSet.allOf(ScConstraint.class);
}
Set<ScConstraint> s = EnumSet.of(ScConstraint.CONSTRAINT_A);
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), // Supplier
Set::add, // Accumulator
Set::addAll // Combiner
);
System.out.println(tbd);
}
enum ScConstraint {
CONSTRAINT_A,
CONSTRAINT_B,
CONSTRAINT_C
}
}在这个例子中，() -> EnumSet.noneOf(ScConstraint.class)就是一个Supplier，它会在每次调用时返回一个新的空的EnumSet<ScConstraint>。 Set::add 作为accumulator，将流中的每个元素添加到EnumSet中。Set::addAll 作为combiner，合并两个EnumSet。
方式二：使用Collectors.toCollection()
另一种更简洁的方式是使用Collectors.toCollection()，它接受一个Supplier作为参数，并返回一个Collector。
import java.util.EnumSet;
import java.util.HashSet;
import java.util.Set;
import java.util.stream.Collectors;
public class EnumSetCollector {
public static void main(String[] args) {
HashSet<Integer> nlsCandidates = new HashSet<>();
nlsCandidates.add(1);
nlsCandidates.add(2);
nlsCandidates.add(3);
EnumSet<ScConstraint>[] rowConstraints = new EnumSet[30];
for (int i = 0; i < 30; i++) {
rowConstraints[i] = EnumSet.allOf(ScConstraint.class);
}
Set<ScConstraint> s = EnumSet.of(ScConstraint.CONSTRAINT_A);
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);
}
enum ScConstraint {
CONSTRAINT_A,
CONSTRAINT_B,
CONSTRAINT_C
}
}这种方式更加简洁，可读性也更好。 Collectors.toCollection() 内部已经处理了accumulator和combiner的逻辑，只需要提供一个创建EnumSet的supplier即可。
注意事项
- 确保Supplier返回的是一个新的EnumSet实例，而不是共享的实例。
- 在并行流中使用collect()时，combiner参数非常重要，需要正确地合并不同线程的结果。
- 选择合适的收集器，Collectors.toCollection()通常是收集到EnumSet的最佳选择。
总结
在使用Stream.collect()方法收集EnumSet时，正确提供Supplier是关键。避免直接提供EnumSet实例，而是使用Lambda表达式或Collectors.toCollection()来创建Supplier。理解collect()方法的各个参数的含义，可以帮助我们更好地利用Stream API，编写出高效、简洁的代码。
