Use DruidLeaderSelector in CliCoordinator.HearbeatSupplier (apache#16215

)
monetate · Mar 28, 2024 · 3471352 · 3471352
1 parent 524842a
commit 3471352
Showing 1 changed file with 5 additions and 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
@@ -42,6 +42,7 @@
 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,20 +466,20 @@ public CoordinatorCustomDutyGroups get()
 
   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
     public Supplier<Map<String, Object>> get()
     {
       return () -> {
         Map<String, Object> heartbeatTags = new HashMap<>();
-        heartbeatTags.put("leader", coordinator.isLeader() ? 1 : 0);
+        heartbeatTags.put("leader", leaderSelector.isLeader() ? 1 : 0);
 
         return heartbeatTags;
       };