Java中的Map.compute

1.功能简介

　　　　简单的说就是，给出一个key值和一个函数，然后这个函数根据key对应的键值对[key,value]计算出一个新的value，就叫newValue吧

　　　　　　如果这个newValue的值是null，则从原来的map中移除key，compute返回null,

　　　　　　如果这个newValue的值不为null，则更新key对应的值为newValue，compute返回newValue。

2.应用例子

// 统计一个字符串中每个字符出现的次数   
 public static void main(String[] args) {
        Map<Character, Integer> map = new HashMap<>();
        String hello = "Hello World!";

        for (int i = 0; i < hello.length(); i++) {
            char key = hello.charAt(i);
            map.compute(key, (k, v) -> {
                if (Objects.isNull(v)) {
                    v = 1;
                } else {
                    v += 1;
                }
                return v;
            });
        }
        System.out.println(map.toString());
    }

输出

{ =1, !=1, r=1, d=1, e=1, W=1, H=1, l=3, o=2}

 

3.JDK中的源码：

default V compute(K key,BiFunction<? super K, ? super V, ? extends V> remappingFunction) {
        Objects.requireNonNull(remappingFunction);
        V oldValue = get(key);

        V newValue = remappingFunction.apply(key, oldValue);
        if (newValue == null) {
            // delete mapping
            if (oldValue != null || containsKey(key)) {
                // something to remove
                remove(key);
                return null;
            } else {
                // nothing to do. Leave things as they were.
                return null;
            }
        } else {
            // add or replace old mapping
            put(key, newValue);
            return newValue;
        }
    }