Apache/druid
1
0
Fork 0
mirror of https://github.com/apache/druid.git synced 2025-02-17 07:25:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use DruidLeaderSelector in CliCoordinator.HearbeatSupplier (#16215)

Browse Source
This commit is contained in:
Rishabh Singh 2024-03-28 21:42:33 +05:30 committed by GitHub
parent 524842a3bb
commit 3471352dac
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
services/src/main/java/org/apache/druid/cli/CliCoordinator.java
View File

@ -42,6 +42,7 @@ import org.apache.druid.client.DirectDruidClientFactory;
import org.apache.druid.client.HttpServerInventoryViewResource; import org.apache.druid.client.HttpServerInventoryViewResource;
import org.apache.druid.client.InternalQueryConfig; import org.apache.druid.client.InternalQueryConfig;
import org.apache.druid.client.coordinator.Coordinator; import org.apache.druid.client.coordinator.Coordinator;
import org.apache.druid.discovery.DruidLeaderSelector;
import org.apache.druid.discovery.NodeRole; import org.apache.druid.discovery.NodeRole;
import org.apache.druid.error.DruidException; import org.apache.druid.error.DruidException;
import org.apache.druid.guice.ConfigProvider; 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 static class HeartbeatSupplier implements Provider<Supplier<Map<String, Object>>>
{ {
private final DruidCoordinator coordinator; private final DruidLeaderSelector leaderSelector;
@Inject @Inject
public HeartbeatSupplier(DruidCoordinator coordinator) public HeartbeatSupplier(@Coordinator DruidLeaderSelector leaderSelector)
{ {
this.coordinator = coordinator; this.leaderSelector = leaderSelector;
} }
@Override @Override
@ -478,7 +479,7 @@ public class CliCoordinator extends ServerRunnable
{ {
return () -> { return () -> {
Map<String, Object> heartbeatTags = new HashMap<>(); Map<String, Object> heartbeatTags = new HashMap<>();
heartbeatTags.put("leader", coordinator.isLeader() ? 1 : 0); heartbeatTags.put("leader", leaderSelector.isLeader() ? 1 : 0);
return heartbeatTags; return heartbeatTags;
}; };