目录
一、computeIfAbsent()
二、removeIfAbsent()
一、computeIfAbsent()
编程中经常遇到这种数据结构,判断一个map中是否存在这个key,如果存在则处理value的数据,
如果不存在,则创建一个满足value要求的数据结构放到value中。
以前常用的方法如下:
编程中经常遇到这种数据结构,判断一个map中是否存在这个key,如果存在则处理value的数据,
如果不存在,则创建一个满足value要求的数据结构放到value中。
以前常用的方法如下:
public class TestComputeIfAbsent {
static HashMap<String, Set<String>> hashMap = new HashMap<>();
public static void main(String[] args) {
Set<String> set = new HashSet<>();
set.add("zhangSan");
hashMap.put("china", set);
// 判断map中是否存在,如果存在则添加元素到set中,如果不存在则新建set添加到hashMap中
if(hashMap.containsKey("china")) {
hashMap.get("china").add("liSi");
} else {
Set<String> setTmp = new HashSet<>();
setTmp.add("liSi");
hashMap.put("china", setTmp);
}
System.out.println(hashMap.toString());
}
}
官方非常的贴心,为了满足广大用户的要求,加入了computeIfAbsent() 这个api,
使用后以上代码变成了下面的形式:
public class TestComputeIfAbsent {
static HashMap<String, Set<String>> hashMap = new HashMap<>();
public static void main(String[] args) {
Set<String> set = new HashSet<>();
set.add("zhangSan");
hashMap.put("china", set);
// after JDK1.8
hashMap.computeIfAbsent("china", key -> getValues(key)).add("liSi");
System.out.println(hashMap.toString());
}
public static HashSet getValues(String key) {
return new HashSet();
}
}
hashMap.computeIfAbsent("china", key -> getValues(key)).add("liSi");的意思表示key为“China”的建值对
是否存在,返回的是value的值。
如果存在则获取china的值,并操作值的set添加数据“lisi"。
如果不存在,则调用方法,新创建set结构,将"lisi"添加到set中,再存入到hashMap中。