java 8流：在新对象中分组和存储总和，并合并映射

Question

我有一个类 Row，例如：

class Row {
   public Long id1;
   public String id2;
   public Long metric1;
   public Long metric2;

   public Stats getStats() {
      return new Stats(metric1, metric2);
   }
}

还有一个类 Stats：

class Stats{
    public Long totalMetric1;
    public Long totalMetric2;

    public void addMetric1(Long metric1) {
       this.totalMetric1 = this.totalMetric1 + metric1;
    }

    public void addMetric2(Long metric2) {
       this.totalMetric2 = this.totalMetric2 + metric2;
    }
}

我有一个行列表

List<Row> rowList;

我需要将其转换为按id1和id2分组的地图，并且我需要将度量数据以这种形式汇总到Stats对象中

Map<Long, Map<String, Stats>>

我正在使用 java 流来生成它，但停留在这一点：

Map<Long, Map<String, List<Stats>>> map = stream.collect(
            Collectors.groupingBy(r->r.id1(), 
            Collectors.groupingBy(r->r.id2,
            Collectors.mapping(r->r.getStats(), Collectors.toList()))));

如何将列表转换为另一个对象，该对象包含该列表中所有对象的总和？

还有没有办法使用 java 流将上述所需形式的两个输出映射合并到第三个？

例子：-

输入：行列表

<1,"ABC", 1, 2>
<1,"ABC", 2, 2>
<1,"XYZ", 1, 2>
<2,"ABC", 1, 2>
<2,"XYZ", 1, 2>
<3,"XYZ", 1, 0>
<3,"XYZ", 2, 1>
<3,"XYZ", 2, 3>

结果：按字段 1、字段 2 分组的地图，字段 3 和字段 4 的总和

1 - ABC - 3,4
    XYZ - 1,2
2 - ABC - 1,2
    XYZ - 1,2
3 - XYZ - 5,4

Answer 1

我的建议是使用比嵌套集合更简单的方法。在 Row 类中，添加

public Pair<Long,String> getIds() {
   return new Pair<>(id1,id2);
}

在 Stats 类中，添加

public Stats merge(Stats other) {
    return new Stats(totalMetric1+other.totalMetric1, totalMetric2 + other.totalMetric2);
}

然后写类似

      Map<Pair<Long, String>, Stats> stats = rowList.stream().
              collect(Collectors.toMap(Row::getIds,Row::getStats, (s1,s2) -> s1.merge(s2)));

如果您对番石榴不过敏（而且您不应该如此，这是每个项目中包含的最简单的库之一，至少对我来说是这样），您可以用更优雅和可读的方式编写它

      Table<Long, String, Stats> table = rowList.stream().
            collect(Tables.toTable(Row::getId1, Row::getId2, Row::getStats,(s1,s2) -> s1.merge(s2),HashBasedTable::create));

无需使用 Pair 或嵌套地图。