use hash for better transaction table partitioning

Author: zinal

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

@@ -31,10 +31,14 @@ public QueryServiceExecutorExt(YdbContext ctx, int transactionLevel, boolean aut
     }
 
     private Status upsertAndCommit(QueryTransaction localTx) {
-        String sql = "DECLARE $trans_id AS Text; "
+        String sql = "DECLARE $trans_hash AS Int32; DECLARE $trans_id AS Text; "
                 + "UPSERT INTO `" + processUndeterminedTable
-                + "` (trans_id, trans_tv) VALUES ($trans_id, CurrentUtcTimestamp());";
-        Params params = Params.of("$trans_id", PrimitiveValue.newText(localTx.getId()));
+                + "` (trans_hash, trans_id, trans_tv) "
+                + "VALUES ($trans_hash, $trans_id, CurrentUtcTimestamp());";
+        Params params = Params.of(
+                "$trans_id", PrimitiveValue.newText(localTx.getId()),
+                "$trans_hash", PrimitiveValue.newInt32(localTx.getId().hashCode())
+        );
         return localTx.createQueryWithCommit(sql, params)
                 .execute()
                 .join()
@@ -43,10 +47,13 @@ private Status upsertAndCommit(QueryTransaction localTx) {
 
     private boolean checkTransaction(YdbContext ctx, String transId,
             YdbValidator validator, YdbTracer tracer) throws SQLException {
-        String sql = "DECLARE $trans_id AS Text; "
+        String sql = "DECLARE $trans_hash AS Int32; DECLARE $trans_id AS Text; "
                 + "SELECT trans_id, trans_tv FROM `" + processUndeterminedTable
-                + "` WHERE trans_id=$trans_id;";
-        Params params = Params.of("$trans_id", PrimitiveValue.newText(transId));
+                + "` WHERE trans_hash=$trans_hash AND trans_id=$trans_id;";
+        Params params = Params.of(
+                "$trans_id", PrimitiveValue.newText(transId),
+                "$trans_hash", PrimitiveValue.newInt32(transId.hashCode())
+        );
         Result<DataQueryResult> result = ctx.getRetryCtx().supplyResult(
                 session -> session.executeDataQuery(sql, TxControl.onlineRo(), params))
                 .join();

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

@@ -355,11 +355,13 @@ public void ensureTransactionTableExists() throws SQLException {
         }
         LOGGER.log(Level.FINE, "Using table {} for UNDETERMINED processing", tableName);
         String sqlCreate = "CREATE TABLE IF NOT EXISTS `" + tableName
-                + "` (trans_id Text NOT NULL, trans_tv Timestamp,"
-                + "   PRIMARY KEY (trans_id)) WITH ("
+                + "` (trans_hash Int32 NOT NULL, trans_id Text NOT NULL, trans_tv Timestamp,"
+                + "   PRIMARY KEY (trans_hash, trans_id)) WITH ("
                 + "TTL=Interval('PT60M') ON trans_tv,"
+                /*
                 + "AUTO_PARTITIONING_MIN_PARTITIONS_COUNT=100,"
                 + "AUTO_PARTITIONING_MAX_PARTITIONS_COUNT=150,"
+                */
                 + "AUTO_PARTITIONING_BY_LOAD=ENABLED,"
                 + "AUTO_PARTITIONING_BY_SIZE=ENABLED,"
                 + "AUTO_PARTITIONING_PARTITION_SIZE_MB=100"