AtomicInteger 和比较操作

AtomicInteger and Compare operations

我正在尝试使用 AtomicInteger 变量作为锁。 所以,想知道下面发布的代码是否是线程安全的。我知道 incrementAndGet() 是一个原子操作。但是,我不确定后续的“==”操作是否也是原子操作(如果在我进行比较时值增加到 2 会怎么样。)。所以发布这个问题来听听你的想法。

private AtomicInteger count = new AtomicInteger(0); //Shared variable
private final int max = 1;

if(count.incrementAndGet() == max) {
  //Do something
}

因为incrementAndGet()是一个原子操作,最多一个线程会得到1的结果(只要AtomicInteger没有被设置为小于1的值)。这是由 AtomicInteger 保证的。虽然实际比较不是这个原子操作的一部分,但是只有接收到1的线程才会运行进入if块。这是因为比较会比较两个 int 类型的值:incrementAndGet()) 和 max 的结果。 AtomicInteger 在其中不起作用,因为当 count 的内部状态发生变化时,原语 int 不会改变它的值。 tl;dr:您的代码相当于:

final int val = count.incrementAndGet(); // (it makes no difference if val is not 'final`)
if (val == max) {
    // ...
}

Java 提供了许多其他方法来实现线程安全(Lock, CountDownLatch, Semaphore, ...), but since you are referring to an AtomicInteger, I believe AtomicInteger.compareAndSet() 对您的要求更准确:

if (count.compareAndSet(0, 1)) {
    // executed if count == 0
    count.set(0);
}