중첩 GroupBy를 사용하여 Java 8에서 다단계 그룹화

이 문서에서는 중첩 클래스를 처리할 때 다단계 그룹화를 구현하는 방법을 살펴봅니다. Java 8. 특히 목표는 key1 필드로 항목을 그룹화한 다음 각 항목 그룹에 대해 key2 필드로 항목을 추가로 그룹화하는 것입니다. 궁극적으로 출력은 key1을 외부 키로 사용하고 key2를 하위 항목 목록으로 매핑하는 맵이어야 합니다.

초기 접근 방식에서는 이를 달성하기 위해 Collectors.groupingBy 메서드를 사용하려고 시도하지만, 여러 키로 단일 항목을 직접 그룹화할 수 없습니다. 이를 극복하기 위해 flatMap 작업이 활용됩니다.

한 가지 방법은 수집하기 전에 항목과 하위 항목의 조합을 보관하기 위해 Map.entrySet을 사용하여 임시 쌍을 만드는 것입니다. Java 9에서 사용할 수 있는 다른 접근 방식은 flatMap 작업을 수집기에서 직접 수행할 수 있는 flatMapping 수집기를 활용합니다.

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