使用Stream.collect()正确收集EnumSet

使用Stream.collect()正确收集EnumSet

第一段引用上面的摘要:

本文介绍了在使用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。 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,编写出高效、简洁的代码。

© 版权声明
THE END
喜欢就支持一下吧
点赞8 分享