探索Java Stream API自定义收集器的魅力

在Java 8引入的Stream API中，收集器（Collector）是一个强大的工具，它允许开发者以声明式的方式处理集合数据。尽管内置的收集器如toList、toSet和groupingBy等已经覆盖了许多常见场景，但在某些特定情况下，自定义收集器却能带来意想不到的灵活性和效率提升。

自定义收集器的必要性

假设我们需要对一个学生列表进行处理，最终得到一个按成绩分组的学生姓名列表。虽然使用内置的groupingBy可以实现分组，但要进一步处理分组结果（如提取姓名列表）则需要额外的步骤。这时，自定义收集器便能大展身手。

自定义收集器的构建

一个完整的收集器由三个核心组件构成：supplier、accumulator和combiner，以及两个可选组件：finisher和characteristics。

Supplier

supplier负责提供初始容器。例如，如果我们需要收集学生姓名，可以使用ArrayList::new。

Supplier<ArrayList<String>> supplier = ArrayList::new;

Accumulator

accumulator定义了如何将元素添加到容器中。对于学生姓名的收集，我们可以这样实现：

BiConsumer<ArrayList<String>, Student> accumulator = (list, student) -> list.add(student.getName());

Combiner

combiner用于并行处理时的结果合并。假设有两个列表需要合并，我们可以使用如下实现：

BinaryOperator<ArrayList<String>> combiner = (list1, list2) -> {
    list1.addAll(list2);
    return list1;
};

Finisher

finisher是可选的，用于对最终结果进行转换。如果不需要额外处理，可以直接使用Function.identity()。

Function<ArrayList<String>, ArrayList<String>> finisher = Function.identity();

Characteristics

characteristics定义了收集器的特性，如CONCURRENT、UNORDERED等。

Set<Characteristics> characteristics = Collections.unmodifiableSet(EnumSet.of(Characteristics.IDENTITY_FINISH));

完整的自定义收集器

将上述组件组合起来，我们便得到了一个完整的自定义收集器：

Collector<Student, ArrayList<String>, ArrayList<String>> studentNameCollector = Collector.of(
    ArrayList::new,
    (list, student) -> list.add(student.getName()),
    (list1, list2) -> {
        list1.addAll(list2);
        return list1;
    },
    Function.identity(),
    Characteristics.IDENTITY_FINISH
);

应用自定义收集器

现在，我们可以使用这个自定义收集器来处理学生列表：

List<Student> students = Arrays.asList(
    new Student("Alice", 90),
    new Student("Bob", 85),
    new Student("Charlie", 90)
);

ArrayList<String> names = students.stream()
    .collect(studentNameCollector);

System.out.println(names); // 输出: [Alice, Bob, Charlie]