Apache
/
druid
mirror of https://github.com/apache/druid.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Use DruidLeaderSelector in CliCoordinator.HearbeatSupplier (#16215)

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.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;
};