【Java用法】computeIfAbsent() 的使用

一、背景描述

编程中经常遇到这种数据结构，判断一个map中是否存在这个key，如果存在则处理value的数据，如果不存在，则创建一个满足value要求的数据结构放到value中。在 JDK1.7 之前，经常这么干，这样做起来功能是可以实现的，但是代码不够优雅。

二、功能发现

于是呢，在经历了时间的检验，终于千呼万唤始出来，在 Java8 版本中，加入了 computeIfAbsent() 这个api，computeIfAbsent() 是 Java8 中 Map 接口新增的一个方法，用于在 Map 中根据指定的键获取对应的值，如果该键不存在，则使用指定的函数计算出一个默认值并将其存储到 Map 中，最后返回该默认值。

语法如下：

V Map.computeIfAbsent(K key, Function<? super K, ? extends V> mappingFunction)
1

其中，key 表示要获取值的键，mappingFunction 表示计算默认值的函数。具体来说，如果 Map 中已经存在指定的键，则直接返回该键对应的值；否则，使用 mappingFunction 计算出一个默认值，并将该键和默认值存储到 Map 中，最后返回该默认值。

三、示例代码

示例代码如下：

import java.util.HashMap;
import java.util.Map;

public class TestComputeIfAbsent {

    public static void main(String[] args) {
        Map<String, Integer> map = new HashMap<>();
        map.put("apple", 1);
        map.put("banana", 2);

        int orangeCount = map.computeIfAbsent("orange", k -> 0);
        System.out.println(orangeCount); // 输出 0

        int appleCount = map.computeIfAbsent("apple", k -> 0);
        System.out.println(appleCount); // 输出 1
    }

}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18

在上面的示例中，首先创建了一个 Map 对象，并向其中添加了两个键值对。然后，使用 computeIfAbsent() 方法获取了一个不存在的键 “orange” 对应的值，由于该键不存在，因此使用指定的函数 k -> 0 计算出了一个默认值 0，并将该键和默认值存储到 Map 中，最后返回了该默认值。接着，使用 computeIfAbsent() 方法获取了一个已经存在的键 “apple” 对应的值，由于该键已经存在，因此直接返回了该键对应的值 1。

本文结束！