diff --git a/db/db_properties_test.cc b/db/db_properties_test.cc index e0ad3ab9f..31afc2cfd 100644 --- a/db/db_properties_test.cc +++ b/db/db_properties_test.cc @@ -491,6 +491,7 @@ TEST_F(DBPropertiesTest, NumImmutableMemTable) { std::string big_value(1000000 * 2, 'x'); std::string num; + uint64_t value; SetPerfLevel(kEnableTime); ASSERT_TRUE(GetPerfLevel() == kEnableTime); @@ -555,11 +556,11 @@ TEST_F(DBPropertiesTest, NumImmutableMemTable) { ASSERT_TRUE(dbfull()->GetProperty( handles_[1], DB::Properties::kNumImmutableMemTableFlushed, &num)); ASSERT_EQ(num, "3"); - ASSERT_TRUE(dbfull()->GetProperty( - handles_[1], "rocksdb.cur-size-active-mem-table", &num)); + ASSERT_TRUE(dbfull()->GetIntProperty( + handles_[1], "rocksdb.cur-size-active-mem-table", &value)); // "384" is the size of the metadata of two empty skiplists, this would // break if we change the default skiplist implementation - ASSERT_EQ(num, "384"); + ASSERT_GE(value, 384); uint64_t int_num; uint64_t base_total_size; diff --git a/db/memtable.cc b/db/memtable.cc index f12365298..9e059409f 100644 --- a/db/memtable.cc +++ b/db/memtable.cc @@ -17,6 +17,7 @@ #include "db/merge_context.h" #include "db/merge_helper.h" #include "db/pinned_iterators_manager.h" +#include "port/port.h" #include "rocksdb/comparator.h" #include "rocksdb/env.h" #include "rocksdb/iterator.h" @@ -27,7 +28,9 @@ #include "table/iterator_wrapper.h" #include "table/merger.h" #include "util/arena.h" +#include "util/autovector.h" #include "util/coding.h" +#include "util/memory_usage.h" #include "util/murmurhash.h" #include "util/mutexlock.h" #include "util/perf_context_imp.h" @@ -105,17 +108,21 @@ MemTable::MemTable(const InternalKeyComparator& cmp, MemTable::~MemTable() { assert(refs_ == 0); } size_t MemTable::ApproximateMemoryUsage() { - size_t arena_usage = arena_.ApproximateMemoryUsage(); - size_t table_usage = table_->ApproximateMemoryUsage(); - table_usage += range_del_table_->ApproximateMemoryUsage(); - // let MAX_USAGE = std::numeric_limits::max() - // then if arena_usage + total_usage >= MAX_USAGE, return MAX_USAGE. - // the following variation is to avoid numeric overflow. - if (arena_usage >= std::numeric_limits::max() - table_usage) { - return std::numeric_limits::max(); + autovector usages = {arena_.ApproximateMemoryUsage(), + table_->ApproximateMemoryUsage(), + range_del_table_->ApproximateMemoryUsage(), + rocksdb::ApproximateMemoryUsage(insert_hints_)}; + size_t total_usage = 0; + for (size_t usage : usages) { + // If usage + total_usage >= kMaxSizet, return kMaxSizet. + // the following variation is to avoid numeric overflow. + if (usage >= port::kMaxSizet - total_usage) { + return port::kMaxSizet; + } + total_usage += usage; } // otherwise, return the actual usage - return arena_usage + table_usage; + return total_usage; } bool MemTable::ShouldFlushNow() const { diff --git a/util/autovector.h b/util/autovector.h index 74fcc70ec..cfe703ed9 100644 --- a/util/autovector.h +++ b/util/autovector.h @@ -6,15 +6,18 @@ #include #include -#include +#include #include +#include #include namespace rocksdb { #ifdef ROCKSDB_LITE template -class autovector : public std::vector {}; +class autovector : public std::vector { + using std::vector::vector; +}; #else // A vector that leverages pre-allocated stack-based array to achieve better // performance for array with small amount of items. @@ -165,6 +168,13 @@ class autovector { typedef std::reverse_iterator const_reverse_iterator; autovector() = default; + + autovector(std::initializer_list init_list) { + for (const T& item : init_list) { + push_back(item); + } + } + ~autovector() = default; // -- Immutable operations diff --git a/util/memory_usage.h b/util/memory_usage.h new file mode 100644 index 000000000..cfe0025fe --- /dev/null +++ b/util/memory_usage.h @@ -0,0 +1,25 @@ +// Copyright (c) 2011-present, Facebook, Inc. All rights reserved. +// This source code is licensed under the BSD-style license found in the +// LICENSE file in the root directory of this source tree. An additional grant +// of patent rights can be found in the PATENTS file in the same directory. + +#pragma once + +#include + +namespace rocksdb { + +// Helper methods to estimate memroy usage by std containers. + +template +size_t ApproximateMemoryUsage( + const std::unordered_map& umap) { + typedef std::unordered_map Map; + return sizeof(umap) + + // Size of all items plus a next pointer for each item. + (sizeof(typename Map::value_type) + sizeof(void*)) * umap.size() + + // Size of hash buckets. + umap.bucket_count() * sizeof(void*); +} + +} // namespace rocksdb