diff options
author | don-dron <don-dron@yandex-team.com> | 2024-07-02 12:57:53 +0300 |
---|---|---|
committer | don-dron <don-dron@yandex-team.com> | 2024-07-02 13:08:32 +0300 |
commit | dd2d8a66405a840578159546cbf0e9c3f4dd6350 (patch) | |
tree | 011993f401aa7526958759bfacdcd81d4de1473b | |
parent | 77685066e8c73adcf14993f5272e2600c6d3c7bd (diff) | |
download | ydb-dd2d8a66405a840578159546cbf0e9c3f4dd6350.tar.gz |
YT-21931: Track memory for ahead blocks in write sessions
0b4b67710dab4d7b83e11414d549bafda875db40
-rw-r--r-- | yt/yt/core/misc/memory_usage_tracker.cpp | 7 | ||||
-rw-r--r-- | yt/yt/core/misc/memory_usage_tracker.h | 3 |
2 files changed, 8 insertions, 2 deletions
diff --git a/yt/yt/core/misc/memory_usage_tracker.cpp b/yt/yt/core/misc/memory_usage_tracker.cpp index 7d100e8254..cd1176ee06 100644 --- a/yt/yt/core/misc/memory_usage_tracker.cpp +++ b/yt/yt/core/misc/memory_usage_tracker.cpp @@ -226,11 +226,16 @@ TError TMemoryUsageTrackerGuard::SetSizeGeneric(i64 size, auto acquirer) return {}; } -void TMemoryUsageTrackerGuard::IncrementSize(i64 sizeDelta) +void TMemoryUsageTrackerGuard::IncreaseSize(i64 sizeDelta) { SetSize(Size_ + sizeDelta); } +void TMemoryUsageTrackerGuard::DecreaseSize(i64 sizeDelta) +{ + SetSize(Size_ - sizeDelta); +} + TMemoryUsageTrackerGuard TMemoryUsageTrackerGuard::TransferMemory(i64 size) { YT_VERIFY(Size_ >= size); diff --git a/yt/yt/core/misc/memory_usage_tracker.h b/yt/yt/core/misc/memory_usage_tracker.h index 74957dfb00..cacc9517c8 100644 --- a/yt/yt/core/misc/memory_usage_tracker.h +++ b/yt/yt/core/misc/memory_usage_tracker.h @@ -79,7 +79,8 @@ public: i64 GetSize() const; void SetSize(i64 size); TError TrySetSize(i64 size); - void IncrementSize(i64 sizeDelta); + void IncreaseSize(i64 sizeDelta); + void DecreaseSize(i64 sizeDelta); TMemoryUsageTrackerGuard TransferMemory(i64 size); private: |