如果当前值小于另一个值，则进行原子更新

Question

我正在寻找像 AtomicInteger 或 LongAddr 这样的东西：

• 如果值小于MAX，则递增，其中MAX 是某个用户定义的值。
• 返回一个值，指示原子是否增加。

用例：

• 我有一个任务队列。
• 只有MAX 任务应该同时运行。
• 当一个新任务添加到队列中时，如果正在进行的任务数少于MAX，我想运行它

我不能使用AtomicInteger 或LongAddr 的原因是它们只允许您与特定值而不是值范围进行比较。

澄清：我不希望解决方案实际执行任务。我的用例涉及将网络请求传递给 Jetty。它使用单个线程来驱动多个网络请求。任何触发Executor 的解决方案都无法达到此目的，因为这样我最终会为每个网络请求提供一个线程。

Answer 1

Andy Turner provided an excellent answer 但我发现这个解决方案更具可读性。本质上，我们只需要new Semaphore(MAX) 和Semaphore.tryAcquire()。

如果您深入研究Semaphore 的源代码，您会发现其实现类似于安迪的答案。

这里是一些示例代码：

Semaphore semaphore = new Semaphore(MAX);
// ... much later ...
public void addTask(Runnable task)
{
  if (semaphore.tryAcquire())
    task.run();
  else
    queue.add(task);
}

public void afterTaskComplete(Runnable task)
{
  semaphore.release();
}

Answer 2

使用compareAndSet():

boolean incrementToTheMax(AtomicInteger atomicInt, int max) {
  while (true) {
    int value = atomicInt.get();
    if (value >= max) {
      // The counter has already reached max, so don't increment it.
      return false;
    }
    if (atomicInt.compareAndSet(value, value+1)) {
      // If we reach here, the atomic integer still had the value "value";
      // and so we incremented it.
      return true;
    }
    // If we reach here, some other thread atomically updated the value.
    // Rats! Loop, and try to increment of again.
  }
}