YT-26906: start session from tx

* Changelog entry
  Type: feature
  Component: cpp-sdk

Start distributed session methods (file/table) now support attaching to transaction.
commit_hash:0a40dfd6d556f9890fa5abccf29c0baf33df4e7d

2 files changed, 20 insertions, 16 deletions

diff --git a/yt/cpp/mapreduce/interface/client.h b/yt/cpp/mapreduce/interface/client.h
index e5032e10970..27c248afc1e 100644
--- a/yt/cpp/mapreduce/interface/client.h
+++ b/yt/cpp/mapreduce/interface/client.h
@@ -150,6 +150,24 @@ public:
         const TStartTransactionOptions& options = TStartTransactionOptions()) = 0;
 
     ///
+    /// @brief Initialize distributed file session and request cookies for participants.
+    ///
+    /// @see [YT doc](https://ytsaurus.tech/docs/en/api/commands#start_distributed_write_file_session)
+    virtual TDistributedWriteFileSessionWithCookies StartDistributedWriteFileSession(
+        const TRichYPath& richPath,
+        i64 cookieCount,
+        const TStartDistributedWriteFileOptions& options = {}) = 0;
+
+    ///
+    /// @brief Initialize distributed table session and request cookies for participants.
+    ///
+    /// @see [YT doc](https://ytsaurus.tech/docs/en/api/commands.html#start_distributed_write_session)
+    virtual TDistributedWriteTableSessionWithCookies StartDistributedWriteTableSession(
+        const TRichYPath& richPath,
+        i64 cookieCount,
+        const TStartDistributedWriteTableOptions& options = {}) = 0;
+
+    ///
     /// @brief Change properties of table.
     ///
     /// Allows to:
@@ -580,14 +598,6 @@ public:
     /// Distributed table write API
     /// @{
 
-    /// @brief Initialize distributed session and request cookies for participants.
-    ///
-    /// @see [YT doc](https://ytsaurus.tech/docs/en/api/commands.html#start_distributed_write_session)
-    virtual TDistributedWriteTableSessionWithCookies StartDistributedWriteTableSession(
-        const TRichYPath& richPath,
-        i64 cookieCount,
-        const TStartDistributedWriteTableOptions& options = {}) = 0;
-
     /// @brief Ping distributed session to prolong its main transaction lifetime.
     ///
     /// @see [YT doc](https://ytsaurus.tech/docs/en/api/commands.html#ping_distributed_write_session)
@@ -609,14 +619,6 @@ public:
     /// Distributed file write API
     /// @{
 
-    /// @brief Initialize distributed session and request cookies for participants.
-    ///
-    /// @see [YT doc](https://ytsaurus.tech/docs/en/api/commands.html#start_distributed_write_file_session)
-    virtual TDistributedWriteFileSessionWithCookies StartDistributedWriteFileSession(
-        const TRichYPath& richPath,
-        i64 cookieCount,
-        const TStartDistributedWriteFileOptions& options = {}) = 0;
-
     /// @brief Ping distributed session to prolong its main transaction lifetime.
     ///
     /// @see [YT doc](https://ytsaurus.tech/docs/en/api/commands.html#ping_distributed_write_file_session)
diff --git a/yt/cpp/mapreduce/interface/raw_client.h b/yt/cpp/mapreduce/interface/raw_client.h
index 23b311e457a..0c451951a07 100644
--- a/yt/cpp/mapreduce/interface/raw_client.h
+++ b/yt/cpp/mapreduce/interface/raw_client.h
@@ -338,6 +338,7 @@ public:
 
     virtual TDistributedWriteTableSessionWithCookies StartDistributedWriteTableSession(
         TMutationId& mutationId,
+        const TTransactionId& transactionId,
         const TRichYPath& richPath,
         i64 cookieCount,
         const TStartDistributedWriteTableOptions& options = {}) = 0;
@@ -359,6 +360,7 @@ public:
 
     virtual TDistributedWriteFileSessionWithCookies StartDistributedWriteFileSession(
         TMutationId& mutationId,
+        const TTransactionId& transactionId,
         const TRichYPath& richPath,
         i64 cookieCount,
         const TStartDistributedWriteFileOptions& options = {}) = 0;