[[424111]]

今天棧長分享一個實用的 Java 8 開發技能，那就是 Map 接口中增加的 compute 方法，給 Map 集合計算更新用的。

compute簡介

如下所示，Java 8 在 Map 和 ConcurrentMap 接口中都增加了 3 個 compute 方法，說明也是支持多線程并發安全操作的。

這三個方法的區別：

•  compute：計算并更新值
•  computeIfAbsent：Value不存在時才計算
•  computeIfPresent：Value存在時才計算

compute有啥用?

話說這有什么卵用？

先看看沒用 Java 8 的一個小示例： 

/**  
* 公眾號：Java技術棧  
*/  
private static void preJava8() {  
    List<String> animals = Arrays.asList("dog", "cat", "cat", "dog", "fish", "dog");  
    Map<String, Integer> map = new HashMap<>();  
    for(String animal : animals){  
        Integer count = map.get(animal);  
        map.put(animal, count == null ? 1 : ++count);  
    }  
    System.out.println(map);  
} 

輸出：

{cat=2, fish=1, dog=3}

這是一個統計一個列表中每個動物的數量，代碼再怎么精簡都需要一步 get 操作，判斷集合中是否有元素再確定是初始化：1，還是需要 +1。

很多時候，這個 get 操作顯然是毫無必要的，所以 Java 8 提供了 3 個 compute 方法，來看看怎么用吧！

Java 8 compute 實現方式： 

/**  
* 公眾號：Java技術棧  
*/  
private static void inJava8() {  
    List<String> animals = Arrays.asList("dog", "cat", "cat", "dog", "fish", "dog");  
    Map<String, Integer> map = new HashMap<>();  
    for(String animal : animals){  
        map.compute(animal, (k, v) -> v == null ? 1 : ++v);  
    }  
    System.out.println(map);  
} 

使用 compute 方法一行搞定，省去了需要使用 get 取值再判斷的冗余操作，直接就可以獲取元素值并計算更新，是不是很方便呢？

compute源碼分析

這還是一個默認方法，為什么是默認方法，也是為了不改動其所有實現類，關于默認方法的定義可以關注公眾號Java技術棧獲取 Java 8+ 系列教程。 

/**  
* 公眾號：Java技術棧  
*/  
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)) { // 舊值存在時  
            // 移除該鍵值  
            remove(key);  
            return null;  
        } else {  
            // nothing to do. Leave things as they were.  
            return null;  
        }  
    } else { // 新值不為空  
        // 添加或者覆蓋舊值  
        put(key, newValue);  
        return newValue;  
    }  
} 

實現邏輯其實也很簡單，其實就是結合了 Java 8 的函數式編程讓代碼變得更簡單了，Java 也越來越聰明了。

另外兩個方法我就不演示了，在特定的場合肯定也肯定特別有用，大家知道就好，需要的時候要知道拿來用。

本節教程所有實戰源碼已上傳到這個倉庫：

https://github.com/javastacks/javastack

本次的分享就到這里了，希望對大家有用。覺得不錯，在看、轉發分享一下哦~