MSSQL Client routing regression (#1365)

Fixes #1361 Signed-off-by: Thomas Segismont <[email protected]>
eclipse-vertx · Oct 25, 2023 · 4bc31ae · 4bc31ae
1 parent 68eb652
commit 4bc31ae
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 8 deletions.
diff --git a/vertx-mssql-client/src/main/java/io/vertx/mssqlclient/impl/MSSQLConnectionFactory.java b/vertx-mssql-client/src/main/java/io/vertx/mssqlclient/impl/MSSQLConnectionFactory.java
@@ -16,6 +16,7 @@
 import io.vertx.core.Promise;
 import io.vertx.core.impl.ContextInternal;
 import io.vertx.core.impl.VertxInternal;
+import io.vertx.core.net.HostAndPort;
 import io.vertx.core.net.NetSocket;
 import io.vertx.core.net.SocketAddress;
 import io.vertx.core.net.impl.NetSocketInternal;
@@ -60,13 +61,18 @@ private Future<Connection> connectOrRedirect(MSSQLConnectOptions options, Contex
       )
       .compose(connBase -> {
         MSSQLSocketConnection conn = (MSSQLSocketConnection) connBase;
-        SocketAddress alternateServer = conn.getAlternateServer();
+        HostAndPort alternateServer = conn.getAlternateServer();
         if (alternateServer == null) {
           return context.succeededFuture(conn);
         }
         Promise<Void> closePromise = context.promise();
         conn.close(null, closePromise);
-        return closePromise.future().transform(v -> connectOrRedirect(options, context, redirections + 1));
+        return closePromise.future().transform(v -> {
+          MSSQLConnectOptions connectOptions = new MSSQLConnectOptions(options)
+            .setHost(alternateServer.host())
+            .setPort(alternateServer.port());
+          return connectOrRedirect(connectOptions, context, redirections + 1);
+        });
       });
   }
 

diff --git a/vertx-mssql-client/src/main/java/io/vertx/mssqlclient/impl/MSSQLSocketConnection.java b/vertx-mssql-client/src/main/java/io/vertx/mssqlclient/impl/MSSQLSocketConnection.java
@@ -20,7 +20,7 @@
 import io.vertx.core.impl.ContextInternal;
 import io.vertx.core.impl.future.PromiseInternal;
 import io.vertx.core.net.ClientSSLOptions;
-import io.vertx.core.net.SocketAddress;
+import io.vertx.core.net.HostAndPort;
 import io.vertx.core.net.impl.NetSocketInternal;
 import io.vertx.core.net.impl.SSLHelper;
 import io.vertx.core.net.impl.SslChannelProvider;
@@ -52,7 +52,7 @@ public class MSSQLSocketConnection extends SocketConnectionBase {
   private final SSLHelper sslHelper;
 
   private MSSQLDatabaseMetadata databaseMetadata;
-  private SocketAddress alternateServer;
+  private HostAndPort alternateServer;
 
   MSSQLSocketConnection(NetSocketInternal socket,
                         SSLHelper sslHelper,
@@ -181,11 +181,11 @@ private void setDatabaseMetadata(MSSQLDatabaseMetadata metadata) {
     this.databaseMetadata = metadata;
   }
 
-  public SocketAddress getAlternateServer() {
+  public HostAndPort getAlternateServer() {
     return alternateServer;
   }
 
-  public void setAlternateServer(SocketAddress alternateServer) {
+  public void setAlternateServer(HostAndPort alternateServer) {
     this.alternateServer = alternateServer;
   }
 }
diff --git a/vertx-mssql-client/src/main/java/io/vertx/mssqlclient/impl/codec/InitCommandCodec.java b/vertx-mssql-client/src/main/java/io/vertx/mssqlclient/impl/codec/InitCommandCodec.java
@@ -12,7 +12,7 @@
 package io.vertx.mssqlclient.impl.codec;
 
 import io.netty.buffer.ByteBuf;
-import io.vertx.core.net.SocketAddress;
+import io.vertx.core.net.HostAndPort;
 import io.vertx.mssqlclient.MSSQLConnectOptions;
 import io.vertx.mssqlclient.impl.MSSQLSocketConnection;
 import io.vertx.mssqlclient.impl.protocol.client.login.LoginPacket;
@@ -220,6 +220,6 @@ protected void handleRouting(ByteBuf payload) {
       throw new IllegalStateException("ProtocolProperty value of zero is not allowed when Protocol is TCP-IP");
     }
     String host = readUnsignedShortLengthString(payload);
-    conn.setAlternateServer(SocketAddress.inetSocketAddress(port, host.toLowerCase(ENGLISH)));
+    conn.setAlternateServer(HostAndPort.create(host.toLowerCase(ENGLISH), port));
   }
 }