java 8 stream:在新对象中分组并存储sum,并合并地图

时间:2017-03-23 05:35:03

标签: java lambda java-8 java-stream

我有一个类Row,如:

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

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

和班级统计:

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

1 个答案:

答案 0 :(得分: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&lt;&gt;或嵌套地图。