Map的computeIfAbsent使用场景和方法
|
1 2 3 |
default V computeIfAbsent(K key, Function<? super K, ? extends V> mappingFunction) { ... } |
我们在复杂map操作（put操作）时候有的时候不知道此时当前key对应的value值是否存在，这里，我们如果使用常规的代码编写，代码量比较大
例如我们定义一个场景：存在一个数组，我们需要将当前数组中相同的数存储到一个对应List集合中
常规实现
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
@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方法实现
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
@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测试数据puteIfAbsent(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中。
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
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() 方法后,使用后以上代码变成了下面的形式（便捷、高效、代码更加优美，但可阅读性降低）：
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
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测试数据puteIfAbsent( "china" , key -> getValues(key)).add( "liSi" ); System.out.println(hashMap.toString()); }
public static HashSet getValues(String key) { return new HashSet(); } } |
hashMap测试数据puteIfAbsent([china], key -> getValues(key)).add([liSi]);的意思表示key为[China]的建值对是否存在，返回的是value的值。
如果存在则获取china的值，并操作值的set添加数据[lisi"。
如果不存在，则调用方法，新创建set结构，将"lisi"添加到set中，再存入到hashMap中。
二、computeIfAbsent() 的源码实现
这个方法是JDK8中Map类新增的一个方法，用来实现当一个KEY的值缺失的时候，使用给定的映射函数重新计算填充KEY的值并返回结果。computeIfAbsent 方法的JDK源码如下：
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
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; } |
以上为个人经验，希望能给大家一个参考，也希望大家多多支持。
原文链接：https://blog.csdn.net/SmallPig_Code/article/details/119647864
查看更多关于基于Map的computeIfAbsent的使用场景和使用方式的详细内容...