Java 8 中的多层分组使用 Nested GroupBy

本文探讨了在处理嵌套类时如何实现多层分组Java 8。具体来说，目标是按 key1 字段对项目进行分组，然后对于每组项目，进一步按 key2 字段对它们进行分组。最终，输出应该是一个以 key1 作为外键的映射，以及一个 key2 到子项列表的映射。

最初的方法尝试使用 Collectors.groupingBy 方法来实现这一点，但是，它是无法直接通过多个键对单个项目进行分组。为了克服这个问题，使用了 flatMap 操作。

一种方法涉及使用 Map.entrySet 创建一个临时对，以在收集之前保存项目和子项目的组合。 Java 9 中提供的另一种方法利用了 flatMapping 收集器，它可以直接在收集器中执行 flatMap 操作。

这是 flatMap 解决方案：

Map>> result = pojo.getItems().stream()
    .flatMap(item -> item.getSubItems().stream()
        .map(sub -> new AbstractMap.SimpleImmutableEntry(item.getKey1(), sub)))
    .collect(Collectors.groupingBy(AbstractMap.SimpleImmutableEntry::getKey,
                Collectors.mapping(Map.Entry::getValue,
                    Collectors.groupingBy(SubItem::getKey2))));

在 Java 8 中使用自定义收集器的替代方案：

static  Collector flatMapping(
    Function super T,? extends Stream extends U>> mapper,
    Collector super U,A,R> downstream) {

    BiConsumer acc = downstream.accumulator();
    return Collector.of(downstream.supplier(),
        (a, t) -> { try(Stream extends U> s=mapper.apply(t)) {
            if(s!=null) s.forEachOrdered(u -> acc.accept(a, u));
        }},
        downstream.combiner(), downstream.finisher(),
        downstream.characteristics().toArray(new Collector.Characteristics[0]));
}

这个自定义收集器可以按如下方式使用：

Map>> result = pojo.getItems().stream()
    .collect(Collectors.groupingBy(Item::getKey1,
                Collectors.flatMapping(item -> item.getSubItems().stream(),
                    Collectors.groupingBy(SubItem::getKey2))));