Merge pull request #20 from ydb-platform/auto_detect_query_type

Auto detect query type

Author: alex268

jdbc/src/main/java/tech/ydb/jdbc/YdbConnection.java

@@ -5,7 +5,7 @@
 
 import javax.annotation.Nullable;
 
-import tech.ydb.jdbc.context.YdbQuery;
+import tech.ydb.jdbc.query.YdbQuery;
 import tech.ydb.jdbc.context.YdbContext;
 import tech.ydb.jdbc.context.YdbExecutor;
 import tech.ydb.table.query.DataQueryResult;

jdbc/src/main/java/tech/ydb/jdbc/YdbConst.java

@@ -89,12 +89,8 @@ public final class YdbConst {
     public static final String CANNOT_LOAD_DATA_FROM_IS = "Unable to load data from input stream: ";
     public static final String CANNOT_LOAD_DATA_FROM_READER = "Unable to load data from reader: ";
     public static final String UNSUPPORTED_QUERY_TYPE_IN_PS = "Query type in prepared statement not supported: ";
-    public static final String UNKNOWN_PARAMETER_IN_BATCH = "New parameter when batch already added: ";
     public static final String STATEMENT_IS_NOT_A_BATCH = "Statement cannot be executed as batch statement: ";
-    public static final String TRY_EXECUTE_ON_BATCH_STATEMENT = "Cannot call #execute method after #addBatch, " +
-            "must use #executeBatch";
-    public static final String INDEXES_PARAMETERS_UNSUPPORTED =
-            "Indexed parameters are not supported in prepared query: ";
+    public static final String MULTI_TYPES_IN_ONE_QUERY = "Query cannot contain expressions with different types: ";
 
     // Cast errors
 
@@ -148,6 +144,7 @@ public final class YdbConst {
     // All indexed parameters will have this 'p' as a prefix, setInteger(1, "test") -> setInteger("p1", "test")
     public static final String INDEXED_PARAMETER_PREFIX = "p";
     public static final String VARIABLE_PARAMETER_PREFIX = "$";
+    public static final String AUTO_GENERATED_PARAMETER_PREFIX = VARIABLE_PARAMETER_PREFIX + "jp";
     public static final String DEFAULT_BATCH_PARAMETER = "$values";
     public static final String OPTIONAL_TYPE_SUFFIX = "?";
 

jdbc/src/main/java/tech/ydb/jdbc/common/JdbcLexer.java

@@ -10,6 +10,7 @@
 import tech.ydb.core.grpc.GrpcTransport;
 import tech.ydb.core.grpc.GrpcTransportBuilder;
 import tech.ydb.jdbc.exception.YdbConfigurationException;
+import tech.ydb.jdbc.query.YdbQueryOptions;
 import tech.ydb.jdbc.settings.ParsedProperty;
 import tech.ydb.jdbc.settings.YdbClientProperties;
 import tech.ydb.jdbc.settings.YdbClientProperty;
@@ -39,15 +40,17 @@ public class YdbContext implements AutoCloseable {
     private final GrpcTransport grpcTransport;
     private final PooledTableClient tableClient;
     private final SchemeClient schemeClient;
-    private final boolean autoResizeSessionPool;
+    private final YdbQueryOptions queryOptions;
 
+    private final boolean autoResizeSessionPool;
     private final AtomicInteger connectionsCount = new AtomicInteger();
 
     private YdbContext(YdbConfig config, GrpcTransport transport, PooledTableClient tableClient, boolean autoResize) {
         this.config = config;
         this.grpcTransport = Objects.requireNonNull(transport);
         this.tableClient = Objects.requireNonNull(tableClient);
         this.schemeClient = SchemeClient.newClient(transport).build();
+        this.queryOptions = new YdbQueryOptions(config.getOperationProperties());
         this.autoResizeSessionPool = autoResize;
     }
 
@@ -67,6 +70,10 @@ public String getUrl() {
         return config.getUrl();
     }
 
+    public YdbQueryOptions getQueryOptions() {
+        return queryOptions;
+    }
+
     public int getConnectionsCount() {
         return connectionsCount.get();
     }