Added property to force scans and bulks

Author: alex268

jdbc/src/main/java/tech/ydb/jdbc/settings/YdbConfig.java

@@ -177,6 +177,8 @@ public DriverPropertyInfo[] toPropertyInfo() throws SQLException {
             YdbOperationProperties.TRANSACTION_LEVEL.toInfo(properties),
             YdbOperationProperties.SCHEME_QUERY_TX_MODE.toInfo(properties),
             YdbOperationProperties.SCAN_QUERY_TX_MODE.toInfo(properties),
+            YdbOperationProperties.BULK_QUERY_TX_MODE.toInfo(properties),
+            YdbOperationProperties.FORCE_SCAN_BULKS.toInfo(properties),
 
             YdbQueryProperties.DISABLE_PREPARE_DATAQUERY.toInfo(properties),
             YdbQueryProperties.DISABLE_AUTO_PREPARED_BATCHES.toInfo(properties),

jdbc/src/main/java/tech/ydb/jdbc/settings/YdbOperationProperties.java

@@ -53,6 +53,11 @@ public class YdbOperationProperties {
             FakeTxMode.ERROR
     );
 
+    static final YdbProperty<Boolean> FORCE_SCAN_BULKS = YdbProperty.bool("forceScanAndBulk",
+            "Force usage of bulk upserts instead of upserts/inserts and scan query for selects",
+            false
+    );
+
     private static final int MAX_ROWS = 1000; // TODO: how to figure out the max rows of current connection?
 
     private final YdbValue<Duration> joinDuration;
@@ -68,6 +73,8 @@ public class YdbOperationProperties {
     private final YdbValue<FakeTxMode> schemeQueryTxMode;
     private final YdbValue<FakeTxMode> bulkQueryTxMode;
 
+    private final YdbValue<Boolean> forceScanBulks;
+
     public YdbOperationProperties(YdbConfig config) throws SQLException {
         Properties props = config.getProperties();
 
@@ -83,6 +90,8 @@ public YdbOperationProperties(YdbConfig config) throws SQLException {
         this.scanQueryTxMode = SCAN_QUERY_TX_MODE.readValue(props);
         this.schemeQueryTxMode = SCHEME_QUERY_TX_MODE.readValue(props);
         this.bulkQueryTxMode = BULK_QUERY_TX_MODE.readValue(props);
+
+        this.forceScanBulks = FORCE_SCAN_BULKS.readValue(props);
     }
 
     public Duration getJoinDuration() {
@@ -113,6 +122,10 @@ public FakeTxMode getBulkQueryTxMode() {
         return bulkQueryTxMode.getValue();
     }
 
+    public boolean getForceScanBulks() {
+        return forceScanBulks.getValue();
+    }
+
     public Duration getSessionTimeout() {
         return sessionTimeout.getValue();
     }

jdbc/src/test/java/tech/ydb/jdbc/settings/YdbDriverProperitesTest.java

@@ -328,6 +328,8 @@ static DriverPropertyInfo[] defaultPropertyInfo(@Nullable String localDatacenter
             new DriverPropertyInfo("transactionLevel", "8"),
             new DriverPropertyInfo("schemeQueryTxMode", "ERROR"),
             new DriverPropertyInfo("scanQueryTxMode", "ERROR"),
+            new DriverPropertyInfo("bulkUpsertQueryTxMode", "ERROR"),
+            new DriverPropertyInfo("forceScanAndBulk", "false"),
             new DriverPropertyInfo("disablePrepareDataQuery", "false"),
             new DriverPropertyInfo("disableAutoPreparedBatches", "false"),
             new DriverPropertyInfo("disableDetectSqlOperations", "false"),
@@ -365,6 +367,8 @@ static DriverPropertyInfo[] customizedPropertyInfo() {
             new DriverPropertyInfo("transactionLevel", "16"),
             new DriverPropertyInfo("schemeQueryTxMode", "SHADOW_COMMIT"),
             new DriverPropertyInfo("scanQueryTxMode", "FAKE_TX"),
+            new DriverPropertyInfo("bulkUpsertQueryTxMode", "SHADOW_COMMIT"),
+            new DriverPropertyInfo("forceScanAndBulk", "true"),
             new DriverPropertyInfo("disablePrepareDataQuery", "true"),
             new DriverPropertyInfo("disableAutoPreparedBatches", "true"),
             new DriverPropertyInfo("disableDetectSqlOperations", "true"),