324. Java Stream API - 实现 Collector 接口：自定义你的流式收集器

324. Java Stream API - 实现Collector接口：自定义你的流式收集器

在 Java Stream API 中，Collector是终端操作collect()背后的核心接口。如果你希望完全掌控数据收集的过程——比如构造特定的数据结构、在并行流中定制合并方式等——那就需要了解如何实现这个接口。

🔍 为什么要实现Collector接口？

虽然大多数时候我们可以使用 Java 提供的标准收集器（如toList()、groupingBy()、joining()），但当你需要：

• 创建一个自定义结构（如：合并值到一个特殊格式的字符串、构造非标准 Map 等）
• 控制并行流合并逻辑（比如收集到线程安全容器）
• 优化性能（跳过不必要的转换）

那么实现Collector接口就是最灵活且强大的方式。

☕ 自定义 Collector 的三种方式

📐 理解Collector<T, A, R>的三个类型参数

我们先看一下接口定义：

interfaceCollector<T,A,R>{...}

• T：流中元素的类型
• A：中间可变容器类型（Accumulator 类型）
• R：最终返回结果类型（Result 类型）

🚗 举例说明

➡️ 注意：我们平时用Collectors.toList()等时，看不到A的具体类型，是因为返回类型中对它进行了隐藏（用?代替了）。

📋 代码示例：查看Collector类型参数

Collection<String>strings=List.of("two","three","four","five","six","seven","eight","nine","ten","eleven","twelve");// toListCollector<String,?,List<String>>listCollector=Collectors.toList();List<String>list=strings.stream().collect(listCollector);System.out.println("Size of list = "+list.size());// 11// toSetCollector<String,?,Set<String>>setCollector=Collectors.toSet();Set<String>set=strings.stream().collect(setCollector);System.out.println("Size of set = "+set.size());// 11// groupingBy + countingCollector<String,?,Map<Integer,Long>>groupingCollector=Collectors.groupingBy(String::length,Collectors.counting());Map<Integer,Long>map=strings.stream().collect(groupingCollector);System.out.println("Size of map = "+map.size());// 4

这些例子中，?代表我们无需关心中间类型 A——由 API 实现者决定，我们只关心结果类型 R。

🧪 补充示例：自定义 Collector 实现（首字母统计）

我们实现一个 collector，用于按单词首字母统计出现次数：

classFirstLetterCountCollectorimplementsCollector<String,Map<Character,Integer>,Map<Character,Integer>>{@OverridepublicSupplier<Map<Character,Integer>>supplier(){returnHashMap::new;}@OverridepublicBiConsumer<Map<Character,Integer>,String>accumulator(){return(map,word)->{charfirst=word.charAt(0);map.merge(first,1,Integer::sum);};}@OverridepublicBinaryOperator<Map<Character,Integer>>combiner(){return(map1,map2)->{map2.forEach((k,v)->map1.merge(k,v,Integer::sum));returnmap1;};}@OverridepublicFunction<Map<Character,Integer>,Map<Character,Integer>>finisher(){returnFunction.identity();// 直接返回中间容器}@OverridepublicSet<Characteristics>characteristics(){returnSet.of(Characteristics.IDENTITY_FINISH);}}

使用：

List<String>words=List.of("apple","apricot","banana","blueberry","cherry");Map<Character,Integer>counts=words.stream().collect(newFirstLetterCountCollector());System.out.println(counts);// {a=2, b=2, c=1}

🧠 总结：自定义 Collector 的 5 个核心方法

✅ Collector 特性（characteristics()）常见值