Added support of scheme and explain queries to PreparedStatement

Author: alex268

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

@@ -86,7 +86,6 @@ public final class YdbConst {
     public static final String DATABASE_UNAVAILABLE = "Database is unavailable: ";
     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 STATEMENT_IS_NOT_A_BATCH = "Statement cannot be executed as batch statement: ";
     public static final String MULTI_TYPES_IN_ONE_QUERY = "Query cannot contain expressions with different types: ";
     public static final String SCAN_QUERY_INSIDE_TRANSACTION = "Scan query cannot be executed inside active "

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

@@ -26,6 +26,7 @@
 import tech.ydb.jdbc.YdbConst;
 import tech.ydb.jdbc.YdbPrepareMode;
 import tech.ydb.jdbc.exception.ExceptionFactory;
+import tech.ydb.jdbc.query.QueryType;
 import tech.ydb.jdbc.query.YdbPreparedQuery;
 import tech.ydb.jdbc.query.YdbQuery;
 import tech.ydb.jdbc.query.params.BatchedQuery;
@@ -345,6 +346,10 @@ public YdbPreparedQuery findOrPrepareParams(YdbQuery query, YdbPrepareMode mode)
             }
         }
 
+        if (query.getType() == QueryType.EXPLAIN_QUERY || query.getType() == QueryType.SCHEME_QUERY) {
+            return new InMemoryQuery(query, queryOptions.isDeclareJdbcParameters());
+        }
+
         if (query.getYqlBatcher() != null && mode == YdbPrepareMode.AUTO) {
             Map<String, Type> types = queryParamsCache.getIfPresent(query.getOriginQuery());
             if (types == null) {

jdbc/src/main/java/tech/ydb/jdbc/impl/YdbConnectionImpl.java

@@ -31,7 +31,6 @@
 import tech.ydb.jdbc.context.YdbExecutor;
 import tech.ydb.jdbc.context.YdbValidator;
 import tech.ydb.jdbc.query.ExplainedQuery;
-import tech.ydb.jdbc.query.QueryType;
 import tech.ydb.jdbc.query.YdbPreparedQuery;
 import tech.ydb.jdbc.query.YdbQuery;
 import tech.ydb.jdbc.settings.FakeTxMode;
@@ -307,10 +306,6 @@ private YdbPreparedStatement prepareStatement(String sql, int resultSetType, Ydb
 
         YdbQuery query = ctx.findOrParseYdbQuery(sql);
 
-        if (query.getType() != QueryType.DATA_QUERY && query.getType() != QueryType.SCAN_QUERY) {
-            throw new SQLException(YdbConst.UNSUPPORTED_QUERY_TYPE_IN_PS + query.getType());
-        }
-
         YdbPreparedQuery params = ctx.findOrPrepareParams(query, mode);
         return new YdbPreparedStatementImpl(this, query, params, resultSetType);
     }

jdbc/src/main/java/tech/ydb/jdbc/impl/YdbPreparedStatementImpl.java

@@ -132,6 +132,12 @@ public boolean execute() throws SQLException {
             case SCAN_QUERY:
                 newState = executeScanQuery(query, prepared.getQueryText(prms), prms);
                 break;
+            case SCHEME_QUERY:
+                newState = executeSchemeQuery(query);
+                break;
+            case EXPLAIN_QUERY:
+                newState = executeExplainQuery(query);
+                break;
             default:
                 throw new IllegalStateException("Internal error. Unsupported query type " + query.getType());
         }