17370845950

在Java中，ConcurrentHashMap 的 compute 方法提供了一种线程安全的方式来更新映射中的值。它结合了读取、计算和写入操作，避免了显式的加锁，适用于高并发场景。

compute 方法的基本用法

compute(K key, BiFunction super K, ? super V, ? extends V> remappingFunction) 方法会根据给定的 key 和一个重映射函数来计算新的值。如果函数返回 null，该 key 会被移除。

常见使用场景包括计数器累加、集合构建等。例如，统计单词出现次数：

ConcurrentHashMap map = new ConcurrentHashMap<>();
String word = "hello";

map.compute(word, (k, v) -> (v == null) ? 1 : v + 1);

这段代码的意思是：如果 word 不存在（v 为 null），就设置为 1；否则在原值基础上加 1。整个操作是原子的，无需额外同步。

computeIfPresent 和 computeIfAbsent

除了 compute，还有两个更精确的方法：

• computeIfAbsent：仅当 key 不存在时才计算并插入值，常用于缓存加载
• computeIfPresent：仅当 key 存在时才进行计算更新

比如用 computeIfAbsent 初始化并发缓存：

ConcurrentHashMap> cache = new ConcurrentHashMap<>();

cache.computeIfAbsent("users", k -> new ArrayList<>()).add("alice");

这行代码确保 key 对应的 list 被创建后，再添加元素。多个线程同时调用也不会冲突。

注意事项与线程安全

虽然 compute 系列方法本身是线程安全的，但 remappingFunction 应尽量保持简洁且无副作用。不要在函数内部执行耗时操作或修改外部状态，以免影响性能或引发死锁。

另外，回调函数中不应尝试对同一个 map 做其他 compute 操作，可能导致死循环或死锁。

基本上就这些。合理使用 compute 方法可以让并发代码更简洁、安全。