業(yè)務(wù)中需要將一組數(shù)據(jù)分類后收集總和，原本可以使用Collectors.summingInt()，但是我們的數(shù)據(jù)源是BigDecimal類型的，而Java8原生只提供了summingInt、summingLong、summingDouble三種基礎(chǔ)類型的方法。

于是就自己動(dòng)手豐衣足食吧。。

自定義工具類

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 {

// 創(chuàng)建一個(gè)計(jì)算用的容器

private final Supplier supplier;

// 計(jì)算邏輯

private final BiConsumer accumulator;

// 合并邏輯

private final BinaryOperator combiner;

// 返回最終計(jì)算值

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 = combiner;

this.finisher = finisher;

this.characteristics = characteristics;

}

CollectorImpl(Supplier supplier, BiConsumer accumulator, BinaryOperator combiner,

Set characteristics) {

this(supplier, accumulator, combiner, castingIdentity(), characteristics);

}

@Override

public BiConsumer accumulator() {

return accumulator;

}

@Override

public Supplier supplier() {

return supplier;

}

@Override

public BinaryOperator combiner() {

return combiner;

}

@Override

public Function finisher() {

return finisher;

}

@Override

public Set characteristics() {

return characteristics;

}

}

@SuppressWarnings("unchecked")

private static Function castingIdentity() {

return i -> (R) i;

}

}

自定義函數(shù)式接口

@FunctionalInterface

public interface ToBigDecimalFunction {

BigDecimal applyAsInt(T value);

}

測試入口

public class AnswerApp {

public static void main(String[] args) {

List list = Lists.newArrayList();

for (int i = 0; i < 24; i++) {

list.add(BigDecimal.valueOf(i + 10.2121543));

}

// 方式1

BigDecimal sum = list.stream().collect(MyCollectors.summingBigDecimal(e -> e));

System.out.println(sum.doubleValue());

// 方式2

Optional reduce = list.stream().reduce(BigDecimal::add);

System.out.println(reduce.orElse(BigDecimal.valueOf(0)));

}

}

// OUTPUT: 521.0917032

補(bǔ)充：Collectors擴(kuò)展接口 實(shí)現(xiàn)BigDecimal的相加

第一步

創(chuàng)建ToBigDecimalFunction接口

import java.math.BigDecimal;

@FunctionalInterface

public interface ToBigDecimalFunction {

BigDecimal applyAsBigDecimal(T value);

}

第二步

創(chuàng)建工具類 實(shí)現(xiàn)接口

import java.math.BigDecimal;

import java.util.Collections;

import java.util.Set;

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;

public class CollectorsUtil {

static final Set CH_NOID = Collections.emptySet();

private CollectorsUtil() {

}

@SuppressWarnings("unchecked")

private static Function castingIdentity() {

return i -> (R) i;

}

/**

* Simple implementation class for {@code Collector}.

*

* @param

* the type of elements to be collected

* @param

* the type of the result

*/

static class CollectorImpl implements Collector {

private final Supplier supplier;

private final BiConsumer accumulator;

private final BinaryOperator combiner;

private final Function finisher;

private final Set characteristics;

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(Supplier supplier, BiConsumer accumulator, BinaryOperator combiner,

Set characteristics) {

this(supplier, accumulator, combiner, castingIdentity(), characteristics);

}

@Override

public BiConsumer accumulator() {

return accumulator;

}

@Override

public Supplier supplier() {

return supplier;

}

@Override

public BinaryOperator combiner() {

return combiner;

}

@Override

public Function finisher() {

return finisher;

}

@Override

public Set characteristics() {

return characteristics;

}

}

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

return new CollectorImpl<>(() -> new BigDecimal[1], (a, t) -> {

if (a[0] == null) {

a[0] = BigDecimal.ZERO;

}

a[0] = a[0].add(mapper.applyAsBigDecimal(t));

}, (a, b) -> {

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

return a;

}, a -> a[0], CH_NOID);

}

}

使用測試

import com.example.javademo.JavaDemoApplicationTests;

import com.example.javademo.pojo.Student;

import com.example.javademo.utils.DataUtils;

import org.junit.Test;

import java.math.BigDecimal;

import java.util.stream.Collectors;

public class TestBigDecimal extends JavaDemoApplicationTests {

@Test

public void testGroupByAfterBigdecimal(){

/*

自定義實(shí)現(xiàn)對(duì)分組后的集合，屬性為bigdecmal進(jìn)行相加

*/

System.out.println(DataUtils.getData().stream().collect(Collectors.groupingBy(Student::getSchool,CollectorsUtil.summingBigDecimal(Student::getMoney))));

//歸約造作

BigDecimal reduce = DataUtils.getData().stream().map(Student::getMoney).reduce(BigDecimal.ZERO, BigDecimal::add);

System.out.println(reduce);

int sum = DataUtils.getData().stream().mapToInt(Student::getAge).sum();

System.out.println(sum);

}

}

以上為個(gè)人經(jīng)驗(yàn)，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教。

總結(jié)

以上是生活随笔為你收集整理的java 相加的函数_Java8 Collectors求和功能的自定义扩展操作的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。