合并地图和修改值

时间:2015-10-12 16:40:36

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

有两张地图,我正在尝试将它们合并为一张地图(finalResp)。

Map<String, String[]> map1 = new HashMap<>();
Map<String, String> map2 = new HashMap<>();

HashMap<String, String> finalResp = new HashMap<String, String>();

解决方案 - 在Java 8之前 - 实现如下:

for (Map.Entry<String, String[]> entry : map1.entrySet()) {
    if (map2.containsKey(entry.getKey())) {
        String newValue  = changetoAnother(map1.get(entry.getKey()), map2.get(entry.getKey()));
        finalResp.put(entry.getKey(), newValue);
    }
}

使用Java 8,我陷入了困境:

HashMap<String, String> map3 = new HashMap<>(map2);
map1.forEach((k, v) -> map3.merge(k, v, (i, j) -> mergeValue(i, j) ));

如何检查地图1中是否存在地图2密钥并修改值?

1 个答案:

答案 0 :(得分:4)

一种可能的方法是过滤不需要的元素(不包含在map2中)并将结果收集到新的地图中:

Map<String, String> finalResp = 
    map1.entrySet().stream().filter(e -> map2.containsKey(e.getKey()))
                            .collect(Collectors.toMap(
                                Entry::getKey, 
                                e -> changetoAnother(e.getValue(), map2.get(e.getKey()))
                            ));

另一种方法是创建map2的副本,保留此Map中包含map1个键的所有键,最后通过应用函数替换所有值changetoAnother

Map<String, String> result = new HashMap<>(map2);
result.keySet().retainAll(map1.keySet());
result.replaceAll((k, v) -> changetoAnother(map1.get(k), v));

请注意,第一个解决方案的优势在于它可以轻松推广到任何两个地图:

private <K, V, V1, V2> Map<K, V> merge(Map<K, V1> map1, Map<K, V2> map2, BiFunction<V1, V2, V> mergeFunction) {
    return map1.entrySet().stream()
                          .filter(e -> map2.containsKey(e.getKey()))
                          .collect(Collectors.toMap(
                              Entry::getKey, 
                              e -> mergeFunction.apply(e.getValue(), map2.get(e.getKey()))
                          ));
}

Map<String, String> finalResp = merge(map1, map2, (v1, v2) -> changetoAnother(v1, v2));