在Java8函数样式中，如何将值映射到已存在的键值对

Question

我有一张地图：

Map<String, List<Object>> dataMap;

现在，我想向地图添加新的键值对，如下所示：

if(dataMap.contains(key)) {
    List<Object> list = dataMap.get(key);
    list.add(someNewObject);
    dataMap.put(key, list);
} else {
    List<Object> list = new ArrayList();
    list.add(someNewObject)
    dataMap.put(key, list);
}

我该如何使用Java8功能样式来做到这一点？

Answer 1

您可以使用computeIfAbsent。

如果不存在映射，则只需将密钥与新的空列表相关联即可创建一个，然后将值添加到其中。

dataMap.computeIfAbsent(key, k -> new ArrayList<>()).add(someNewObject);

如文档所述，它将返回与指定键关联的当前（现有或计算得出的）值，因此您可以将调用与链接起来ArrayList#add。当然，这是假设原始地图中的值不是固定大小的列表（我不知道您如何填充它）…

顺便说一句，如果您有权访问原始数据源，则可以从中获取流并Collectors.groupingBy直接使用。