java 相加的函数_Java8 Collectors求和功能的自定义扩展操作

业务中需要将一组数据分类后收集总和，原本可以使用Collectors.summingInt()，但是我们的数据源是BigDecimal类型的，而Java8原生只提供了summingInt、summingLong、summingDouble三种基础类型的方法。

于是就自己动手丰衣足食吧。。

自定义工具类

public class MyCollectors {

private MyCollectors() {

}

// public static Collector summingBigDecimal(Function super T, BigDecimal> mapper) {}

// BigDecimal 类型的集合求和

public static Collector summingBigDecimal(ToBigDecimalFunction super T> mapper) {

return new CollectorImpl<>(

() -> new BigDecimal[] { BigDecimal.ZERO },

(a, t) -> a[0] = a[0].add(mapper.applyAsInt(t)),

(a, b) -> {

a[0] = a[0].add(b[0]);

return a;

},

a -> a[0],

Collections.emptySet()

);

}

static class CollectorImpl implements Collector {

// 创建一个计算用的容器

private final Supplier supplier;

// 计算逻辑

private final BiConsumer accumulator;

// 合并逻辑

private final BinaryOperator combiner;

// 返回最终计算值

private final Function finisher;

// 空Set

private final Set characteristics;

CollectorImpl(Supplier supplier, BiConsumer accumulator, BinaryOperator combiner,

Function finisher, Set characteristics) {

this.supplier = supplier;

this.accumulator = accumulator;

this.combiner = co