Java 8 加入地图 Collectors.toMap

Question

我正在尝试在地图中收集对象列表过程的结果，并且它 returns 是一张地图。我想我应该用 Collectors.toMap 来做，但我还没找到方法。

这是代码：

public class Car {
    List<VersionCar> versions;
    public List<VersionCar> getVersions() {
        return versions;
      }
    }

public class VersionCar {
  private String wheelsKey;
  private String engineKey;
  public String getWheelsKey() {
     return wheelsKey;
  }
  public String getEngineKey() {
    return engineKey;
  }
}

处理方法：

private static Map<String,Set<String>> processObjects(VersionCar version) {
    Map<String,Set<String>> mapItems = new HashMap<>();
    mapItems.put("engine", new HashSet<>(Arrays.asList(version.getEngineKey())));
    mapItems.put("wheels", new HashSet<>(Arrays.asList(version.getWheelsKey())));
    return mapItems;
}

我的最终代码是：

Map<String,Set<String>> mapAllItems =
                car.getVersions().stream()
                   .map(versionCar -> processObjects(versionCar))
                   .collect(Collectors.toMap()); // here I don't know like collect the map.

我的想法是处理版本列表，最后得到一个包含两个项目的地图：车轮和引擎，但有一个 set<> 包含所有版本的所有不同项目。您有什么想法吗？我可以使用 Collectors.toMap 或其他选项来做到这一点吗？

Answer 1

您可以使用 Collectors.toMap(keyMapper, valueMapper, mergeFunction)。最后一个参数用于解决与同一键关联的值之间的冲突。

例如：

    Map<String, Set<String>> mapAllItems =
            car.getVersions().stream()
                    .map(versionCar -> processObjects(versionCar))
                    .flatMap(m -> m.entrySet().stream())
                    .collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue,
                            (firstSet, secondSet) -> {
                                    Set<String> result = new HashSet<>();
                                    result.addAll(firstSet);
                                    result.addAll(secondSet);
                                    return result;
                            }
                    ));

Answer 2

在这种情况下您要使用的运算符可能是 "reduce"

car.getVersions().stream()
        .map(versionCar -> processObjects(versionCar))
        .reduce((map1, map2) -> {
                    map2.forEach((key, subset) -> map1.get(key).addAll(subset));
                    return map1;
                })
        .orElse(new HashMap<>());

"reduce" 中使用的 lambda 是一个 BinaryOperator，它合并了 2 个映射和 return 合并后的映射。

"orElse" 只是为了 return 在您的初始集合（版本）为空的情况下。从类型的角度来看，它摆脱了 "Optional"

Answer 3

为了得到mapAllItems，我们不需要也不应该定义processObjects方法：

Map<String, Set<String>> mapAllItems = new HashMap<>();
mapAllItems.put("engine", car.getVersions().stream().map(v -> v.getEngineKey()).collect(Collectors.toSet()));
mapAllItems.put("wheels", car.getVersions().stream().map(v -> v.getWheelsKey()).collect(Collectors.toSet()));

或通过 AbstractMap.SimpleEntry which is lighter than the Map created byprocessObjects`:

mapAllItems = car.getVersions().stream()
    .flatMap(v -> Stream.of(new SimpleEntry<>("engine", v.getEngineKey()), new SimpleEntry<>("wheels", v.getWheelsKey())))
    .collect(Collectors.groupingBy(e -> e.getKey(), Collectors.mapping(e -> e.getValue(), Collectors.toSet())));

Java 8 加入地图 Collectors.toMap

Java 8 Join Map with Collectors.toMap

java

java-8

java-stream

collectors