Question

假设我有3辆汽车，每辆汽车有3个发动机，每个发动机有3个零件。如何使用Java8流创建map of <engine_model, List<parts>>？

在常规java中，我会执行以下操作：

Map<String, List<Parts>> map = new HashMap<String, List<Parts>>();
for (Car car: cars){
  for (Engine engine: car.getEngines()){
    if (!map.contains(engine.getModel())){
      map.put(engine.getModel(), new ArrayList<Part>());
    }
    for (Part  part: engine.getParts()){
      map.get(engine.getModel()).add(part);
    }
  }
}

此处map是map，它将引擎的模型（type String）映射到与该模型相关的所有部分的列表。不同的车可以有相同的发动机型号。

我尝试执行以下操作：

carList.stream().flatMap(p -> p.getEngines().stream())
  .collect(Collectors.groupingBy(p -> p.getModel(),
               Collectors.mapping(Engine::getParts, Collectors.toList())))

问题是上面的使用我得到了 Map<String, List<List<Part>>> 而且我要 Map<String, List<Part>>

我想一个同等的问题是：如何在上述示例的上下文中将List of Lists更改为List？

Answer 1

首先，您甚至可以简化您的命令式变体：

Map<String, List<Part>> map = new HashMap<>();
for(Car car: cars) {
    for(Engine engine: car.getEngines()) {
        map.computeIfAbsent(engine.getModel(), key -> new ArrayList<>())
           .addAll(engine.getParts());
    }
}

在决定功能变体是否真的是一种改进时，这很重要。 flatMap + groupingBy收集器是正确的方法，但是您有一个（Part s）列表的事实是另一个障碍。在Java 9中，您可以编写

Map<String, List<Part>> map = cars.stream()
    .flatMap(car -> car.getEngines().stream())
    .collect(Collectors.groupingBy(Engine::getModel,
                 Collectors.flatMapping(e -> e.getParts().stream(), Collectors.toList())));

但在那之前，我们必须手动组装一个等效的收集器：

Map<String, List<Part>> map = cars.stream()
    .flatMap(car -> car.getEngines().stream())
    .collect(Collectors.groupingBy(Engine::getModel,
                 Collector.of(ArrayList::new, (l,e) -> l.addAll(e.getParts()),
                              (l1,l2) -> { l1.addAll(l2); return l1; })));

所以我现在考虑继续使用循环。

如何使用Java8流从嵌套对象组装Map <key，value =“”>？

1 个答案: