收集器按多个字段嵌套分组

时间:2019-04-10 08:56:01

标签: java spring java-stream collectors

您将如何使用收集器以便在第二级按多个字段分组。例如:

"someList": {
        "firstLevelElementX": {
              "secondLevelElementW": 2,
              "secondLevelElementZ": 3,
              "secondLevelElementK": 7
        },
        "firstLevelElementY": {
              "secondLevelElementW": 1,
              "secondLevelElementZ": 3,
              "secondLevelElementK": 10
        }
}

我尝试创建一个包含“ secondLevel”元素的类,并按此类分组,但无法使其起作用:

@Data
@AllArgsConstructor
public class someClass{
    private String firstLevelElement;
    private Long secondLevelElementW;
    private Long secondLevelElementZ;
    private Long secondLevelElementK= 0L;

}

这就是我的方法:

Map<String,Map<String,Long>> someList =
            events.stream().collect(
                Collectors.groupingBy(
                    someDAO::getFirstLevelElement,
                    Collectors.groupingBy(
                        someClass::getSecondLevelFields,
                        Collectors.counting())
                )
            );

有一些解决方案建议使用n级分组(使用链),但我宁愿使它更简单,更简洁。

编辑

为了阐明自己,我将尝试提供一个更好的示例,这是我的清单:

{
      "date": "2019-04-08 08:28:01.0",
      "source": "maint",
      "severity": "HARMLESS",
      "site": "USA",
      "hostname": "usaHost"
    },
    {
      "date": "2019-04-08 08:28:01.0",
      "source": "CPU_Checker",
      "severity": "MINOR",
      "site": "GERMANY",
      "hostname": "germanyHost"
    },
    {
      "date": "2019-04-02 08:28:01.0",
      "source": "maint",
      "severity": "HARMLESS",
      "site": "USA",
      "hostname": "anotherUsaHost"
    }

我想在第二级上对'source'和'severity'使用group-by,因此输出应如下所示:

"eventList": {
        "USA": {
              "maint": 2,
              "HARMLESS": 2
        },
        "GERMANY": {
              "CPU_checker": 1,
              "MINOR": 1
        }
}

2 个答案:

答案 0 :(得分:1)

您可以汇总第二个groupingBy收集器中的第二级字段值:

Map<String, Map<Long, List<someClass>>> result = elements.stream().collect(
        groupingBy(someClass::getFirstLevelElement,
            groupingBy(s -> 
                s.getSecondLevelElementK() + s.getSecondLevelElementW() + s.getSecondLevelElementZ()))
);

答案 1 :(得分:1)

如果您可以使用 Java 9 或更高版本,则可以使用Collectors.flatMapping()来实现:

Map<String, Map<String, Long>> eventList = list.stream()
        .collect(Collectors.groupingBy(MyObject::getSite, Collectors.flatMapping(
                o -> Stream.of(o.getSource(), o.getSeverity()),
                Collectors.groupingBy(Function.identity(), Collectors.counting())
        )));

结果将是这样:

{
    USA={maint=2, HARMLESS=2}, 
    GERMANY={CPU_Checker=1, MINOR=1}
}

如果您不能使用 Java 9 ,则可以自己实现flatMapping()函数。您可以查看Java 9 Collectors.flatMapping rewritten in Java 8,它可以为您提供帮助。