9 自定义收集器
在自定义收集器前,我们再确定下 Collector 接口函数接收参数和该实现的方法。
public interface Collector<T, A, R> {
// A 结果类型,T 中间操作类型,R 最终返回类型,一般情况下,A=R
Supplier<A> supplier(); // 源数据对象类型(中间操作对象类型)
BiConsumer<A, T> accumulator();// T 中间操作类型
BinaryOperator<A> combiner(); //并发 合并部分结果
Function<A, R> finisher();// 可选,对结果集的转换
Set<Characteristics> characteristics(); // 当前收集器的特性
enum Characteristics {
CONCURRENT,// 标识收集器是一个并发的
UNORDERED,// 收集操作不能保证保留顺序
IDENTITY_FINISH // 标识 finisher 就是 identity(同一性) 函数,类型转换要求必须成功。
}
}
自定义一个set收集器,我们需要分别实现五个方法,事例代码:
package cn.zxhysy.jdk8.stream2;
import java.util.*;
import java.util.function.BiConsumer;
import java.util.function.BinaryOperator;
import java.util.function.Function;
import java.util.function.Supplier;
import java.util.stream.Collector;
import static java.util.stream.Collector.Characteristics.IDENTITY_FINISH;
import static java.util.stream.Collector.Characteristics.UNORDERED;
public class MySetCollector<T> implements Collector<T, Set<T>, Set<T>> {
@Override
public Supplier supplier() {
System.out.println("suppliier invoked!");
return HashSet::new;
}
@Override
public BiConsumer<Set<T>, T> accumulator() {
System.out.println("accmulator invoked!");
return Set::add;
// return HashSet::add;// 不可以写这个,因为类型可能与supplier中的类型冲突
}
@Override
public BinaryOperator<Set<T>> combiner() {
System.out.println("combiner invoked!");
return (set1, set2) -> {
set1.addAll<