Map.computeIfPresent() 方法的使用

一.简介

JDK8中新增了Map.computeIfPresent()方法，computeIfPresent() 方法对 hashMap 中指定 key 的值进行重新计算，前提是该 key 存在于 hashMap 中。

computeIfPresent() 方法的语法为：

hashmap.computeIfPresent(K key, BiFunction remappingFunction)

注意：hashmap 是 HashMap 类的一个对象。

参数说明：

• key - 键
• remappingFunction - 重新映射函数，用于重新计算值

返回值

如果 key 对应的 value 不存在，则返回该 null，如果存在，则返回通过 remappingFunction 重新计算后的值。

二.源码

    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;
        }
    }

三.实例

1.简单实例

import java.util.HashMap;

class Main {
    public static void main(String[] args) {
        // 创建一个 HashMap
        HashMap<String, Integer> prices = new HashMap<>();

        // 往HashMap中添加映射关系
        prices.put("Shoes", 200);
        prices.put("Bag", 300);
        prices.put("Pant", 150);
        System.out.println("HashMap: " + prices);

        // 重新计算鞋加上10%的增值税后的价值
        int shoesPrice = prices.computeIfPresent("Shoes", (key, value) -> value + value * 10/100);
        System.out.println("Price of Shoes after VAT: " + shoesPrice);

        // 输出更新后的HashMap
        System.out.println("Updated HashMap: " + prices);
    }
}

执行以上程序输出结果为：

HashMap: {Pant=150, Bag=300, Shoes=200}
Price of Shoes after VAT: 220
Updated HashMap: {Pant=150, Bag=300, Shoes=220}}

2.开发实例

    /**
     * 将正在查看的未读消息MsgId缓存起来，防止ws重复发送已读消息
     */
    private static final Map<String, Set<String>> readMsgCache = new ConcurrentHashMap<>(2048);

    /**
     * 组装正在查看未读消息的MsgId缓存Key
     * @param appId
     * @param receive
     * @return
     */
    private static String getReadMsgCacheKey(String appId, Long receive) {
        return appId + "_" + receive;
    }

    /**
     * 将正在查看的消息缓存起来
     * @param appId 应用ID
     * @param receiverAccountId 接收人ID
     * @param msgIds 消息ID
     */
    public static void cacheReadMsg(String appId, Long receiverAccountId, List<String> msgIds) {
        readMsgCache.merge(getReadMsgCacheKey(appId, receiverAccountId), new HashSet<>(msgIds), (o, n) -> {
            o.addAll(n);
            return o;
        });
    }

    /**
     * 将缓存的未读消息移除
     * @param appId 应用ID
     * @param receiverAccountId 接收人ID
     * @param msgIds 消息ID
     */
    public static void removeCache(String appId, Long receiverAccountId, List<String> msgIds) {
        readMsgCache.computeIfPresent(getReadMsgCacheKey(appId, receiverAccountId), (k, v) -> {
            msgIds.forEach(v::remove);
            return v;
        });
    }

    /**
     * 获取正在查看未读消息的MsgId，防止ws重复发送未读消息
     * @param appId            应用ID
     * @param receiveAccountId 接收人ID
     * @return
     */
    public static Set<String> getReadMsgCache(String appId, Long receiveAccountId) {
        return readMsgCache.getOrDefault(getReadMsgCacheKey(appId, receiveAccountId), Collections.emptySet());
    }