Map中的computeIfAbsent()方法

调用java集合Map.computeIfAbsent()方法

java版本JDK1.8中，Map是我们经常使用的，在面对复杂Map时，我们怎么更好的去维护呢？

比如，这里的复杂Map，即<K,V>中V是一个Collection集合：

我们先拿到Key值对应的Collection，再往Collection中加入元素。但是这会出现一个问题！！！

如果get(key)的结果为null，那么代码就变成这样了：

就会出现异常。

所以，为了不抛出异常，你的代码需要能够处理key不在map里的情况。比如，你可能会这样处理：

也就是当key不在map中时，提前new 一个 Collection放进去。

虽然这样的做法可以解决抛异常的情况，但是代码却变复杂了，那么我们要怎么简化这段代码呢？？

答：我们的 map.computeIfAbsent();方法就派上用场了。

做法如下：

解释一下：

computeIfAbsent()方法的效果是：如果key不在Map中，那么会new一个ArrayList作为键值对的值，把这个不存在的key作为键值对的键，然后放到map里面去，并且返回键值对的值。如果key存在，那么就返回这个键对应的值，相当于是get方法。

例子如下：

package com.yicheng.test.md5Test;

import java.util.*;

/**
 * @Author yimeng
 * @Date 2024/3/22 22:13
 * @PackageName:com.yicheng.test.md5Test
 * @ClassName: Test
 * @Description: TODO
 * @Version 1.0
 */
public class Test {
    public static void main(String[] args) {
        Map<String, List<Integer>> dataMap = new HashMap<>();

        // 初始化键值对，值为 List 类型
        List<Integer> list1 = new ArrayList<>();
        list1.add(10);
        list1.add(20);

        dataMap.put("numbers1", list1);

        // 使用 computeIfAbsent 方法操作 List。当dataMap中的numbers1键存在，那么就相当于是dataMap.get("numbers1").add(30)
        dataMap.computeIfAbsent("numbers1", key -> new ArrayList<>())
                .add(30);

        dataMap.put("numbers2", null);
        // 使用 computeIfAbsent 方法操作 List。
        List<Integer> numbers2 = dataMap.computeIfAbsent("numbers2", key -> new LinkedList<>());
        numbers2.add(100);

        // 输出 Map 中的键值对
        System.out.println("Map 中的键值对：");
        for (Map.Entry<String, List<Integer>> entry : dataMap.entrySet()) {
            System.out.println(entry.getKey() + " -> " + entry.getValue());
        }
    }
}

get(键)的值不存在于Map、get(键)的值为null，效果都是一样的。

package com.yicheng.test.md5Test;

import java.util.*;

/**
 * @Author yimeng
 * @Date 2024/3/22 22:13
 * @PackageName:com.yicheng.test.md5Test
 * @ClassName: Test
 * @Description: TODO
 * @Version 1.0
 */
public class Test {
    public static void main(String[] args) {
        Map<String, List<Integer>> dataMap = new HashMap<>();

        // 初始化键值对，值为 List 类型
        List<Integer> list1 = new ArrayList<>();
        list1.add(10);
        list1.add(20);

        dataMap.put("numbers1", list1);

        // 使用 computeIfAbsent 方法操作 List
        dataMap.computeIfAbsent("numbers1", key -> new ArrayList<>())
                .add(30);

        dataMap.put("numbers2", null);
        // 使用 computeIfAbsent 方法操作 List
        List<Integer> numbers2 = dataMap.computeIfAbsent("numbers2", key -> new LinkedList<>());
        numbers2.add(100);
        
        // 使用 computeIfAbsent 方法操作 List
        List<Integer> numbers3 = dataMap.computeIfAbsent("numbers21234", key -> new LinkedList<>());
        numbers3.add(100);

        // 输出 Map 中的键值对
        System.out.println("Map 中的键值对：");
        for (Map.Entry<String, List<Integer>> entry : dataMap.entrySet()) {
            System.out.println(entry.getKey() + " -> " + entry.getValue());
        }
    }
}

computeIfAbsent方法是顶级接口Map中的方法。所以只要是实现了Map接口的实现类，都可以使用这个方法。HashMap也能用。

computeIfAbsent方法的声明如下(他是Map中的默认方法，所以所有的Map实现类都会有这个方法，且效果都一样)：

default V computeIfAbsent(K key,Function<? super K, ? extends V> mappingFunction) {
        Objects.requireNonNull(mappingFunction);
        V v;
        if ((v = get(key)) == null) {
            V newValue;
            if ((newValue = mappingFunction.apply(key)) != null) {
                put(key, newValue);
                return newValue;
            }
        }
        return v;
}