Map中的computeIfAbsent()和computeIfPresent()---java 8新特性

今天寫的一個方法，如下，實現的功能就是將數據庫中的表字段獲取，存在map 中，將前端傳過來的數據存在map 中。

/**
     * 給表中添加數據
     * @param object   數據
     */
    protected void fillData(String type,Object object){
        JSONArray jsonArr = changeObjetToJsonArray(object);
        if(jsonArr==null || jsonArr.isEmpty()){
            return;
        }
        //遍歷jsonarray
        GetDateByTableName getDateByTableName=new GetDateByTableName(testNumber,imsi);
        for(int i=0;i<jsonArr.size();i++){
            List<String[]> rowList=new ArrayList<>();
            JSONObject jsonItem=jsonArr.optJSONObject(i);
            if(jsonItem!=null){
                String typeName=BaseDataPreHandle.typeNamePreHandle(type, jsonItem);
                TypeForTableName typeForTableName=Enum.valueOf(TypeForTableName.class,typeName.toUpperCase());
                String tableName=typeForTableName.getTableName();
                /**mapTableColumns.computeIfAbsent(tableName,k -> getFieldByTableName(tableName));*/
                if(mapTableColumns.get(tableName)==null){
                    mapTableColumns.put(tableName,getFieldByTableName(tableName));
                }
                String[] row= getDateByTableName.getDatasbyName(tableName,jsonItem,mapTableColumns.get(tableName));
                if(mapTableValues.get(tableName)!=null){
                    rowList=mapTableValues.get(tableName);
                }
                rowList.add(row);
                mapTableValues.put(tableName,rowList);
            }
        }

    }

然後我寫完用sonarLint 檢測發現兩個異味，竟然提示我不要這麼寫，而是提示使用Map.computeIfAbsent()方法，當時很納悶

使用了lambda表達式，然後我就看了一下computeIfAbsent()源碼，如下

default V computeIfAbsent(K key,
            Function<? super K, ? extends V> mappingFunction) {
        Objects.requireNonNull(mappingFunction);
        V v;
        if ((v = get(key)) == null) {
            V newValue;
            if ((newValue = mappingFunction.apply(key)) != null) {
                put(key, newValue);
                return newValue;
            }
        }

        return v;
    }

我看這個方法的意思是如果get(key)爲null，那麼就創建一個value值，這個value值就是lambda表達式的返回值，並且這個返回值不爲null，纔會將這個這個值存到map對應的key 中。
再來看一下我最開始寫的代碼的需求

if(mapTableColumns.get(tableName)==null){
	mapTableColumns.put(tableName,getFieldByTableName(tableName));
 }

如果map中這個key 對應的值不存在，就將新值賦給這個key。和上面computeIfAbsent() 的方法是一致的，因爲我可以保證新值不爲null。
簡單的說就是，map中key值存在就不更新，不存在就更新。
所以修改爲如下：

mapTableColumns.computeIfAbsent(tableName,k -> getFieldByTableName(tableName));

再接着看我後面的代碼，單獨拿出來

String[] row= getDateByTableName.getDatasbyName(tableName,jsonItem,mapTableColumns.get(tableName));
     if(mapTableValues.get(tableName)!=null){
          rowList=mapTableValues.get(tableName);
      }
      rowList.add(row);
      mapTableValues.put(tableName,rowList);

這個的意思是，有一個list集合，如果map.get(key)不爲空，就將這個值賦給list集合，如果這個值爲空就跳過，之後，將list添加一條數據，然後將新的list集合添加到這個map對應的key 值中。
簡單的說就是更新map 對應key 的值。不管map.get(key)是否爲空。

那這樣可以使用computeIfAbsent()方法嗎，顯然是不可以的，因爲源代碼很清楚，只有當get(key)爲空時纔會進行更新，不爲空的話直接就返回了。

所以computeIfAbsent() 方法的用法總結:

 只有在當前 Map 中 key 對應的值不存在或爲 null 時
 才調用 mappingFunction
 並在 mappingFunction 執行結果非 null 時
 將結果跟 key 關聯．
 mappingFunction 爲空時 將拋出空指針異常

那麼問題來了，現在我的這段代碼snoarLint 檢測出異味，改怎麼修改呢?
這讓我想到了computeIfPresent()方法，看了一下源碼

default V computeIfPresent(K key,
            BiFunction<? super K, ? super V, ? extends V> remappingFunction) {
        Objects.requireNonNull(remappingFunction);
        V oldValue;
        if ((oldValue = get(key)) != null) {
            V newValue = remappingFunction.apply(key, oldValue);
            if (newValue != null) {
                put(key, newValue);
                return newValue;
            } else {
                remove(key);
                return null;
            }
        } else {
            return null;
        }
    }

看源碼的意思是：當get(key)值存在時，並且新值不爲空，就會更新。否則就返回null。

這與我的需求不相符啊還是，我的需求是，不管key值是否爲空，都需要進行更新，我這個更新還是在原來的值上增加數據。

那現在怎樣消除異味呢？

實在沒辦法我使用了computeIfAbsent()和computeIfPresent()兩種方法結合。
先用computeIfAbsent()方法，保證一定可以獲取到get(key)
再用computeIfPresent() 方法，保證一定對get(key)更新。
如下：

	String[] row= getDateByTableName.getDatasbyName(tableName,jsonItem,mapTableColumns.get(tableName));
	
	mapTableValues.computeIfAbsent(tableName,k->new ArrayList<String[]>());
	
	List<String[]> rowList=mapTableValues.get(tableName);
	rowList.add(row);
	
	mapTableValues.computeIfPresent(tableName,(k,v) -> rowList);

最後再檢測，消除異味。雖然這段代碼看上去沒有原來那麼好理解，但是我覺得還是個人對這個方法熟悉與否，put和get方法大家比較熟悉，所以一眼就能看懂，但是computeIfAbsent和computeIfPresent 方法在java 8 中才有，大家用的比較少，所以才顯得陌生。

最後附上這個方法改好的代碼