我有一个地图:
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函数式风格来实现这一点?
您可以使用computeIfAbsent
。
如果映射不存在,只需通过将键与新的空列表关联来创建一个映射,然后将值添加到其中。
dataMap.computeIfAbsent(key, k -> new ArrayList<>()).add(someNewObject);
正如文档所述,它返回与指定键关联的当前(现有或计算的)值,以便您可以将调用与ArrayList#add
链接起来。当然,这是假设原始地图中的值不是固定大小的列表(我不知道你是如何填写的)。。。
顺便说一句,如果您可以访问原始数据源,我会从中获取流并直接使用Collectors.groupingBy
。
这可以通过使用三元运算符来简化。你并不真的需要if-else语句
List<Object> list = dataMap.containsKey(key) ? dataMap.get(key) : new ArrayList<>();
list.add(someNewObject);
dataMap.put(key, list);
您也可以使用compute
方法。
dataMap.compute(key, (k, v) -> {
if(v == null)
return new ArrayList<>();
else {
v.add(someNewObject);
return v;
}
});
您可以使用
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)