AtomicInteger 和比较操作

Question

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

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

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

Answer 1

由于incrementAndGet()是原子操作，因此最多有一个线程会获取1结果（只要 AtomicInteger 不设置为低于 1 的值）。这是由 AtomicInteger 保证的。尽管实际的比较不是该原子操作的一部分，但只有接收到的线程1才会进入该if块。这是因为比较比较的是类型 的两个值： ) 和 的int结果。AtomicInteger 在其中不起作用，因为当内部状态发生变化时，原语不会改变它的值。tl;dr：您的代码相当于：incrementAndGet()maxintcount

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

Java提供了许多其他方法来实现线程安全（Lock，CountDownLatch，Semaphore，...），但由于您指的是 AtomicInteger ，我相信AtomicInteger.compareAndSet()更准确地满足您的要求：

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