mirror of
https://github.com/facebook/rocksdb.git
synced 2024-11-26 16:30:56 +00:00
1543d5d92e
Summary: It is hard to measure acutal memory usage by std containers. Even providing a custom allocator will miss count some of the usage. Here we only do a wild guess on its memory usage. Closes https://github.com/facebook/rocksdb/pull/1511 Differential Revision: D4179945 Pulled By: yiwu-arbug fbshipit-source-id: 32ab929
26 lines
882 B
C++
26 lines
882 B
C++
// 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 <unordered_map>
|
|
|
|
namespace rocksdb {
|
|
|
|
// Helper methods to estimate memroy usage by std containers.
|
|
|
|
template <class Key, class Value, class Hash>
|
|
size_t ApproximateMemoryUsage(
|
|
const std::unordered_map<Key, Value, Hash>& umap) {
|
|
typedef std::unordered_map<Key, Value, Hash> 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
|