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))));