Reverted changes in SshBastionContainer (#13934)

airbytehq · Jun 20, 2022 · de05753 · de05753
1 parent c283d9d
commit de05753
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/...s/bases/base-java/src/main/java/io/airbyte/integrations/base/ssh/SshBastionContainer.java b/...s/bases/base-java/src/main/java/io/airbyte/integrations/base/ssh/SshBastionContainer.java
@@ -40,9 +40,13 @@ public JsonNode getTunnelConfig(final SshTunnel.TunnelMethod tunnelMethod, final
 
     return Jsons.jsonNode(builderWithSchema
         .put("tunnel_method", Jsons.jsonNode(ImmutableMap.builder()
-            .put("tunnel_host", bastion.getHost())
+            .put("tunnel_host",
+                Objects.requireNonNull(bastion.getContainerInfo().getNetworkSettings()
+                    .getNetworks()
+                    .get(((Network.NetworkImpl) network).getName())
+                    .getIpAddress()))
             .put("tunnel_method", tunnelMethod)
-            .put("tunnel_port", bastion.getFirstMappedPort())
+            .put("tunnel_port", bastion.getExposedPorts().get(0))
             .put("tunnel_user", SSH_USER)
             .put("tunnel_user_password", tunnelMethod.equals(SSH_PASSWORD_AUTH) ? SSH_PASSWORD : "")
             .put("ssh_key", tunnelMethod.equals(SSH_KEY_AUTH) ? bastion.execInContainer("cat", "var/bastion/id_rsa").getStdout() : "")