使用groupingBy的类属性的平面映射收集器

Question

我上这堂课

class Assignment {
  private Integer index;
  private List<Quantity> quantities;
}

然后，我有该类对象的列表。

List<Assigment> assignments = new ArrayList<>();

有没有一种方法创建一个Map包含from Assignment和List<Quantity>as值的索引？

到目前为止，这是我尝试过的。

assignments.stream().collect(groupingBy(Assignment::getIndex));

但这给我一个Map<Integer, List<Assignment>>，我想要一个Map<Integer, List<Quantity>>。

我已经尝试过使用forEach方法-它可以工作-但我确定必须有一种方法可以在一个衬里中进行操作-
或者至少仅使用collect和groupingBy方法

Answer 1

似乎没有平面映射收集器可groupingBy在Java8中用作下游，但已被Java9提出并接受：https
://bugs.openjdk.java.net/browse/JDK-8071600

public 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> downstreamAccumulator = downstream.accumulator();
        return Collector.of(downstream.supplier(),
                            (r, t) -> mapper.apply(t).sequential().forEach(u -> downstreamAccumulator.accept(r, u)),
                            downstream.combiner(),
                            downstream.finisher(),
                            downstream.characteristics().stream().toArray(Collector.Characteristics[]::new));
    }

如果使用该代码，并且还添加了一个返回的quantities方法，则可以使用以下代码：Assignment``Stream<Quantity>

Map<Integer, List<Quantity>> result = assignments.stream()
    .collect(groupingBy(Assignment::getIndex, 
        flatMapping(Assignment::quantities, toList())));