前言
以前写过Java8中的自定义收集器,当时只是在文章末尾放了个例子,觉得基本用法挺简单,而且有些东西没搞懂(比如combiner方法到底做什么的),没有专门写,过了一段时间又忘了,所以,即使还是没搞懂combiner方法,还是硬着头皮把使用的经验记录下,方便以后参考。
简介
要实现自定义收集器,只需要实现java.util.stream.Collector<T, A, R>
接口即可,这个接口包含五个无参方法:[supplier, accumulator, combiner, finisher, characteristics]。
泛型简介
泛型含义如下:
- T:缩减操作的输入元素的类型
- A:还原操作的可变累积类型(通常隐藏为实现细节)
- R:还原操作的结果类型
T
不必说,收集什么泛型的列表,就输入什么类型,比如我对一个Student列表进行收集计算,那么T肯定是Student。A
是计算过程中用来盛放计算结果的容器,一般都是List,Set等等。R
就比较好理解,就是收集完成后返回的类型,需要注意的是,当characteristics()中包含Characteristics.IDENTITY_FINISH时,。
方法简介
- characteristics 表示收集计算的方式,返回类型为
Set<Characteristics>
,其中Characteristics
是一个枚举类型,指示收集器属性的特征,可用于优化缩减实现。它只有三个值[CONCURRENT, UNORDERED, IDENTITY_FINISH]
,注释翻译过来分别是:1.表示此收集器是并发的,这意味着结果容器可以支持与来自多个线程的相同结果容器同时调用的累加器函数。如果CONCURRENT收集器也不是UNORDERED,那么只有在应用于无序数据源时才应同时评估它。2.指示集合操作不承诺保留输入元素的遭遇顺序。(如果结果容器没有内在顺序,例如Set,则可能是这样。)3.表示整理器功能是标识功能,可以省略。 如果设置,则必须是从A到R的未经检查的强制转换成功的情况。 - supplier 该方法返回一个
Supplier<A>
类型的结果,表示在计算过程中,如何初始化一个临时容器,比如A=List,那么一般返回ArrayList::new - accumulator 核心方法,关键的计算逻辑都放在这里,定义了如何把一个个元素放入临时容器中,返回类型为
BiConsumer<A, T>
- combiner 返回一个
BinaryOperator<A>
类型的结果,个人理解是如何合并临时容器,但是在实际应用中没碰到执行过,所以一般直接返回null - finisher 定义了如何把临时容器转换为输出结果,返回类型为
Function<A, R>
,需要注意的是,当方法characteristics
的返回值中包含Characteristics.IDENTITY_FINISH
,则必须保证从A到R能够强制转换成功,此时该方法固定返回Function.identity()
即可。否则会报错。比如A为ArrayList,而R为HashMap的情况就会报错。
实战
添加依赖
<dependency>
<groupId>com.alibaba</groupId>
<artifactId>fastjson</artifactId>
<version>1.2.33</version>
</dependency>
需求1
有一个学生列表,求计算学生的总分
数据结构
public class Student {
private String id;
private Course course;
private double score;
public Student() {
}
public Student(String id, Course course, double score) {
this.id = id;
this.course = course;
this.score = score;
}
@Override
public String toString() {
return JSON.toJSONString(this, true);
}
public enum Course {
LANGUAGE, MATHEMATICS, ENGLISH, TOTAL
}
// 省略setter、getter方法
}
列表定义
public static final List<Student> students =
List.of(new Student("1", Student.Course.ENGLISH, 78), new Student("1", Student.Course.LANGUAGE, 71),
new Student("1", Student.Course.MATHEMATICS, 82), new Student("2", Student.Course.ENGLISH, 69),
new Student("2", Student.Course.LANGUAGE, 66), new Student("2", Student.Course.MATHEMATICS, 46),
new Student("3", Student.Course.ENGLISH, 78), new Student("3", Student.Course.LANGUAGE, 88),
new Student("3", Student.Course.MATHEMATICS, 100), new Student("4", Student.Course.ENGLISH, 68),
new Student("4", Student.Course.LANGUAGE, 84), new Student("4", Student.Course.MATHEMATICS, 90),
new Student("5", Student.Course.ENGLISH, 74), new Student("5", Student.Course.LANGUAGE, 59),
new Student("5", Student.Course.MATHEMATICS, 87));
自定义收集器
public class StudentCollector implements Collector<Student, List<Student>, List<Student>> {
@Override
public Supplier<List<Student>> supplier() {
return ArrayList::new;
}
@Override
public BiConsumer<Set<Student>, Student> accumulator() {
return (set, student) -> {
Predicate<Student> studentPredicate = s -> s.getId().equals(student.getId());
boolean b = set.stream().noneMatch(studentPredicate);
if (b) {
student.setCourse(Student.Course.TOTAL);
set.add(student);
}
set.stream().filter(studentPredicate).forEach(s -> s.setScore(s.getScore() + student.getScore()));
};
}
@Override
public BinaryOperator<List<Student>> combiner() {
return null;
}
@Override
public Function<List<Student>, List<Student>> finisher() {
return Function.identity();
}
@Override
public Set<Characteristics> characteristics() {
return EnumSet.of(Characteristics.IDENTITY_FINISH);
}
}
调用
List<Student> studentList = students.stream().collect(new StudentCollector());
System.out.println(studentList);
结果
需求2
将输出结果转化成TreeMap<Double, Student>
,key为学生总分,value为学生本身,并按照总分从高到低排序。
数据结构与学生列表不变。
自定义收集器
public class StudentDiffCollectors implements Collector<Student, Set<Student>, TreeMap<Double, Student>> {
@Override
public Supplier<Set<Student>> supplier() {
return HashSet::new;
}
@Override
public BiConsumer<Set<Student>, Student> accumulator() {
return (set, student) -> {
Predicate<Student> studentPredicate = s -> s.getId().equals(student.getId());
boolean b = set.stream().noneMatch(studentPredicate);
if (b) {
student.setCourse(Student.Course.TOTAL);
set.add(student);
}
set.stream().filter(studentPredicate).forEach(s -> s.setScore(s.getScore() + student.getScore()));
};
}
@Override
public BinaryOperator<Set<Student>> combiner() {
return null;
}
@Override
public Function<Set<Student>, TreeMap<Double, Student>> finisher() {
return list -> {
TreeMap<Double, Student> doubleStudentTreeMap = new TreeMap<>(Comparator.reverseOrder());
list.forEach(student -> doubleStudentTreeMap.put(student.getScore(), student));
return doubleStudentTreeMap;
};
}
@Override
public Set<Characteristics> characteristics() {
return EnumSet.of(Characteristics.CONCURRENT, Characteristics.UNORDERED);
}
}
调用
TreeMap<Double, Student> collect = students.stream().collect(new StudentDiffCollectors());
System.out.println(collect);
结果
进阶
除了实现接口以外,jdk还帮我们提供了更简单的实现,只需要调用Collector.of()
方法即可,如下所示:
Collector<Student, List<Student>, List<Student>> studentCollector = Collector.of(ArrayList::new, (l, r) -> {
// 计算逻辑
}, (l, r) -> null, Function.identity(), Collector.Characteristics.IDENTITY_FINISH);
由于它符合A强转为R的条件,因此,可以更简化为:
Collector<Student, List<Student>, List<Student>> studentCollector = Collector.of(ArrayList::new, (l, r) -> {
// 计算逻辑
}, (l, r) -> null);
它的效果和实现接口Collector<Student, List<Student>, List<Student>>
是一样的,但看起来更简洁明了!