YT-24267: add probe put blocks API interface

commit_hash:6c17cb04a6f1f5619cf06502eda433c7588b35d7

2 files changed, 11 insertions, 0 deletions

diff --git a/yt/yt/client/chunk_client/config.cpp b/yt/yt/client/chunk_client/config.cpp
index 5d7a05e091..6c367cdafe 100644
--- a/yt/yt/client/chunk_client/config.cpp
+++ b/yt/yt/client/chunk_client/config.cpp
@@ -268,6 +268,8 @@ void TReplicationWriterConfig::Register(TRegistrar registrar)
         .DefaultNew();
     registrar.Parameter("node_rpc_timeout", &TThis::NodeRpcTimeout)
         .Default(TDuration::Seconds(300));
+    registrar.Parameter("probe_put_blocks_timeout", &TThis::ProbePutBlocksTimeout)
+        .Default(TDuration::Seconds(60));
     registrar.Parameter("upload_replication_factor", &TThis::UploadReplicationFactor)
         .GreaterThanOrEqual(1)
         .Default(2);
@@ -301,6 +303,9 @@ void TReplicationWriterConfig::Register(TRegistrar registrar)
     registrar.Parameter("enable_local_throttling", &TThis::EnableLocalThrottling)
         .Default(false);
 
+    registrar.Parameter("use_probe_put_blocks", &TThis::UseProbePutBlocks)
+        .Default(false);
+
     registrar.Preprocessor([] (TThis* config) {
         config->NodeChannel->RetryBackoffTime = TDuration::Seconds(10);
         config->NodeChannel->RetryAttempts = 100;
diff --git a/yt/yt/client/chunk_client/config.h b/yt/yt/client/chunk_client/config.h
index d703c2da3a..ad17f5f757 100644
--- a/yt/yt/client/chunk_client/config.h
+++ b/yt/yt/client/chunk_client/config.h
@@ -331,6 +331,8 @@ struct TReplicationWriterConfig
      */
     TDuration NodeRpcTimeout;
 
+    TDuration ProbePutBlocksTimeout;
+
     NRpc::TRetryingChannelConfigPtr NodeChannel;
 
     int UploadReplicationFactor;
@@ -366,6 +368,10 @@ struct TReplicationWriterConfig
     //! If |true| network throttlers will be applied even in case of requests to local host.
     bool EnableLocalThrottling;
 
+    //! Enable write protocol with probe put blocks.
+    //! Acquiring resources for putting blocks before invoking PutBlocks.
+    bool UseProbePutBlocks;
+
     int GetDirectUploadNodeCount();
 
     REGISTER_YSON_STRUCT(TReplicationWriterConfig);