java8 stream 多字段分组、多个列汇总统计

原创已于 2022-05-02 23:40:03 修改 · 4.7k 阅读

12 ·

CC 4.0 BY-SA版权

文章标签：

#java #开发语言

于 2022-05-02 22:51:28 首次发布

java8新特性专栏收录该内容

2 篇文章

订阅专栏

本文详细介绍了Java中Collector的collectingAndThen方法，该方法用于在收集操作完成后执行额外的转换。通过示例展示了如何使用collectingAndThen进行分组聚合，并对聚合结果进行计算，例如计算面积和电量的乘积。这些示例展示了如何在处理数据流时灵活地组合收集器和转换函数，以实现复杂的数据处理需求。

Java中的类Collector的collectionAndThen(Collector下游，函数完成器)方法，该方法采用Collector，以便我们可以执行其他整理转换。

public static <T, A, R, RR> 
       Collector <T, A, RR> 
       collectingAndThen(Collector <T, A, R> downstream, 
                         Function <R, RR> finisher)
                                                             

Where,

用法

T：输入元素的类型
A：下游Collector的中间堆积类型
R：下游Collector的结果类型
RR：结果Collector的结果类型

参数：此方法接受下面列出的两个参数

downstream:它是Collector的一个实例，即我们可以在这里使用任何Collector。
finisher:它是一个函数的实例，该函数将应用于下游Collector的最终结果。

返回值：返回一个执行下游Collector动作的Collector，然后在finisher函数的帮助下执行附加的整理步骤。

分组聚合返回多个计算结果

  Map<String,  Map<String, Object>>  groupAndSum = result.stream().collect(Collectors.groupingBy(
            m ->m.get("type").toString(),Collectors.collectingAndThen(
               Collectors.toList(), m -> {
                 Map<String, Object> map = new HashMap();
                 final double   kwh=m.stream().mapToDouble(t->(Double)t.get("daykwh")).sum();
                 final BigDecimal   area=m.stream().map(a->Convert.toBigDecimal(a.get("area"))).reduce(BigDecimal.ZERO,BigDecimal::add);
                 map.put("kwh",kwh);
                 map.put("area",area);
                 return map;
        })));

分组统计并用统计的结果在做运算

  Map<String, BigDecimal> sum = result.stream().collect(Collectors.groupingBy(
            m -> m.get("type").toString(), Collectors.collectingAndThen(
               Collectors.toList(), m -> {
                  /*Convert.toBigDecimal 类型转换工具*/
                 final BigDecimal kwh = m.stream().map(a -> Convert.toBigDecimal(a.get("daykwh"))).reduce(BigDecimal.ZERO, BigDecimal::add);
                 final BigDecimal area = m.stream().map(a -> Convert.toBigDecimal(a.get("area"))).reduce(BigDecimal.ZERO, BigDecimal::add);
                 return kwh.multiply(area).setScale(2, BigDecimal.ROUND_HALF_UP);
     })));