Fix agent consul forwarder discovery

Prefer usage of service host qhen available. With IP per container, it is not relevant to use the node host.
criteo · Jul 22, 2024 · 25bb5c8 · 25bb5c8
1 parent c189855
commit 25bb5c8
Showing 1 changed file with 12 additions and 4 deletions.
diff --git a/agent/src/main/java/com/criteo/hadoop/garmadon/agent/ConsulConnection.java b/agent/src/main/java/com/criteo/hadoop/garmadon/agent/ConsulConnection.java
@@ -63,12 +63,20 @@ private List<HealthService> getHealthyEndPoints() {
     @Override
     public void establishConnection() {
         for (; ; ) {
-            List<HealthService> nodes = getHealthyEndPoints();
+            List<HealthService> healthServices = getHealthyEndPoints();
 
-            HealthService electedNode = nodes.get(ThreadLocalRandom.current().nextInt(nodes.size()));
+            HealthService randomHealthyService = healthServices.get(ThreadLocalRandom.current().nextInt(healthServices.size()));
 
-            String host = electedNode.getNode().getAddress();
-            Integer port = electedNode.getService().getPort();
+            String nodeHost = randomHealthyService.getNode().getAddress();
+            String serviceHost = randomHealthyService.getService().getAddress();
+
+            String host;
+            if (serviceHost.isEmpty()) {
+                host = nodeHost;
+            } else {
+                host = serviceHost;
+            }
+            Integer port = randomHealthyService.getService().getPort();
 
             if (LOGGER.isDebugEnabled()) {
                 LOGGER.debug("will use forwarder at " + host + ":" + port);