如何统计特定用户的REST API轮询次数？ConcurrentHashMap是否最优？

需求：当特定用户的REST API轮询次数达到指定数值（示例中为5次）且满足另一附加条件时，触发通知。当前采用以userId为键、AtomicInteger为值的ConcurrentHashMap实现，相关Java代码如下：

public class MyClass{
    private static final ConcurrentHashMap<Integer, AtomicInteger> quoteRequestCount = new ConcurrentHashMap<>();


    mymeth(){
        map.putIfAbsent(<id of user>, new AtomicInteger(0));
        map.get(<id of user>).getAndIncrement();

        // 一系列操作

        AtomicInteger hitCount = map.get(userId);
        if (hitCount == 5 && /* 依赖上述操作的条件 */)
        {
             // 发送通知
        }
    }
}

请问ConcurrentHashMap是否为该场景下的最理想选择？

先修正原代码的一个错误 #

首先要指出：你代码里hitCount == 5的写法是错误的——AtomicInteger是对象，==比较的是引用地址，应该用hitCount.get() == 5才能正确判断计数的数值。

ConcurrentHashMap的适用性分析 #

回到核心问题：ConcurrentHashMap在这个多线程计数触发通知的场景下是合格的选择，但称不上“最理想”，具体拆解如下：

• 它的优势：
  • 天生线程安全，完美适配REST API多线程调用的场景；
  • putIfAbsent是原子操作，能避免多个线程同时初始化同一个用户的AtomicInteger，不会出现计数初始化重复的问题；
  • 并发性能远优于Hashtable或加锁的普通HashMap，Java8+版本用CAS+synchronized实现，高并发下的性能开销更低。
• 它的局限：
  • 无自动清理机制：如果用户不再调用API，对应的计数条目会一直留在Map中，长期运行会占用不必要的内存；
  • 计数生命周期需要手动维护：比如达到指定次数后，要不要重置计数？要不要移除条目？这些逻辑都得自己实现，否则可能出现重复触发通知的问题。

更优的替代方案（按需选择） #

如果你的场景有额外需求，可以考虑更合适的方案：

• Guava LoadingCache：支持设置过期时间（比如用户1小时不调用就自动清理计数），还能自动加载初始计数，省去手动写putIfAbsent的代码，逻辑更简洁；
• Redis（分布式场景）：如果你的服务是集群部署，ConcurrentHashMap只能实现单节点计数，这时候必须用Redis这类分布式缓存来做跨节点的统一计数。

总结 #

如果只是单节点服务、对内存占用要求不高，ConcurrentHashMap完全够用，但要修正代码里的判断错误，并且补充必要的计数清理/重置逻辑；如果有分布式部署、内存优化等需求，它就不是最理想的选择，需要更换对应的方案。

内容的提问来源于stack exchange，提问作者Janitta James