diff options
| author | ermolovd <ermolovd@yandex-team.com> | 2024-04-12 21:38:02 +0300 |
|---|---|---|
| committer | ermolovd <ermolovd@yandex-team.com> | 2024-04-13 00:43:22 +0300 |
| commit | 68bc6e97593f4c41266b2ddca54eff8d23416fee (patch) | |
| tree | 2db6ceb72132062d0f9335a8b81dab81ae19993d /yt | |
| parent | 39d2b82b888f5a640099579def110a7066555ee6 (diff) | |
| download | ydb-68bc6e97593f4c41266b2ddca54eff8d23416fee.tar.gz | |
BufferMemory usage for retryful writer v2
ca66c0e73475f3596f6fb2e4e655f2ec378a71a9
Diffstat (limited to 'yt')
| -rw-r--r-- | yt/cpp/mapreduce/client/retryful_writer_v2.cpp | 5 | ||||
| -rw-r--r-- | yt/cpp/mapreduce/client/retryful_writer_v2.h | 2 |
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 cabc6cf8fa..40297500ae 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 c344fd0c6c..bda55d96a6 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; |