From 9e91ad6ac30ce190b22d5ee5f8f06817365e1bc7 Mon Sep 17 00:00:00 2001
From: windust <freddy@spinningnoodle.com>
Date: Thu, 2 Nov 2023 14:53:05 -0700
Subject: [PATCH] Timeout when doing take (to poll)

This allows graceful termination of remotedispatcher
---
 .../www/haystack/client/dispatchers/RemoteDispatcher.java       | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/core/src/main/java/com/expedia/www/haystack/client/dispatchers/RemoteDispatcher.java b/core/src/main/java/com/expedia/www/haystack/client/dispatchers/RemoteDispatcher.java
index dd42624..2f3a597 100644
--- a/core/src/main/java/com/expedia/www/haystack/client/dispatchers/RemoteDispatcher.java
+++ b/core/src/main/java/com/expedia/www/haystack/client/dispatchers/RemoteDispatcher.java
@@ -82,7 +82,7 @@ public void run() {
             while (running.get() || !(acceptQueue.isEmpty())) {
 
                 try (Sample timer = sendTimer.start()) {
-                    Span span = acceptQueue.take();
+                    Span span = acceptQueue.poll(1000, TimeUnit.MILLISECONDS);
                     try {
                         client.send(span);
                     } catch (ClientException e) {