Browse Source

BufferMemory usage for retryful writer v2
ca66c0e73475f3596f6fb2e4e655f2ec378a71a9

ermolovd 11 months ago
parent
commit
68bc6e9759

+ 5 - 0
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_) {

+ 2 - 0
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;