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

在使用 Java Stream API 进行数据处理时，collect() 方法是一个强大的工具，可以将流中的元素收集到各种数据结构中。当需要将流中的元素收集到一个 EnumSet 时，理解 collect() 方法的正确使用方式至关重要。本文将深入探讨如何为 Stream.collect() 提供正确的 Supplier，以避免常见的编译错误。

理解 Stream.collect() 的三参数形式

Stream.collect() 方法有多种重载形式，其中一种接受三个参数：

<R> R collect(Supplier<R> supplier,
              BiConsumer<R, ? super T> accumulator,
              BiConsumer<R, R> combiner);

• Supplier<R> supplier: 这是一个函数，用于提供一个可变的容器对象，用于累积流中的元素。
• BiConsumer<R, ? super T> accumulator: 这是一个函数，接受容器对象和流中的元素，并将元素添加到容器中。
• BiConsumer<R, R> combiner: 这是一个函数，用于在并行流处理中合并多个容器对象。

关键在于 Supplier 参数必须是一个函数，它能够提供一个新的 EnumSet 实例。直接传递一个 EnumSet 实例会导致类型不匹配的编译错误。

解决方案一：使用三参数 collect() 方法

为了解决这个问题，我们需要提供一个 Supplier，它能够返回一个空的 EnumSet。可以使用 lambda 表达式来实现：

EnumSet<ScConstraint> tbd = nlsCandidates.stream()
    .flatMap(p -> rowConstraints[10 * k + p].stream())
    .filter(cstr -> !s.contains(cstr.RC))
    .collect(
        () -> EnumSet.noneOf(ScConstraint.class), // Supplier: 提供一个新的空 EnumSet
        Set::add,                                   // Accumulator: 将元素添加到 EnumSet
        Set::addAll                                  // Combiner: 合并 EnumSet (用于并行流)
    );

在这个例子中，() -> EnumSet.noneOf(ScConstraint.class) 是一个 Supplier，它在每次调用时都会创建一个新的、空的 EnumSet。Set::add 是一个 BiConsumer，它将流中的元素添加到 EnumSet 中。Set::addAll 是一个 BiConsumer，它用于在并行流处理中合并多个 EnumSet。

法语写作助手

法语助手旗下的AI智能写作平台，支持语法、拼写自动纠错，一键改写、润色你的法语作文。

31

查看详情

解决方案二：使用 Collectors.toCollection()

另一种更简洁的方法是使用 Collectors.toCollection() 方法。它接受一个 Supplier 作为参数，并返回一个 Collector，该 Collector 将流中的元素收集到由 Supplier 提供的集合中。

import java.util.stream.Collectors;

EnumSet<ScConstraint> tbd = nlsCandidates.stream()
    .flatMap(p -> rowConstraints[10 * k + p].stream())
    .filter(cstr -> !s.contains(cstr.RC))
    .collect(Collectors.toCollection(
        () -> EnumSet.noneOf(ScConstraint.class)  // Supplier: 提供一个新的空 EnumSet
    ));

在这个例子中，Collectors.toCollection(() -> EnumSet.noneOf(ScConstraint.class)) 创建了一个 Collector，它将流中的元素收集到一个新的、空的 EnumSet 中。

总结

当使用 Stream.collect() 方法收集 EnumSet 时，务必确保 Supplier 参数是一个函数，它能够提供一个新的 EnumSet 实例。可以使用 lambda 表达式或 Collectors.toCollection() 方法来实现。理解 Supplier 的作用是避免编译错误的关键。选择哪种方法取决于个人偏好和代码的可读性。Collectors.toCollection() 通常更简洁，但在某些情况下，三参数 collect() 方法可能更灵活。

以上就是使用 Stream.collect() 正确收集 EnumSet 的方法的详细内容，更多请关注php中文网其它相关文章！