如何使用java 8 stream和lambda来flatMap一个groupingBy结果

时间:2015-08-27 14:49:25

标签: java lambda java-8 java-stream collectors

我有一个包含其他对象列表的对象,我想返回由容器的某些属性映射的包含对象的平面图。任何一个是否可以只使用流和lambdas?

public class Selling{
   String clientName;
   double total;
   List<Product> products;
}

public class Product{
   String name;
   String value;
}

让我们提供一系列操作:

List<Selling> operations = new ArrayList<>();

operations.stream()
     .filter(s -> s.getTotal > 10)
     .collect(groupingBy(Selling::getClientName, mapping(Selling::getProducts, toList());

结果将是善意的

Map<String, List<List<Product>>> 

但是我想像

那样扁平化
Map<String, List<Product>>

2 个答案:

答案 0 :(得分:8)

在JDK9中,有一个名为flatMapping的新标准收集器可以通过以下方式实现:

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) -> {
                try (Stream<? extends U> result = mapper.apply(t)) {
                    if (result != null)
                        result.sequential().forEach(u -> downstreamAccumulator.accept(r, u));
                }
            },
            downstream.combiner(), downstream.finisher(),
            downstream.characteristics().toArray(new Collector.Characteristics[0]));
}

您可以将其添加到项目中并使用如下:

operations.stream()
   .filter(s -> s.getTotal() > 10)
   .collect(groupingBy(Selling::getClientName, 
              flatMapping(s -> s.getProducts().stream(), toList())));

答案 1 :(得分:7)

您可以尝试以下方式:

Map<String, List<Product>> res = operations.parallelStream().filter(s -> s.getTotal() > 10)
    .collect(groupingBy(Selling::getClientName, mapping(Selling::getProducts,
        Collector.of(ArrayList::new, List::addAll, (x, y) -> {
            x.addAll(y);
            return x;
        }))));