在使用 Java Stream API 时，collect() 方法是一个强大的工具，可以将流中的元素收集到不同的数据结构中。当需要将流中的元素收集到 EnumSet 时，可能会遇到一些问题，特别是关于 Supplier 的使用。
问题在于 Stream.collect() 方法有一个重载版本，它接受三个参数：一个 Supplier，一个 BiConsumer 和另一个 BiConsumer。 很多人容易混淆 Supplier 的作用，导致编译错误。
理解 Stream.collect() 的三个参数版本
Stream.collect(Supplier
-
Supplier
supplier : 一个函数，用于创建一个新的可变结果容器。 每次调用都会创建一个新的容器。 -
BiConsumer
accumulator : 一个函数，用于将流中的元素合并到结果容器中。 -
BiConsumer
combiner : 一个函数，用于合并两个结果容器。 在并行流处理中，多个结果容器会被合并成一个。
关键在于 Supplier 必须是一个函数，它提供一个新的可变对象，用于累积流中的元素，并作为流执行的结果返回。
正确使用 Supplier 创建 EnumSet
以下是一个正确的示例，展示了如何使用 Stream.collect() 将流中的元素收集到 EnumSet 中：
Lht蓝海豚(大维)团购导航系统是一套开源程序，采用PHP+MySql平台开发，具有强大的多规则API采集功能、精确化引导消费功能，卓越的负载能力和访问速度，全面支持第三方整合（微博、短信等）前台功能团购名站：显示分类下的所有团购网站,点击团购站,可直接进行查看、推荐、收藏精品商城：为网民推荐展示非团购网站的网上商城今日团购：显示各团购网站正在进行的团购团购排行：今日团购商品可按价格、行业、折扣、
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 nlsCandidates = new HashSet<>();
nlsCandidates.add(1);
nlsCandidates.add(2);
EnumSet[] rowConstraints = new EnumSet[20];
for (int i = 0; i < 20; i++) {
rowConstraints[i] = EnumSet.of(ScConstraint.CONSTRAINT_1, ScConstraint.CONSTRAINT_2);
}
Set s = EnumSet.of(ScConstraint.CONSTRAINT_1);
int k = 0;
EnumSet 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
使用 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 nlsCandidates = new HashSet<>();
nlsCandidates.add(1);
nlsCandidates.add(2);
EnumSet[] rowConstraints = new EnumSet[20];
for (int i = 0; i < 20; i++) {
rowConstraints[i] = EnumSet.of(ScConstraint.CONSTRAINT_1, ScConstraint.CONSTRAINT_2);
}
Set s = EnumSet.of(ScConstraint.CONSTRAINT_1);
int k = 0;
EnumSet 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。 通过理解这些概念，可以避免常见的编译错误，并编写出更清晰、更高效的代码。
