YT-20969: C++ methods for reading table partitions

* Changelog entry
Type: feature
Component: cpp-mapreduce-sdk

Introduce methods for reading table partition
commit_hash:ab818edd21ccb2a9c5f3d5f010686c3314532192

1 files changed, 18 insertions, 0 deletions

diff --git a/yt/cpp/mapreduce/http_client/raw_client.cpp b/yt/cpp/mapreduce/http_client/raw_client.cpp
index 734282c2cf5..ab8fbf5821c 100644
--- a/yt/cpp/mapreduce/http_client/raw_client.cpp
+++ b/yt/cpp/mapreduce/http_client/raw_client.cpp
@@ -755,6 +755,24 @@ std::unique_ptr<IInputStream> THttpRawClient::ReadTable(
     return std::make_unique<NHttpClient::THttpResponseStream>(std::move(responseInfo));
 }
 
+std::unique_ptr<IInputStream> THttpRawClient::ReadTablePartition(
+    const TString& cookie,
+    const TMaybe<TFormat>& format,
+    const TTablePartitionReaderOptions& options)
+{
+    TMutationId mutationId;
+    THttpHeader header("GET", "api/v4/read_table_partition", /*isApi*/ false);
+    header.SetOutputFormat(format);
+    header.SetResponseCompression(ToString(Context_.Config->AcceptEncoding));
+    auto params = NRawClient::SerializeParamsForReadTablePartition(cookie, options);
+    header.MergeParameters(params);
+
+    TRequestConfig config;
+    config.IsHeavy = true;
+    auto responseInfo = RequestWithoutRetry(Context_, mutationId, header, /*body*/ {}, config);
+    return std::make_unique<NHttpClient::THttpResponseStream>(std::move(responseInfo));
+}
+
 std::unique_ptr<IInputStream> THttpRawClient::ReadBlobTable(
     const TTransactionId& transactionId,
     const TRichYPath& path,