在 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 。

Answer 2

这可以通过使用三元运算符来简化。 你真的不需要 if-else 语句

List<Object> list = dataMap.containsKey(key) ?  dataMap.get(key) : new ArrayList<>();
list.add(someNewObject);
dataMap.put(key, list);

Answer 3

您也可以使用compute方法。

dataMap.compute(key, (k, v) -> {
                              if(v == null)
                                  return new ArrayList<>();
                              else {
                                   v.add(someNewObject);
                                   return v;
                              }
                         });

Answer 4

您可以使用

dataMap.compute(key,(k,v)->v!=null?v:new ArrayList<>()).add(someNewObject)

或者

dataMap.merge(key,new ArrayList<>(),(v1,v2)->v1!=null?v1:v2).add(someNewObject)