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

1 files changed, 6 insertions, 12 deletions

diff --git a/yt/cpp/mapreduce/http_client/raw_client.cpp b/yt/cpp/mapreduce/http_client/raw_client.cpp
index 8d896f93e45..010e0a47f42 100644
--- a/yt/cpp/mapreduce/http_client/raw_client.cpp
+++ b/yt/cpp/mapreduce/http_client/raw_client.cpp
@@ -931,20 +931,17 @@ T DeserializeStartWriteSessionResponse(TNode node)
 
 TDistributedWriteTableSessionWithCookies THttpRawClient::StartDistributedWriteTableSession(
     TMutationId& mutationId,
+    const TTransactionId& transactionId,
     const TRichYPath& richPath,
     i64 cookieCount,
-    const TStartDistributedWriteTableOptions& /*options*/)
+    const TStartDistributedWriteTableOptions& options)
 {
     // NB(achains): C++ client by default uses v3 api while v4 is not fully supported.
     // Explicit command path is needed until v4 is not default version.
     THttpHeader header("GET", "api/v4/start_distributed_write_session", /*isApi*/ false);
     header.AddMutationId();
 
-    TNode parameters;
-    parameters["path"] = PathToNode(richPath);
-    parameters["cookie_count"] = cookieCount;
-
-    header.MergeParameters(parameters);
+    header.MergeParameters(NRawClient::SerializeParamsForStartDistributedTableSession(transactionId, richPath, cookieCount, options));
 
     auto responseInfo = RequestWithoutRetry(Context_, mutationId, header);
 
@@ -1014,20 +1011,17 @@ std::unique_ptr<IOutputStreamWithResponse> THttpRawClient::WriteTableFragment(
 
 TDistributedWriteFileSessionWithCookies THttpRawClient::StartDistributedWriteFileSession(
     TMutationId& mutationId,
+    const TTransactionId& transactionId,
     const TRichYPath& richPath,
     i64 cookieCount,
-    const TStartDistributedWriteFileOptions& /*options*/)
+    const TStartDistributedWriteFileOptions& options)
 {
     // NB(achains): C++ client by default uses v3 api while v4 is not fully supported.
     // Explicit command path is needed until v4 is not default version.
     THttpHeader header("GET", "api/v4/start_distributed_write_file_session", /*isApi*/ false);
     header.AddMutationId();
 
-    TNode parameters;
-    parameters["path"] = PathToNode(richPath);
-    parameters["cookie_count"] = cookieCount;
-
-    header.MergeParameters(parameters);
+    header.MergeParameters(NRawClient::SerializeParamsForStartDistributedFileSession(transactionId, richPath, cookieCount, options));
 
     auto responseInfo = RequestWithoutRetry(Context_, mutationId, header);