YT-21942: Adjustments to adaptive row count feature

First ver
commit_hash:2463cc9c403515c1e5605fdcfcecc0f7318b4b70

1 files changed, 4 insertions, 1 deletions

diff --git a/yt/yt/client/table_client/adapters.cpp b/yt/yt/client/table_client/adapters.cpp
index 35b21ce8cf..648a208ea4 100644
--- a/yt/yt/client/table_client/adapters.cpp
+++ b/yt/yt/client/table_client/adapters.cpp
@@ -212,7 +212,9 @@ void PipeReaderToWriterByBatches(
                 continue;
             }
 
-            if (!batch->IsEmpty() && pipeDelay != TDuration::Zero()) {
+            auto rowsRead = batch->GetRowCount();
+
+            if (pipeDelay != TDuration::Zero()) {
                 TDelayedExecutor::WaitForDuration(pipeDelay);
             }
 
@@ -228,6 +230,7 @@ void PipeReaderToWriterByBatches(
             }
 
             if (optionsUpdater) {
+                options.MaxRowsPerRead = rowsRead;
                 optionsUpdater(&options, timer.GetElapsedTime());
             }
         }