使用Stream.collect()正确收集EnumSet-java教程-PHP中文网

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

聖光之護

发布： 2025-08-20 11:50:01

原创

708人浏览过

使用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<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，编写出高效、简洁的代码。

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