jdk 8カスタムCollector実現対象の合計平均値
1、まず二つのインターフェースSumOperator、Average Operatorを定義する。
すべての需要と機能を実現するクラスは、SumOperatorインターフェースを実現しなければならない。平均値を求める機能を実現するクラスは、Average Operatorインターフェースを実現しなければならない。以下は2つのインターフェース定義である。
すべての需要と機能を実現するクラスは、SumOperatorインターフェースを実現しなければならない。平均値を求める機能を実現するクラスは、Average Operatorインターフェースを実現しなければならない。以下は2つのインターフェース定義である。
public interface SumOperator<T> {
/**
* sum
* @desc:
* @author:
* @createTime: 2020/4/29 12:37
* @param t
* @return: T
*/
T sum(T t);
}
public interface AverageOperator<T> extends SumOperator<T>{
/**
* divide
* @desc:
* @author:
* @createTime: 2020/4/29 12:37
* @param number
* @return: T
*/
T divide(long number);
}
2、次は一つのクラスをカスタマイズして、この二つのCollectorを生成します。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;
/**
* CustomStream
*
* @desc: TODO 、
* @version:
* @createTime: 2020/4/29 12:23
* @author:
*/
public class CustomCollectors{
public static<T extends SumOperator<T>> Collector<T, List<T>, T> ofSum() {
return new CollectorImpl<T>(
list -> list.stream().reduce((t1, t2) -> t1.sum(t2)).get()
);
}
public static<T extends SumOperator<T>> Collector<T, List<T>, T> ofSum(Function<Optional<T>, T> function) {
return new CollectorImpl<T>(
list -> list.stream().reduce(function.apply(list.stream().findFirst()), (t1, t2) -> t1.sum(t2))
);
}
public static<T extends AverageOperator<T>> Collector<T, List<T>, T> ofAverage() {
return new CollectorImpl<T>(
list -> list.stream().reduce((t1, t2) -> t1.sum(t2)).get().divide(list.size())
);
}
public static<T extends AverageOperator<T>> Collector<T, List<T>, T> ofAverage(Function<Optional<T>, T> function) {
return new CollectorImpl<T>(
list -> list.stream().reduce(function.apply(list.stream().findFirst()), (t1, t2) -> t1.sum(t2)).divide(list.size())
);
}
static class CollectorImpl<T> implements Collector<T, List<T>, T>{
private Function<List<T>, T> finisher;
private CollectorImpl(Function<List<T>, T> finisher){
this.finisher = finisher;
}
@Override
public Supplier<List<T>> supplier() {
return ArrayList::new;
}
@Override
public BiConsumer<List<T>, T> accumulator() {
return List::add;
}
@Override
public BinaryOperator<List<T>> combiner() {
return (left, right) -> {
left.addAll(right); return left;
};
}
@Override
public Function<List<T>, T> finisher() {
return finisher;
}
@Override
public Set<Characteristics> characteristics() {
return Collections.emptySet();
}
}
}
3、テストimport lombok.Getter;
import lombok.Setter;
import lombok.experimental.Accessors;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.stream.Collector;
import java.util.stream.Collectors;
public class Test {
@Getter
@Setter
@Accessors(chain = true)
static class Student implements AverageOperator<Student> {
private String name;
private int term;
private double mathScore;
private double englishScore;
@Override
public Student divide(long number) {
this.mathScore /= number;
this.englishScore /= number;
return this;
}
@Override
public Student sum(Student student) {
this.mathScore += student.mathScore;
this.englishScore += student.englishScore;
return this;
}
}
public static void main(String[] args) {
List<Student> students = new ArrayList<Student>(){
{
add(new Student().setName(" ").setTerm(1).setMathScore(90).setEnglishScore(80));
add(new Student().setName(" ").setTerm(2).setMathScore(85).setEnglishScore(77));
add(new Student().setName(" ").setTerm(3).setMathScore(95).setEnglishScore(78));
add(new Student().setName(" ").setTerm(4).setMathScore(88).setEnglishScore(75));
add(new Student().setName(" ").setTerm(5).setMathScore(95).setEnglishScore(78));
add(new Student().setName(" ").setTerm(6).setMathScore(88).setEnglishScore(75));
add(new Student().setName(" ").setTerm(1).setMathScore(90).setEnglishScore(60));
add(new Student().setName(" ").setTerm(2).setMathScore(95).setEnglishScore(67));
add(new Student().setName(" ").setTerm(3).setMathScore(98).setEnglishScore(60));
add(new Student().setName(" ").setTerm(4).setMathScore(98).setEnglishScore(70));
add(new Student().setName(" ").setTerm(5).setMathScore(100).setEnglishScore(68));
add(new Student().setName(" ").setTerm(6).setMathScore(96).setEnglishScore(75));
}
};
//
Collector<Student, List<Student>, Student> collector = CustomCollectors.ofAverage(optional -> {
Student source = optional.get();
Student target = new Student();
target.setName(source.getName());
return target;
});
Map<String, Student> map = students.stream().collect(Collectors.groupingBy(Student::getName, collector));
map.forEach((name, student) -> {
System.out.println(String.format(" 【%s】 ;( :%f, :%f)", name, student.getMathScore(), student.getEnglishScore()));
});
}
}
印刷結果は以下の通りです。 【 】 ;( :96.166667, :66.666667)
【 】 ;( :90.166667, :77.166667)