使用Java lambda从两个映射构建映射

Question

我有一些代码需要帮助...我正在尝试使用两个地图作为源并同时使用java lambdas构建地图

Map<String, List<String>> motorsMapping = new HashMap<>();

motorsMapping.put("CAR", Collections.singletonList("AUDI"));
motorsMapping.put("CAR_AND_BIKE", Arrays.asList("AUDI", "BMW"));
motorsMapping.put("BIKE", Collections.singletonList("BMW"));

Map<String, List<String>> models = new HashMap<>();
models.put("AUDI", Arrays.asList("A1", "Q2"));
models.put("BMW", Arrays.asList("X1", "X5"));


motorsMapping.keySet().forEach(key -> {
     Map<String, List<String>> result = new HashMap<>();
     result.put(key, new ArrayList<>());
     motorsMapping.get(key).forEach(value -> models.getOrDefault(value, Collections.emptyList()).forEach(property -> result.get(key).add(property)));
});

我可以用你们在上面看到的foreach做到这一点，但我试图用lambdas做到这一点。。。像这样的

Map<String, List<String>> collected = motorsMapping
    .entrySet()
    .stream()
    .map(entry -> {
        return entry.getValue()
            .stream()
            .map(key -> models.getOrDefault(key, Collections.emptyList()))
            .flatMap(List::stream)
            .collect(Collectors.groupingBy(entry.getKey(), Collectors.toList()));
    }
)...

我想要的输出是这样的

CAR -> A1, Q2
BIKE -> X1, X5
CAR_AND_BIKE -> A1, Q2, X1, X5

但我无法理解

Answer 1

这项任务可能可以通过一些简单的方法来完成，例如，replaceAll，以便：

Map<String, List<String>> result = motorsMapping;
result.replaceAll((vehicle, categories) -> categories.stream()
        .flatMap(category -> models.getOrDefault(category, Collections.emptyList()).stream())
        .collect(Collectors.toList()));

Answer 2

有很多方法可以实现这一点，下面是一个使用Java9方法Map.entry（）和Collectors.flatMap（）：

Map<String, List<String>> collected = motorsMapping.entrySet().stream()
        .flatMap(e -> e.getValue().stream()
                .map(i -> Map.entry(e.getKey(), models.get(i).stream())))
        .collect(Collectors.groupingBy(Map.Entry::getKey, 
                Collectors.flatMapping(Map.Entry::getValue, Collectors.toList())));

如果您仅限于Java 8，则可以使用这种方法：

Map<String, List<String>> collected = motorsMapping.entrySet().stream()
        .flatMap(e -> e.getValue().stream().flatMap(i -> models.get(i).stream())
                .map(i -> new AbstractMap.SimpleEntry<>(e.getKey(), i)))
        .collect(Collectors.groupingBy(Map.Entry::getKey,
                Collectors.mapping(Map.Entry::getValue, Collectors.toList())));

最后一个使用AbstractMap. SimpleEntry（）代替Map.entry（）。

这两种解决方案都使用flatMap（）来获取每个键的模型列表。之后，再次使用flatMap（）获取每个模型的键。最后，两者都使用收集器。groupingBy（）获取结果：

{
  CAR_AND_BIKE: [A1, Q2, X1, X5],
  CAR: [A1, Q2], 
  BIKE: [X1, X5]
}

Answer 3

因为你已经在java-8下面标记了这个问题，所以我假设你有Java8，并且不能访问9Java中的平面地图。在java-8中，你可以使用：Collectors.to地图

Map< String, List< String > > collected = motorsMapping.entrySet( )
            .stream( )
            .collect( Collectors.toMap( Map.Entry::getKey
                    , motorMap -> motorMap.getValue( )
                                    .stream( )
                                    .flatMap( value -> models.get( value ).stream( ) )
                                    .collect( Collectors.toList( ) ) ) );
collected.entrySet( ).stream( ).forEach( System.out::println );

这是输出：

CAR_AND_BIKE=[A1, Q2, X1, X5]
CAR=[A1, Q2]
BIKE=[X1, X5]