Map中computeIfAbsent() 的作用和底层实现

小半 • 2022年5月23日下午9:58 • Java • 阅读 181

一、computeIfAbsent() 的作用

最近在开发中，发现同事经常使用Map的computeIfAbsent（）方法进行编程，于是对他的实现和作用产生了小兴趣，下面用两个demo案例来简单介绍一下它的作用，然后再对底层实现进行进一步阅读。

作用：判断一个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());
  }

在使用了Map的computeIfAbsent() 方法后,使用后以上代码变成了下面的形式（便捷、高效、代码更加优美，但可阅读性降低）：

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

二、computeIfAbsent() 的源码实现

这个方法是JDK8中Map类新增的一个方法，用来实现当一个KEY的值缺失的时候，使用给定的映射函数重新计算填充KEY的值并返回结果。computeIfAbsent 方法的JDK源码如下：

default V computeIfAbsent(K key, Function<? super K, ? extends V> mappingFunction) {
	Objects.requireNonNull(mappingFunction);
	V v;
	// 尝试获取KEY的值，如果获取不到KEY
	if ((v = get(key)) == null) {
	    V newValue;
	    // 利用传入的计算函数，得到新的值
	    if ((newValue = mappingFunction.apply(key)) != null) {
	        // 将KEY的值填充为函数计算的结果
	        put(key, newValue);
	        // 返回计算的结果
	        return newValue;
	    }
	}
	// 如果KEY的值存在，则直接返回
	return v;
}