commit abd1e9ce0327c5e80bf262b7a17fccce895463c8
author Martin Fick US <mfick@nvidia.com> Thu Oct 02 13:46:38 2025 -0700
committer Gerrit Code Review <noreply-gerritcodereview@google.com> Thu Oct 02 13:46:38 2025 -0700
tree c438c4b544dd1a54da14c15a387b7686294f5600
parent 80e774ae968915d7becd10794eaaf2d67fe68e22
parent 7e4ca18195c8aaf510643d48ad6ab1cd71040499

Merge "Replace Semaphore with AtomicInteger"

src/main/java/com/googlesource/gerrit/plugins/quota/TaskQuotaWithPermits.java

diff --git a/src/main/java/com/googlesource/gerrit/plugins/quota/TaskQuotaWithPermits.java b/src/main/java/com/googlesource/gerrit/plugins/quota/TaskQuotaWithPermits.java
index 027ddcf..ea414ca 100644
--- a/src/main/java/com/googlesource/gerrit/plugins/quota/TaskQuotaWithPermits.java
+++ b/src/main/java/com/googlesource/gerrit/plugins/quota/TaskQuotaWithPermits.java
@@ -1,20 +1,25 @@
 package com.googlesource.gerrit.plugins.quota;
 
 import com.google.gerrit.server.git.WorkQueue;
-import java.util.concurrent.Semaphore;
+import java.util.concurrent.atomic.AtomicInteger;
 
 public abstract class TaskQuotaWithPermits implements TaskQuota {
-  protected final Semaphore permits;
+  protected final AtomicInteger permits;
 
   public TaskQuotaWithPermits(int maxPermits) {
-    this.permits = new Semaphore(maxPermits);
+    this.permits = new AtomicInteger(maxPermits);
   }
 
   public boolean isReadyToStart(WorkQueue.Task<?> task) {
-    return permits.tryAcquire();
+    if (permits.decrementAndGet() >= 0) {
+      return true;
+    }
+
+    permits.incrementAndGet();
+    return false;
   }
 
   public void onStop(WorkQueue.Task<?> task) {
-    permits.release();
+    permits.incrementAndGet();
   }
 }