Kotlin:列表中BigDecimal的总和

Question

我有一个我想要过滤的列表,然后返回一个带有金额总和的id地图:

val totalById = list
                    .filter { it.status == StatusEnum.Active }
                    .groupBy { it.item.id }
                    .mapValues { it.value.sumBy { it.amount } }

"it.amount"是BigDecimal,但看起来sumBy只是Int.

对于Java 8,它将是:

Collectors.groupingBy(i-> i.getItem().getId(), Collectors.mapping(Item::getAmount, Collectors.reducing(BigDecimal.ZERO, BigDecimal::add))))

在Kotlin有办法吗？

Answer 1

就像你Collectors.reducing在java中使用的一样,你可以在Kotlin中使用fold或reduce扩展函数:

val bigDecimals: List<BigDecimal> = ...
val sum = bigDecimals.fold(BigDecimal.ZERO) { acc, e -> acc + e }
// or
val sum2 = bigDecimals.fold(BigDecimal.ZERO, BigDecimal::add)

Answer 2

在 Kotlin 1.4 中sumOf，他们介绍了它的用法：

val bigDecimals: List<BigDecimal> = ...
val sum = bigDecimals.sumOf { it }

文档

Answer 3

您可以创建类似于自己的sumByBigDecimal 扩展功能sumByDouble.例如:

/**
 * Returns the sum of all values produced by [selector] function applied to each element in
 * the collection.
 */
inline fun <T> Iterable<T>.sumByBigDecimal(selector: (T) -> BigDecimal): BigDecimal {
    var sum: BigDecimal = BigDecimal.ZERO
    for (element in this) {
        sum += selector(element)
    }
    return sum
}

用法示例:

val totalById = list
        .filter { it.status == StatusEnum.Active }
        .groupBy { it.item.id }
        .mapValues { it.value.sumByBigDecimal { it.amount } }