第一段引用上面的摘要:
本文介绍了在使用Java Stream API的collect()方法收集EnumSet时,如何正确提供Supplier。常见的错误是直接提供一个EnumSet实例,而不是一个生成EnumSet实例的Supplier。本文将详细讲解collect()方法的用法,并提供两种正确的实现方式,帮助开发者避免常见的编译错误。
理解Stream.collect()方法
Stream.collect()方法是Java Stream API中一个强大的终端操作,它允许我们将流中的元素收集到各种数据结构中,例如List、Set、map等。它有多种重载形式,其中一种形式接受三个参数:
- Supplier
supplier: 提供一个用于累积结果的容器。这个容器必须是可变的,并且必须是一个新的实例,而不是共享的。 - BiConsumer
accumulator: 定义如何将流中的元素添加到容器中。 - BiConsumer
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
方式二:使用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,编写出高效、简洁的代码。