Fixing requestId generator bug in PinotBrokerDebug Resource

apache · Jul 30, 2024 · 464b7ab · 464b7ab
1 parent a6cf197
commit 464b7ab
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/pinot-broker/src/main/java/org/apache/pinot/broker/api/resources/PinotBrokerDebug.java b/pinot-broker/src/main/java/org/apache/pinot/broker/api/resources/PinotBrokerDebug.java
@@ -84,7 +84,7 @@
 public class PinotBrokerDebug {
 
   // Request ID is passed to the RoutingManager to rotate the selected replica-group.
-  private final AtomicLong _requestIdGenerator = new AtomicLong();
+  private final static AtomicLong _requestIdGenerator = new AtomicLong();
 
   @Inject
   private BrokerRoutingManager _routingManager;
@@ -272,6 +272,9 @@ public String getServerRoutingStats() {
   }
 
   private long getRequestId() {
+    if (_requestIdGenerator.get() == Long.MAX_VALUE) {
+      _requestIdGenerator.set(0);
+    }
     return _requestIdGenerator.getAndIncrement();
   }