Java无锁计数器实现技巧

本文深入解析了如何在Java中利用while循环结合AtomicInteger的CAS（比较并交换）机制实现高效、线程安全的无锁计数器，强调其核心在于“读取—计算—条件判断—原子更新—失败重试”的自旋重试模式；不仅给出了incrementAndGetOldValue和带阈值限制的tryIncrementToLimit等典型实现，更关键地揭示了避免常见陷阱的要点：条件判断必须紧邻get()置于循环内以保证快照一致性，循环体必须幂等且无副作用，同时兼顾性能与可读性——优先使用内置原子方法，仅在需复合原子操作时手写CAS循环，并合理运用onSpinWait优化高竞争场景。

在 Java 中，用 while 循环配合 AtomicInteger 实现无锁计数器，核心是利用其 compareAndSet（CAS）方法实现“乐观锁”式更新，避免 synchronized 带来的线程阻塞。关键不是写个 while 就行，而是要确保循环体里做的是原子性重试逻辑。

理解 CAS 重试模式的本质

AtomicInteger.compareAndSet(expected, newValue) 只有当当前值等于 expected 时才把值设为 newValue，并返回是否成功。失败说明值已被其他线程修改，此时应重新读取最新值，再计算新目标值，继续尝试——这就是典型的 CAS 自旋重试。

例如实现一个线程安全的“加 1 并返回旧值”操作：

public int incrementAndGetOldValue(AtomicInteger counter) {
    int current;
    int next;
    do {
        current = counter.get();
        next = current + 1;
    } while (!counter.compareAndSet(current, next));
    return current;
}

避免常见陷阱：不要在循环中做非幂等或耗时操作

CAS 循环可能重试多次，因此循环体内必须满足两个条件：

• 所有计算逻辑必须是纯函数式的（不依赖外部可变状态、不产生副作用）
• 不能包含 I/O、锁、sleep、随机数生成等可能阻塞或不可重入的操作
• 如果需要基于旧值做复杂判断（比如“只在大于 100 时才加 1”），要把判断逻辑放进循环内，并确保 compareAndSet 的 expected 仍反映该判断所依据的真实前提

实现带条件的无锁计数逻辑（如限流计数器）

比如实现一个“最多允许累加到 1000，超了就停止”的计数器：

public boolean tryIncrementToLimit(AtomicInteger counter, int limit) {
    int current;
    int next;
    do {
        current = counter.get();
        if (current >= limit) {
            return false; // 已达上限，不再递增
        }
        next = current + 1;
    } while (!counter.compareAndSet(current, next));
    return true;
}

注意：这里的 if (current >= limit) 判断必须放在循环内，且紧邻 get() 之后，才能保证“读取-判断-尝试更新”三者对同一快照生效。若把判断提到循环外，就会因并发导致误判。

性能与可读性的平衡建议

虽然 CAS 循环无锁，但高竞争下会频繁重试，带来 CPU 开销。实际使用中建议：

• 优先考虑 AtomicInteger.incrementAndGet() 等内置方法，它们已高度优化，语义清晰
• 仅当需要复合操作（如“先检查再更新”且中间不能被干扰）时，才手写 CAS 循环
• 必要时可用 Thread.onSpinWait()（Java 9+）提示 CPU 当前处于自旋等待，提升能效
• 避免嵌套多层 CAS 逻辑；复杂状态建议改用 AtomicReference 封装整个对象

本篇关于《Java无锁计数器实现技巧》的介绍就到此结束啦，但是学无止境，想要了解学习更多关于文章的相关知识，请关注golang学习网公众号！