Merge pull request #201 from DataDog/vickenty/drain-loop

Avoid busy loop on client shutdown
DataDog · Sep 2, 2022 · e4fc80f · e4fc80f
2 parents ad4daea + 50ceb1b
commit e4fc80f
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/main/java/com/timgroup/statsd/StatsDProcessor.java b/src/main/java/com/timgroup/statsd/StatsDProcessor.java
@@ -90,6 +90,12 @@ protected void processLoop() {
                             closeSignal.countDown();
                             clientClosed = true;
                         }
+                        if (clientClosed) {
+                            // We are draining highPrioMessages, which is a non-blocking
+                            // queue. Avoid a busy loop if the queue is empty while the aggregator
+                            // is flushing.
+                            Thread.sleep(WAIT_SLEEP_MS);
+                        }
                         continue;
                     }