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

以上就是本文的全部内容了，是否有顺利帮助你解决问题？若是能给你带来学习上的帮助，请大家多多支持golang学习网！更多关于文章的相关知识，也可关注golang学习网公众号。