Regroupement multi-niveaux en Java 8 à l'aide de GroupBy imbriqué

Cet article explore comment implémenter un regroupement multi-niveaux lors du traitement de classes imbriquées dans Java 8. Plus précisément, l'objectif est de regrouper les éléments par un champ clé1, puis, pour chaque groupe d'éléments, de les regrouper davantage par un champ clé2. En fin de compte, le résultat devrait être une carte avec key1 comme clé externe et une carte de key2 vers une liste de sous-éléments.

L'approche initiale tente d'utiliser la méthode Collectors.groupingBy pour y parvenir, cependant, il est Il n'est pas possible de regrouper directement un seul élément par plusieurs touches. Pour surmonter ce problème, une opération flatMap est utilisée.

Une méthode consiste à créer une paire temporaire à l'aide de Map.entrySet pour contenir des combinaisons d'éléments et de sous-éléments avant de les collecter. L'autre approche, disponible dans Java 9, exploite le collecteur flatMapping, qui permet d'effectuer l'opération flatMap directement dans le collecteur.

Voici la solution 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))));

Une alternative utilisant un collecteur personnalisé dans 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]));
}

Ce collecteur personnalisé peut être utilisé comme suit :

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