基于Map的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;

}

以上为个人经验,希望能给大家一个参考,也希望大家多多支持。

以上是 基于Map的computeIfAbsent的使用场景和使用方式 的全部内容, 来源链接: www.h5w3.com/248554.html

回到顶部