(druid) branch master updated: Use DruidLeaderSelector in CliCoordinator.HearbeatSupplier (#16215)

[kfaraz]

This is an automated email from the ASF dual-hosted git repository.

kfaraz pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/druid.git


The following commit(s) were added to refs/heads/master by this push:
     new 3471352dac6 Use DruidLeaderSelector in CliCoordinator.HearbeatSupplier 
(#16215)
3471352dac6 is described below

commit 3471352dac69980fc56970fe99e0c4e76f169cad
Author: Rishabh Singh <6513075+findingr...@users.noreply.github.com>
AuthorDate: Thu Mar 28 21:42:33 2024 +0530

    Use DruidLeaderSelector in CliCoordinator.HearbeatSupplier (#16215)
---
 services/src/main/java/org/apache/druid/cli/CliCoordinator.java | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/services/src/main/java/org/apache/druid/cli/CliCoordinator.java 
b/services/src/main/java/org/apache/druid/cli/CliCoordinator.java
index c6b9e27238c..36e78a51e63 100644
--- a/services/src/main/java/org/apache/druid/cli/CliCoordinator.java
+++ b/services/src/main/java/org/apache/druid/cli/CliCoordinator.java
@@ -42,6 +42,7 @@ import org.apache.druid.client.DirectDruidClientFactory;
 import org.apache.druid.client.HttpServerInventoryViewResource;
 import org.apache.druid.client.InternalQueryConfig;
 import org.apache.druid.client.coordinator.Coordinator;
+import org.apache.druid.discovery.DruidLeaderSelector;
 import org.apache.druid.discovery.NodeRole;
 import org.apache.druid.error.DruidException;
 import org.apache.druid.guice.ConfigProvider;
@@ -465,12 +466,12 @@ public class CliCoordinator extends ServerRunnable
 
   private static class HeartbeatSupplier implements 
Provider<Supplier<Map<String, Object>>>
   {
-    private final DruidCoordinator coordinator;
+    private final DruidLeaderSelector leaderSelector;
 
     @Inject
-    public HeartbeatSupplier(DruidCoordinator coordinator)
+    public HeartbeatSupplier(@Coordinator DruidLeaderSelector leaderSelector)
     {
-      this.coordinator = coordinator;
+      this.leaderSelector = leaderSelector;
     }
 
     @Override
@@ -478,7 +479,7 @@ public class CliCoordinator extends ServerRunnable
     {
       return () -> {
         Map<String, Object> heartbeatTags = new HashMap<>();
-        heartbeatTags.put("leader", coordinator.isLeader() ? 1 : 0);
+        heartbeatTags.put("leader", leaderSelector.isLeader() ? 1 : 0);
 
         return heartbeatTags;
       };


---------------------------------------------------------------------
To unsubscribe, e-mail: commits-unsubscr...@druid.apache.org
For additional commands, e-mail: commits-h...@druid.apache.org