基于Map的computeIfAbsent的使用场景和使用方式

目录

• Map的computeIfAbsent使用场景和方法

• 常规实现

• 使用computeIfAbsent方法实现

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

• 一、computeIfAbsent() 的作用

• 二、computeIfAbsent() 的源码实现

Map的computeIfAbsent使用场景和方法

default V computeIfAbsent(K key, Function<? super K, ? extends V> mappingFunction) {
   ...
}

我们在复杂map操作（put操作）时候有的时候不知道此时当前key对应的value值是否存在，这里，我们如果使用常规的代码编写，代码量比较大

例如我们定义一个场景：存在一个数组，我们需要将当前数组中相同的数存储到一个对应List集合中

常规实现

   @Test
   public void test02() {
       //复杂map的使用场景：首先我们说复杂map，即map的value值为一个list集合或者是一个Set集合，对象或者是其他的集合
       //给定一个场景：现在存在一个数组，我们需要将当前数组中相同的数存储到一个对应List集合中
       int[] nums = {1,2,3,1,3,4,6,7,9,9,1,3,4,5};
       Map<Integer, List<Integer>> map = new HashMap<>();
       //普通的写法
       for (int i = 0; i < nums.length; i++) {
           if (!map.containsKey(nums[i])) {
               ArrayList<Integer> list = new ArrayList<>();
               list.add(nums[i]);
               map.put(nums[i],list);
           }else {
               map.get(nums[i]).add(nums[i]);
           }
       }
       map.forEach((key,value) -> {
           System.out.print(key + ": ");
           System.out.println(value);
       });
   }

使用computeIfAbsent方法实现

   @Test
   public void  test03() {
       int[] nums = {1,2,3,1,3,4,6,7,9,9,1,3,4,5};
       Map<Integer, List<Integer>> map = new HashMap<>();
       //我们使用map的computeIfAbsent解决
       for (int i = 0; i < nums.length; i++) {
           //返回值是该key对应的集合list
           map.computeIfAbsent(nums[i], key -> new ArrayList<Integer>());
           map.get(nums[i]).add(nums[i]);
       }

map.forEach((key,value) -> {
           System.out.print(key + ": ");
           System.out.println(value);
       });
   }

输出结果：

1: [1, 1, 1]
2: [2]
3: [3, 3, 3]
4: [4, 4]
5: [5]
6: [6]
7: [7]
9: [9, 9]

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

一、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;
}

来源：https://blog.csdn.net/SmallPig_Code/article/details/119647864