mirror of
https://github.com/facebook/rocksdb.git
synced 2024-11-30 04:41:49 +00:00
17bc27741f
Summary: Currently it's easy to use a ton of memory with many small OptimisticTransactionDB instances, because each one by default allocates a million mutexes (40 bytes each on my compiler) for validating transactions. It even puts a lot of pressure on the allocator by allocating each one individually! In this change: * Create a new object and option that enables sharing these buckets of mutexes between instances. This is generally good for load balancing potential contention as various DBs become hotter or colder with txn writes. About the only cases where this sharing wouldn't make sense (e.g. each DB usually written by one thread) are cases that would be better off with OccValidationPolicy::kValidateSerial which doesn't use the buckets anyway. * Allocate the mutexes in a contiguous array, for efficiency * Add an option to ensure the mutexes are cache-aligned. In several other places we use cache-aligned mutexes but OptimisticTransactionDB historically does not. It should be a space-time trade-off the user can choose. * Provide some visibility into the memory used by the mutex buckets with an ApproximateMemoryUsage() function (also used in unit testing) * Share code with other users of "striped" mutexes, appropriate refactoring for customization & efficiency (e.g. using FastRange instead of modulus) Pull Request resolved: https://github.com/facebook/rocksdb/pull/11439 Test Plan: unit tests added. Ran sized-up versions of stress test in unit test, including a before-and-after performance test showing no consistent difference. (NOTE: OptimisticTransactionDB not currently covered by db_stress!) Reviewed By: ltamasi Differential Revision: D45796393 Pulled By: pdillinger fbshipit-source-id: ae2b3a26ad91ceeec15debcdc63ff48df6736a54
102 lines
2.8 KiB
C++
102 lines
2.8 KiB
C++
// Copyright (c) 2011-present, Facebook, Inc. All rights reserved.
|
|
// This source code is licensed under both the GPLv2 (found in the
|
|
// COPYING file in the root directory) and Apache 2.0 License
|
|
// (found in the LICENSE.Apache file in the root directory).
|
|
|
|
#include "db/blob/blob_file_cache.h"
|
|
|
|
#include <cassert>
|
|
#include <memory>
|
|
|
|
#include "db/blob/blob_file_reader.h"
|
|
#include "options/cf_options.h"
|
|
#include "rocksdb/cache.h"
|
|
#include "rocksdb/slice.h"
|
|
#include "test_util/sync_point.h"
|
|
#include "trace_replay/io_tracer.h"
|
|
#include "util/hash.h"
|
|
|
|
namespace ROCKSDB_NAMESPACE {
|
|
|
|
BlobFileCache::BlobFileCache(Cache* cache,
|
|
const ImmutableOptions* immutable_options,
|
|
const FileOptions* file_options,
|
|
uint32_t column_family_id,
|
|
HistogramImpl* blob_file_read_hist,
|
|
const std::shared_ptr<IOTracer>& io_tracer)
|
|
: cache_(cache),
|
|
mutex_(kNumberOfMutexStripes),
|
|
immutable_options_(immutable_options),
|
|
file_options_(file_options),
|
|
column_family_id_(column_family_id),
|
|
blob_file_read_hist_(blob_file_read_hist),
|
|
io_tracer_(io_tracer) {
|
|
assert(cache_);
|
|
assert(immutable_options_);
|
|
assert(file_options_);
|
|
}
|
|
|
|
Status BlobFileCache::GetBlobFileReader(
|
|
const ReadOptions& read_options, uint64_t blob_file_number,
|
|
CacheHandleGuard<BlobFileReader>* blob_file_reader) {
|
|
assert(blob_file_reader);
|
|
assert(blob_file_reader->IsEmpty());
|
|
|
|
const Slice key = GetSliceForKey(&blob_file_number);
|
|
|
|
assert(cache_);
|
|
|
|
TypedHandle* handle = cache_.Lookup(key);
|
|
if (handle) {
|
|
*blob_file_reader = cache_.Guard(handle);
|
|
return Status::OK();
|
|
}
|
|
|
|
TEST_SYNC_POINT("BlobFileCache::GetBlobFileReader:DoubleCheck");
|
|
|
|
// Check again while holding mutex
|
|
MutexLock lock(&mutex_.Get(key));
|
|
|
|
handle = cache_.Lookup(key);
|
|
if (handle) {
|
|
*blob_file_reader = cache_.Guard(handle);
|
|
return Status::OK();
|
|
}
|
|
|
|
assert(immutable_options_);
|
|
Statistics* const statistics = immutable_options_->stats;
|
|
|
|
RecordTick(statistics, NO_FILE_OPENS);
|
|
|
|
std::unique_ptr<BlobFileReader> reader;
|
|
|
|
{
|
|
assert(file_options_);
|
|
const Status s = BlobFileReader::Create(
|
|
*immutable_options_, read_options, *file_options_, column_family_id_,
|
|
blob_file_read_hist_, blob_file_number, io_tracer_, &reader);
|
|
if (!s.ok()) {
|
|
RecordTick(statistics, NO_FILE_ERRORS);
|
|
return s;
|
|
}
|
|
}
|
|
|
|
{
|
|
constexpr size_t charge = 1;
|
|
|
|
const Status s = cache_.Insert(key, reader.get(), charge, &handle);
|
|
if (!s.ok()) {
|
|
RecordTick(statistics, NO_FILE_ERRORS);
|
|
return s;
|
|
}
|
|
}
|
|
|
|
reader.release();
|
|
|
|
*blob_file_reader = cache_.Guard(handle);
|
|
|
|
return Status::OK();
|
|
}
|
|
|
|
} // namespace ROCKSDB_NAMESPACE
|