compute源码分析及案例
描述
校验这个计算函数是否为空,为空抛出空指针异常。使用给定的key和获取到的oldValue计算出一个新的值newValue。在新值为空的前提下,如果旧的值不为空,或者这个集合中包含这个key,则删除这个key,并且返回null。在新值为空的前提下,如果旧的值为空,并且集合不包含这个key,什么都不做,直接返回null。如果新值不为空,添加这个新值和指定的key到集合,这里可能是新增操作【新值不为空,集合中不存在这个key】或者是替换操作【新值不为空,集合中这个key,对应的val为null】,返回这个新值。
参数
-
key:与指定值关联的键 -
remappingFunction:计算值的函数
返回值
在新值为空的前提下,如果旧的值不为空,或者这个集合中包含这个key,则删除这个key,并且返回null。在新值为空的前提下,如果旧的值为空,并且集合不包含这个key,什么都不做,直接返回null。如果新值不为空,添加这个新值和指定的key到集合,这里可能是新增操作【新值不为空,集合中不存在这个key】或者是替换操作【新值不为空,集合中这个key,对应的val为null】,返回这个新值。
源码分析
default V compute(K key,
BiFunction<? super K, ? super V, ? extends V> remappingFunction) {
// 校验这个计算函数是否为空,为空抛出空指针异常
Objects.requireNonNull(remappingFunction);
// 使用给定的key获取值,赋值给oldValue
V oldValue = get(key); // 使用给定的key和获取到的oldValue计算出一个新的值newValue V newValue = remappingFunction.apply(key, oldValue); // 判断这个新的值是否为空 if (newValue == null) { // 如果在新值为空的前提下,旧的值不为空,或者这个集合中包含这个key if (oldValue != null || containsKey(key)) { // 删除这个key,并且返回null remove(key); return null; } else { // 如果在新值为空的前提下,旧的值为空,并且集合不包含这个key,什么都不做,直接返回null return null; } } else { // 新值不为空,添加这个新值和指定的key到集合 // 这里可能是新增操作【新值不为空,集合中不存在这个key】或者是替换操作【新值不为空,集合中这个key,对应的val为null】 put(key, newValue); // 返回这个新值 return newValue; } } 复制代码
案例
@Test
public void test_compute_01(){
Map<String, String> map = new HashMap<>();
String key = "aaa";
String msg = "hahh";
// key对应的val不存在,会将key和msg添加到集合中,并返回这个msg,这个执行的是新增操作 String compute = map.compute(key, (k, v) -> (v == null) ? msg : v.concat(msg)); System.out.println(compute); map.forEach((k, v) -> System.out.println("=======key: " + k + "========v: " + v)); } @Test public void test_compute_01_val_null(){ Map<String, String> map = new HashMap<>(); String key = "aaa"; String msg = null; // key对应的val不存在,会将key和msg添加到集合中,并返回这个msg String compute = map.compute(key, (k, v) -> (v == null) ? msg : v.concat(msg)); System.out.println(compute); map.forEach((k, v) -> System.out.println("=======key: " + k + "========v: " + v)); } @Test public void test_compute_02(){ Map<String, String> map = new HashMap<>(); String key = "aaa"; String msg = "hahh"; map.put(key, "uuu"); // key对应的val存在,并且不为空,使用concat函数在原有的val后面追加新值,并返回追加后的新值,然后添加到集合中 String compute = map.compute(key, (k, v) -> (v == null) ? msg : v.concat(msg)); System.out.println(compute); map.forEach((k, v) -> System.out.println("=======key: " + k + "========v: " + v)); } @Test public void test_compute_03(){ //java8 实现计算功能统计字符串出现次数 Map<String, Integer> map = new HashMap<>(); List<String> source = Arrays.asList("hello", "world", "hello", "welcome", "hello", "hello", "welcome", "world"); source.forEach(s -> map.compute(s, (k, v) -> Objects.isNull(v) ? 1 : v + 1)); System.out.println(map); } 复制代码
putIfAbsent源码分析及案例
描述
根据key获取旧的val,如果旧的值不为空,返回旧的val,如果旧的val为空【这个集合要支持val为空】,返回新添加的val,这个新添加的val可能为null。
参数
-
Key:与指定值关联的键 -
value:与指定键关联的值
返回值
如果旧的值不为空,返回旧的val,如果旧的val为空【这个集合要支持val为空】,返回新添加的val,这个新添加的val可能为null。
源码分析
default V putIfAbsent(K key, V value) {
// 根据key获取旧的val
V v = get(key);
// 判断旧的val是否为空
if (v == null) {
// 旧的val为空,将这个key和value添加到集合中 v = put(key, value); } // 如果旧的值不为空,返回旧的val,如果旧的val为空,返回新添加的val,这个新添加的val可能为null return v; } 复制代码
案例
@Test
public void test_putAbsent_01(){
// HashMap是允许键和值为null的
Map<String, String> map = new HashMap<>();
// 验证key和val键值对在集合中都不存在
map.putIfAbsent("aaa", "hhh"); map.forEach((k, v) -> System.out.println("===key: " + k + "===value: " + v)); } @Test public void test_putAbsent_02(){ Map<String, String> map = new HashMap<>(); // HashMap是允许键和值为null的 map.put("aaa", null); // 验证key和val都存在集合中,但是key对应的val为null map.putIfAbsent("aaa", "bbb"); map.forEach((k, v) -> System.out.println("===key: " + k + "===value: " + v)); } @Test public void test_putAbsent_03(){ Map<String, String> map = new HashMap<>(); map.put("aaa", "bbb"); // 验证key和val都存在集合中,并且key对应的val不为null map.putIfAbsent("aaa", "ccc"); map.forEach((k, v) -> System.out.println("===key: " + k + "===value: " + v)); } 复制代码