Java 8 嵌套（多级）分组依据

collections lambda java java-8 java-stream

2022-09-01 23:29:34

我很少有像下面这样的课程

class Pojo {
    List<Item> items;
}

class Item {
    T key1;
    List<SubItem> subItems;
}

class SubItem {
    V key2;
    Object otherAttribute1;
}

我想基于和每个聚合聚合来聚合项目，子项应按以下方式聚合：key1key2

Map<T, Map<V, List<Subitem>>

Java 8 嵌套如何实现这一点？Collectors.groupingBy

我正在尝试一些东西，并卡在一半

pojo.getItems()
    .stream()
    .collect(
        Collectors.groupingBy(Item::getKey1, /* How to group by here SubItem::getKey2*/)
    );

注意：这与级联不同，级联基于此处讨论的同一对象中的字段进行多级聚合groupingBy

答案 1

您无法按多个键对单个项目进行分组，除非您接受该项目可能出现在多个组中。在这种情况下，您希望执行一种操作。flatMap

实现这一目标的一种方法是在收集之前与临时配对一起使用并保持组合。由于缺少标准对类型，因此典型的解决方案是用于：Stream.flatMapItemSubItemMap.Entry

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

不需要这些临时对象的替代方法是直接在收集器中执行操作，但不幸的是，直到Java 9才会出现flatMapping。flatMap

这样，解决方案看起来像这样

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

如果我们不想等待Java 9，我们可以在我们的代码库中添加一个类似的收集器，因为它并不难实现：

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

答案 2