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方法。

例子如下:

java 复制代码
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,效果都是一样的。

java 复制代码
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实现类都会有这个方法,且效果都一样):

java 复制代码
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;
}
相关推荐
Wx-bishekaifayuan几秒前
django电商易购系统-计算机设计毕业源码61059
java·spring boot·spring·spring cloud·django·sqlite·guava
customer085 分钟前
【开源免费】基于SpringBoot+Vue.JS周边产品销售网站(JAVA毕业设计)
java·vue.js·spring boot·后端·spring cloud·java-ee·开源
全栈开发圈7 分钟前
新书速览|Java网络爬虫精解与实践
java·开发语言·爬虫
WaaTong9 分钟前
《重学Java设计模式》之 单例模式
java·单例模式·设计模式
面试鸭11 分钟前
离谱!买个人信息买到网安公司头上???
java·开发语言·职场和发展
沈询-阿里1 小时前
java-智能识别车牌号_基于spring ai和开源国产大模型_qwen vl
java·开发语言
AaVictory.1 小时前
Android 开发 Java中 list实现 按照时间格式 yyyy-MM-dd HH:mm 顺序
android·java·list
LuckyLay1 小时前
Spring学习笔记_27——@EnableLoadTimeWeaving
java·spring boot·spring
向阳12182 小时前
Dubbo负载均衡
java·运维·负载均衡·dubbo