Java8中Collectors求和功能的自定義擴(kuò)展

imtianx 發(fā)布于2019-08-15 13:26 / 3255人閱讀

摘要：起因業(yè)務(wù)中需要將一組數(shù)據(jù)分類(lèi)后收集總和，原本可以使用，但是我們的數(shù)據(jù)源是類(lèi)型的，而原生只提供了三種基礎(chǔ)類(lèi)型的方法。于是就自己動(dòng)手豐衣足食吧。。期望目標(biāo)實(shí)踐依葫蘆先分析一下方法該方法接受類(lèi)型的參數(shù)，返回類(lèi)型的實(shí)例化對(duì)象。

起因

業(yè)務(wù)中需要將一組數(shù)據(jù)分類(lèi)后收集總和，原本可以使用Collectors.summingInt()，但是我們的數(shù)據(jù)源是BigDecimal類(lèi)型的，而Java8原生只提供了summingInt、summingLong、summingDouble三種基礎(chǔ)類(lèi)型的方法。于是就自己動(dòng)手豐衣足食吧。。

期望目標(biāo)：

Map result = Arrays.stream(records).parallel().collect(Collectors.groupingBy(
    Record::getType, CollectorsUtil.summingBigDecimal(Record::getAmount)));

實(shí)踐

1. 依葫蘆
先分析一下Collectors.summingInt()方法

public static  Collector summingInt(ToIntFunction mapper) {
    return new CollectorImpl<>(
        () -> new int[1],
        (a, t) -> { a[0] += mapper.applyAsInt(t); },
        (a, b) -> { a[0] += b[0]; return a; },
        a -> a[0], CH_NOID);
    }

該方法接受ToIntFunction類(lèi)型的參數(shù)，返回CollectorImpl類(lèi)型的實(shí)例化對(duì)象。CollectorImpl是Collector接口的唯一實(shí)現(xiàn)類(lèi)

CollectorImpl(Supplier supplier,
                BiConsumer accumulator,
                BinaryOperator combiner,
                Function finisher,
                Set characteristics) {
        this.supplier = supplier;
        this.accumulator = accumulator;
        this.combiner = combiner;
        this.finisher = finisher;
        this.characteristics = characteristics;
    }

分析CollectorImpl的構(gòu)造器參數(shù)，可知summingInt()方法是這樣的

arg[0]創(chuàng)建一個(gè)計(jì)算用的容器： () -> new int[1]
arg[1]為計(jì)算邏輯： (a, t) -> { a[0] += mapper.applyAsInt(t); }
arg[2]為合并邏輯： (a, b) -> { a[0] += b[0]; return a; }
arg[3]為返回最終計(jì)算值： a -> a[0]
arg[4]為空Set(不知道干什么用。。)： Collections.emptySet()

2. 畫(huà)瓢
很自然的，BigDecimal的就是這樣了

public static  Collector summingBigDecimal(Function mapper) {
    return new CollectorImpl<>(() -> new BigDecimal[] { BigDecimal.ZERO }, (a, t) -> {
        a[0] = a[0].add(mapper.apply(t));
    }, (a, b) -> {
        a[0] = a[0].add(b[0]);
        return a;
    }, a -> a[0], CH_NOID);
}

Java8并行流的一些tips

千萬(wàn)注意共享變量的使用

注意裝箱拆箱的開(kāi)銷(xiāo)

基于數(shù)據(jù)量考慮使用并行流本身的成本

謹(jǐn)慎在并行流中使用事務(wù)