summaryrefslogtreecommitdiff
path: root/memtable
diff options
context:
space:
mode:
authorHui Xiao <huixiao@fb.com>2021-11-05 16:12:11 -0700
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>2021-11-05 16:13:47 -0700
commit3018a3e27ecf66969439c6ad0d0f4c6c50843911 (patch)
treed987d0fe288f84c2aee8ff780025991cde75f893 /memtable
parent5237b39d2eec1587e451777e03930f99007d3a37 (diff)
Minor improvement to CacheReservationManager/WriteBufferManager/CompressionDictBuilding (#9139)
Summary: Pull Request resolved: https://github.com/facebook/rocksdb/pull/9139 Reviewed By: zhichao-cao Differential Revision: D32211415 Pulled By: hx235 fbshipit-source-id: 39ce036ba34e1fb4a1992a33ac6904a4a943301d
Diffstat (limited to 'memtable')
-rw-r--r--memtable/write_buffer_manager.cc24
1 files changed, 12 insertions, 12 deletions
diff --git a/memtable/write_buffer_manager.cc b/memtable/write_buffer_manager.cc
index 48278aaaf..a2a2a340c 100644
--- a/memtable/write_buffer_manager.cc
+++ b/memtable/write_buffer_manager.cc
@@ -23,7 +23,7 @@ WriteBufferManager::WriteBufferManager(size_t _buffer_size,
mutable_limit_(buffer_size_ * 7 / 8),
memory_used_(0),
memory_active_(0),
- cache_rev_mng_(nullptr),
+ cache_res_mgr_(nullptr),
allow_stall_(allow_stall),
stall_active_(false) {
#ifndef ROCKSDB_LITE
@@ -31,7 +31,7 @@ WriteBufferManager::WriteBufferManager(size_t _buffer_size,
// Memtable's memory usage tends to fluctuate frequently
// therefore we set delayed_decrease = true to save some dummy entry
// insertion on memory increase right after memory decrease
- cache_rev_mng_.reset(
+ cache_res_mgr_.reset(
new CacheReservationManager(cache, true /* delayed_decrease */));
}
#else
@@ -47,15 +47,15 @@ WriteBufferManager::~WriteBufferManager() {
}
std::size_t WriteBufferManager::dummy_entries_in_cache_usage() const {
- if (cache_rev_mng_ != nullptr) {
- return cache_rev_mng_->GetTotalReservedCacheSize();
+ if (cache_res_mgr_ != nullptr) {
+ return cache_res_mgr_->GetTotalReservedCacheSize();
} else {
return 0;
}
}
void WriteBufferManager::ReserveMem(size_t mem) {
- if (cache_rev_mng_ != nullptr) {
+ if (cache_res_mgr_ != nullptr) {
ReserveMemWithCache(mem);
} else if (enabled()) {
memory_used_.fetch_add(mem, std::memory_order_relaxed);
@@ -68,15 +68,15 @@ void WriteBufferManager::ReserveMem(size_t mem) {
// Should only be called from write thread
void WriteBufferManager::ReserveMemWithCache(size_t mem) {
#ifndef ROCKSDB_LITE
- assert(cache_rev_mng_ != nullptr);
+ assert(cache_res_mgr_ != nullptr);
// Use a mutex to protect various data structures. Can be optimized to a
// lock-free solution if it ends up with a performance bottleneck.
- std::lock_guard<std::mutex> lock(cache_rev_mng_mu_);
+ std::lock_guard<std::mutex> lock(cache_res_mgr_mu_);
size_t new_mem_used = memory_used_.load(std::memory_order_relaxed) + mem;
memory_used_.store(new_mem_used, std::memory_order_relaxed);
Status s =
- cache_rev_mng_->UpdateCacheReservation<CacheEntryRole::kWriteBuffer>(
+ cache_res_mgr_->UpdateCacheReservation<CacheEntryRole::kWriteBuffer>(
new_mem_used);
// We absorb the error since WriteBufferManager is not able to handle
@@ -97,7 +97,7 @@ void WriteBufferManager::ScheduleFreeMem(size_t mem) {
}
void WriteBufferManager::FreeMem(size_t mem) {
- if (cache_rev_mng_ != nullptr) {
+ if (cache_res_mgr_ != nullptr) {
FreeMemWithCache(mem);
} else if (enabled()) {
memory_used_.fetch_sub(mem, std::memory_order_relaxed);
@@ -108,14 +108,14 @@ void WriteBufferManager::FreeMem(size_t mem) {
void WriteBufferManager::FreeMemWithCache(size_t mem) {
#ifndef ROCKSDB_LITE
- assert(cache_rev_mng_ != nullptr);
+ assert(cache_res_mgr_ != nullptr);
// Use a mutex to protect various data structures. Can be optimized to a
// lock-free solution if it ends up with a performance bottleneck.
- std::lock_guard<std::mutex> lock(cache_rev_mng_mu_);
+ std::lock_guard<std::mutex> lock(cache_res_mgr_mu_);
size_t new_mem_used = memory_used_.load(std::memory_order_relaxed) - mem;
memory_used_.store(new_mem_used, std::memory_order_relaxed);
Status s =
- cache_rev_mng_->UpdateCacheReservation<CacheEntryRole::kWriteBuffer>(
+ cache_res_mgr_->UpdateCacheReservation<CacheEntryRole::kWriteBuffer>(
new_mem_used);
// We absorb the error since WriteBufferManager is not able to handle