java - ConcurrentHashMap 原子获取、增量和替换

Question

是否有一种简单的方法可以从 ConcurrentHashMap 中获取、递增和替换整数？

ConcurrentHashMap<MyResource, AtomicInteger> map;

假设每个值都已经像这样初始化:

map.get(i) = new AtomicInteger(0);

并且我们不会添加任何新 key 。

如何进行获取、递增和替换原子操作(且线程安全)？

像这样的东西不应该工作:

map.get(myResourceKey).incrementAndGet();

因为 2 个线程在递增之前可以获得相同的整数，对吧？

该映射正在跟踪有多少线程正在使用特定资源。我希望仅在未使用资源时(计数器 = 0)发生某些事情。

Answer 1

您可以使用计算来执行您所要求的原子操作。计算方法将返回递增的值，如果是第一次调用，则返回 1。

Map<MyResource, Integer> map = new ConcurrentHashMap<>();

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