Added implementation for useDiscovery

Author: alex268

jdbc/src/main/java/tech/ydb/jdbc/context/YdbContext.java

@@ -10,6 +10,7 @@
 
 import tech.ydb.core.grpc.GrpcTransport;
 import tech.ydb.core.grpc.GrpcTransportBuilder;
+import tech.ydb.core.impl.SingleChannelTransport;
 import tech.ydb.core.settings.BaseRequestSettings;
 import tech.ydb.jdbc.YdbPrepareMode;
 import tech.ydb.jdbc.YdbTracer;
@@ -238,7 +239,7 @@ public static YdbContext createContext(YdbConfig config) throws SQLException {
                 });
             });
 
-            grpcTransport = builder.build();
+            grpcTransport = config.isUseDiscovery() ? builder.build() : new SingleChannelTransport(builder);
 
             PooledTableClient.Builder tableClient = PooledTableClient.newClient(
                     GrpcTableRpc.useTransport(grpcTransport)

jdbc/src/test/java/tech/ydb/jdbc/YdbDriverExampleTest.java

@@ -8,8 +8,9 @@
 import java.sql.Types;
 
 import org.junit.jupiter.api.Assertions;
-import org.junit.jupiter.api.Test;
 import org.junit.jupiter.api.extension.RegisterExtension;
+import org.junit.jupiter.params.ParameterizedTest;
+import org.junit.jupiter.params.provider.EnumSource;
 
 import tech.ydb.test.junit5.YdbHelperExtension;
 
@@ -21,23 +22,10 @@ public class YdbDriverExampleTest {
     @RegisterExtension
     private static final YdbHelperExtension ydb = new YdbHelperExtension();
 
-    private static String jdbcURL() {
-        StringBuilder jdbc = new StringBuilder("jdbc:ydb:")
-                .append(ydb.useTls() ? "grpcs://" : "grpc://")
-                .append(ydb.endpoint())
-                .append("/?database=")
-                .append(ydb.database());
-
-        if (ydb.authToken() != null) {
-            jdbc.append("&").append("token=").append(ydb.authToken());
-        }
-
-        return jdbc.toString();
-    }
-
-    @Test
-    public void testYdb() throws SQLException {
-        try (Connection connection = DriverManager.getConnection(jdbcURL())) {
+    @ParameterizedTest
+    @EnumSource(Mode.class)
+    public void testYdb(Mode mode) throws SQLException {
+        try (Connection connection = DriverManager.getConnection(mode.getJdbcURL())) {
             try {
                 connection.createStatement()
                         .execute("drop table table_sample");
@@ -148,9 +136,10 @@ public void testYdb() throws SQLException {
         }
     }
 
-    @Test
-    public void testYdbNotNull() throws SQLException {
-        try (Connection connection = DriverManager.getConnection(jdbcURL())) {
+    @ParameterizedTest
+    @EnumSource(Mode.class)
+    public void testYdbNotNull(Mode mode) throws SQLException {
+        try (Connection connection = DriverManager.getConnection(mode.getJdbcURL())) {
             try {
                 connection.createStatement().execute("drop table table_sample");
             } catch (SQLException e) {
@@ -249,4 +238,57 @@ public void testYdbNotNull() throws SQLException {
             }
         }
     }
+
+    private enum Mode {
+        BASE {
+            @Override
+            String getJdbcURL() {
+                StringBuilder jdbc = new StringBuilder("jdbc:ydb:")
+                        .append(ydb.useTls() ? "grpcs://" : "grpc://")
+                        .append(ydb.endpoint())
+                        .append("/")
+                        .append(ydb.database());
+
+                if (ydb.authToken() != null) {
+                    jdbc.append("?").append("token=").append(ydb.authToken());
+                }
+
+                return jdbc.toString();
+            }
+        },
+        OLD_STYLE {
+            @Override
+            String getJdbcURL() {
+                StringBuilder jdbc = new StringBuilder("jdbc:ydb:")
+                        .append(ydb.useTls() ? "grpcs://" : "grpc://")
+                        .append(ydb.endpoint())
+                        .append("/?database=")
+                        .append(ydb.database());
+
+                if (ydb.authToken() != null) {
+                    jdbc.append("&").append("token=").append(ydb.authToken());
+                }
+
+                return jdbc.toString();
+            }
+        },
+        NO_DISCOVERY {
+            @Override
+            String getJdbcURL() {
+                StringBuilder jdbc = new StringBuilder("jdbc:ydb:")
+                        .append(ydb.useTls() ? "grpcs://" : "grpc://")
+                        .append(ydb.endpoint())
+                        .append(ydb.database())
+                        .append("?useDiscovery=false");
+
+                if (ydb.authToken() != null) {
+                    jdbc.append("&").append("token=").append(ydb.authToken());
+                }
+
+                return jdbc.toString();
+            }
+        };
+
+        abstract String getJdbcURL();
+    }
 }