如何在 Java 8 中使用 Collectors.groupingBy 实现嵌套分组？

Java 8 中的嵌套（多级）Group By

经常需要根据多个键对对象进行多级分组。在本例中，我们有一组表示数据的类，涉及带有键的项目列表和也带有键的子项目列表。目标是按键对项目进行分组，对于每个聚合，进一步按键对子项目进行分组，从而生成嵌套映射。

使用 Java 8 的 Collectors.groupingBy，我们可以通过使用组合来实现此目的flatMap 和自定义平面映射收集器。我们首先使用 Stream.flatMap 创建一个临时对来保存项目和子项目的组合。

<code class="java">Map<T, Map<V, List<SubItem>>> result = pojo.getItems().stream()
        .flatMap(item -> item.subItems.stream()
                .map(sub -> new AbstractMap.SimpleImmutableEntry<>(item.getKey1(), sub)))
        .collect(Collectors.groupingBy(AbstractMap.SimpleImmutableEntry::getKey,
                Collectors.mapping(Map.Entry::getValue,
                        Collectors.groupingBy(SubItem::getKey2))));</code>

或者，我们可以通过使用平面映射收集器来避免创建临时对象，不幸的是，这不会Java 9 之前可用。通过这种方法，代码可以简化为：

<code class="java">Map<T, Map<V, List<SubItem>>> result = pojo.getItems().stream()
        .collect(Collectors.groupingBy(Item::getKey1,
                Collectors.flatMapping(item -> item.getSubItems().stream(),
                        Collectors.groupingBy(SubItem::getKey2))));</code>

对于可能不想等待 Java 9 的用户，可以实现自定义平面映射收集器：

<code class="java">static <T,U,A,R> Collector<T, ?, R> flatMapping(
        Function<? super T, ? extends Stream<? extends U>> mapper,
        Collector<? super U, A, R> downstream) {

        BiConsumer<A, ? super U> 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]));
}</code>

以上是如何在 Java 8 中使用 Collectors.groupingBy 实现嵌套分组？的详细内容。更多信息请关注PHP中文网其他相关文章！