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

2022-09-02 03:37:39

我有一张地图:

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函数式风格做到这一点?


答案 1

您可以使用 .computeIfAbsent

如果映射不存在,只需通过将键与新的空列表相关联来创建一个映射,然后将值添加到其中。

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

如文档所述,它返回与指定键关联的当前(现有或计算)值,以便您可以将调用与 链接在一起。当然,这假设原始地图中的值不是固定大小的列表(我不知道你是如何填充的)......ArrayList#add

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


答案 2

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

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