From e7e727407f325ed2527d4a14a767b40ba4905113 Mon Sep 17 00:00:00 2001
From: Nikolai Amelichev <entropia@nebius.com>
Date: Tue, 4 Jun 2024 20:46:41 +0200
Subject: [PATCH] Update YDB SDK to v2.2.2

---
 pom.xml                                       |  4 +--
 .../ydb/client/QueryInterceptingSession.java  | 29 ++++++++++++++++++-
 2 files changed, 30 insertions(+), 3 deletions(-)

diff --git a/pom.xml b/pom.xml
index 6c2c2679..411d1e84 100644
--- a/pom.xml
+++ b/pom.xml
@@ -109,8 +109,8 @@
         <os-maven-plugin.version>1.7.1</os-maven-plugin.version>
 
         <!-- YDB SDK 2.x -->
-        <ydb-sdk-v2.version>2.1.10</ydb-sdk-v2.version>
-        <ydb-proto-api.version>1.6.0</ydb-proto-api.version>
+        <ydb-sdk-v2.version>2.2.2</ydb-sdk-v2.version>
+        <ydb-proto-api.version>1.6.2</ydb-proto-api.version>
 
         <!-- build-only dependencies (provided) -->
         <lombok.version>1.18.30</lombok.version>
diff --git a/repository-ydb-v2/src/main/java/tech/ydb/yoj/repository/ydb/client/QueryInterceptingSession.java b/repository-ydb-v2/src/main/java/tech/ydb/yoj/repository/ydb/client/QueryInterceptingSession.java
index 49a047c5..996cc9ae 100644
--- a/repository-ydb-v2/src/main/java/tech/ydb/yoj/repository/ydb/client/QueryInterceptingSession.java
+++ b/repository-ydb-v2/src/main/java/tech/ydb/yoj/repository/ydb/client/QueryInterceptingSession.java
@@ -1,5 +1,6 @@
 package tech.ydb.yoj.repository.ydb.client;
 
+import tech.ydb.common.transaction.TxMode;
 import tech.ydb.core.Result;
 import tech.ydb.core.Status;
 import tech.ydb.core.grpc.GrpcReadStream;
@@ -9,6 +10,7 @@
 import tech.ydb.table.query.DataQueryResult;
 import tech.ydb.table.query.ExplainDataQueryResult;
 import tech.ydb.table.query.Params;
+import tech.ydb.table.query.ReadRowsResult;
 import tech.ydb.table.query.ReadTablePart;
 import tech.ydb.table.result.ResultSetReader;
 import tech.ydb.table.settings.AlterTableSettings;
@@ -26,8 +28,11 @@
 import tech.ydb.table.settings.ExplainDataQuerySettings;
 import tech.ydb.table.settings.KeepAliveSessionSettings;
 import tech.ydb.table.settings.PrepareDataQuerySettings;
+import tech.ydb.table.settings.ReadRowsSettings;
 import tech.ydb.table.settings.ReadTableSettings;
+import tech.ydb.table.settings.RenameTablesSettings;
 import tech.ydb.table.settings.RollbackTxSettings;
+import tech.ydb.table.transaction.TableTransaction;
 import tech.ydb.table.transaction.Transaction;
 import tech.ydb.table.transaction.TxControl;
 import tech.ydb.table.values.ListValue;
@@ -45,7 +50,6 @@ public final class QueryInterceptingSession implements Session {
     private final Session delegate;
     private final QueryInterceptor interceptor;
 
-
     private QueryInterceptingSession(Session delegate, QueryInterceptor interceptor) {
         this.delegate = delegate;
         this.interceptor = interceptor;
@@ -162,6 +166,29 @@ public void close() {
         delegate.close();
     }
 
+    //////////////////////////
+    // @since: YDB SDK v2.2.0:
+
+    @Override
+    public CompletableFuture<Status> renameTables(RenameTablesSettings renameTablesSettings) {
+        return delegate.renameTables(renameTablesSettings);
+    }
+
+    @Override
+    public CompletableFuture<Result<ReadRowsResult>> readRows(String s, ReadRowsSettings readRowsSettings) {
+        return delegate.readRows(s, readRowsSettings);
+    }
+
+    @Override
+    public TableTransaction createNewTransaction(TxMode txMode) {
+        return delegate.createNewTransaction(txMode);
+    }
+
+    @Override
+    public CompletableFuture<Result<TableTransaction>> beginTransaction(TxMode txMode, BeginTxSettings beginTxSettings) {
+        return delegate.beginTransaction(txMode, beginTxSettings);
+    }
+
     public enum QueryType {
         DATA_QUERY,
         SCAN_QUERY,