要实现自定义Collector,必须正确实现supplier、accumulator、combiner,可选finisher和characteristics;例如统计字符串长度分布可用HashMap::new、merge计数、forEach合并,并设IDENTITY_FINISH特征。

要实现一个符合 Collector 接口的自定义变量收集器,核心是正确实现三个函数式组件:supplier(供应者)、accumulator(累加器)和 combiner(合并器),并可选地提供 finisher(终结器)与 characteristics(特征标识)。Java 的 Collectors 工具类背后正是这套机制,而手动编写能让你精准控制聚合逻辑,比如按类型分组、去重计数、构建不可变容器,或收集特定字段到自定义对象中。
明确 Collector 的四大组成部分
一个合法的 Collector<t a r></t> 需要明确定义:
-
supplier():返回一个空的中间容器(如
ArrayList::new、HashMap::new或自定义可变对象); -
accumulator():定义如何将每个元素
T“加入”中间容器A(例如list.add(t)或map.merge(...)); -
combiner():定义两个中间容器
A如何合并(用于并行流,必须满足结合律,如(a1, a2) -> { a1.addAll(a2); return a1; }); -
finisher():将中间容器
A转为最终结果R(如ArrayList::new→ImmutableList.copyOf);若无需转换,可返回Function.identity(); -
characteristics():声明行为特征,常见有
CONCURRENT(支持并发修改)、UNORDERED(不依赖顺序)、IDENTITY_FINISH(finisher 是恒等函数)——正确声明能提升性能或启用并行优化。
实战:收集字符串长度分布(Map)
目标:统计流中每个字符串的长度,输出「长度 → 出现次数」的映射。这比 Collectors.groupingBy(String::length, Collectors.counting()) 更轻量,且完全可控。
代码示例:
Collector<string map long>, Map<integer long>> lengthFreqCollector =
Collector.<string map long>, Map<integer long>>of(
HashMap::new,
(map, s) -> map.merge(s.length(), 1L, Long::sum),
(m1, m2) -> {
m2.forEach((len, count) -> m1.merge(len, count, Long::sum));
return m1;
},
Function.identity(),
Collector.Characteristics.IDENTITY_FINISH
);</integer></string></integer></string>
说明:
- 中间容器与结果类型一致(
Map<integer long></integer>),所以 finisher 是恒等函数; -
merge同时处理 key 不存在和已存在的情况,线程安全需配合ConcurrentHashMap和CONCURRENT特征; - 若想支持并行且避免同步开销,可将 supplier 改为
ConcurrentHashMap::new,combiner 改为ConcurrentHashMap::putAll,并添加CONCURRENT和UNORDERED特征。
进阶:构建不可变聚合对象(如 StatsSummary)
假设需要从 double 流中一次性计算 count、sum、min、max、average —— 用多个 collectors 分别计算效率低,且 average 需依赖 count/sum。此时自定义 collector 最合适。
步骤:
- 定义中间状态类
StatsAccumulator(含 mutable 字段:count、sum、min=+∞、max=−∞); - accumulator 中逐个更新字段(
acc.count++、acc.sum += d、acc.min = Math.min(acc.min, d)等); - combiner 实现「两份统计合并」:count 相加、sum 相加、min 取小、max 取大;
- finisher 将 accumulator 转为不可变
StatsSummary(含 final 字段 + builder 构造); - characteristics 建议设为
CONCURRENT(因 accumulator 是线程安全更新)和UNORDERED(统计不依赖顺序)。
避坑提醒:常见错误与验证方式
写完 collector 后务必验证其正确性,尤其并行场景:
-
combiner 不满足结合律:例如用
list1.addAll(list2)是 OK 的,但用list1.add(list2.get(0))就不行; -
supplier 返回共享实例:不能写
() -> sharedList,每次必须新建容器; - 忽略并发安全:并行流下,多个线程可能同时调用 accumulator,若中间容器非线程安全(如 ArrayList),需加锁或换用线程安全类型;
-
characteristics 声明错误:标了
CONCURRENT却没用线程安全容器,会导致数据竞争;标了IDENTITY_FINISH却提供了非恒等 finisher,会抛异常; - 验证方法:对同一数据源分别用
stream().collect(c)和parallelStream().collect(c),比对结果是否一致。










