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

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

    @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);
        });
    }

    @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);
        });
    }

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());
  }

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();
  }
}

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