BufferMemory usage for retryful writer v2

ca66c0e73475f3596f6fb2e4e655f2ec378a71a9

2 files changed, 7 insertions, 0 deletions

diff --git a/yt/cpp/mapreduce/client/retryful_writer_v2.cpp b/yt/cpp/mapreduce/client/retryful_writer_v2.cpp
index cabc6cf8fa7..40297500ae5 100644
--- a/yt/cpp/mapreduce/client/retryful_writer_v2.cpp
+++ b/yt/cpp/mapreduce/client/retryful_writer_v2.cpp
@@ -338,6 +338,11 @@ void TRetryfulWriterV2::Abort()
     }
 }
 
+size_t TRetryfulWriterV2::GetBufferMemoryUsage() const
+{
+    return BufferSize_ * 4;
+}
+
 void TRetryfulWriterV2::DoFinish()
 {
     if (Sender_) {
diff --git a/yt/cpp/mapreduce/client/retryful_writer_v2.h b/yt/cpp/mapreduce/client/retryful_writer_v2.h
index c344fd0c6c8..bda55d96a60 100644
--- a/yt/cpp/mapreduce/client/retryful_writer_v2.h
+++ b/yt/cpp/mapreduce/client/retryful_writer_v2.h
@@ -32,6 +32,8 @@ public:
     void NotifyRowEnd() override;
     void Abort() override;
 
+    size_t GetBufferMemoryUsage() const override;
+
 protected:
     void DoWrite(const void* buf, size_t len) override;
     void DoFinish() override;