mirror of
https://github.com/facebook/rocksdb.git
synced 2024-12-02 10:15:54 +00:00
ea8bb82fc7
Summary: Add support for IOTracing in blob files Pull Request resolved: https://github.com/facebook/rocksdb/pull/7958 Test Plan: Add a new test and checked manually the trace_file for blob files being recorded during read and write. Reviewed By: ltamasi Differential Revision: D26415950 Pulled By: akankshamahajan15 fbshipit-source-id: 49c2859b3a4f8307e7cb69a92704403a4da46d44
103 lines
2.9 KiB
C++
103 lines
2.9 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 ImmutableCFOptions* immutable_cf_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, kGetSliceNPHash64UnseededFnPtr),
|
|
immutable_cf_options_(immutable_cf_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_cf_options_);
|
|
assert(file_options_);
|
|
}
|
|
|
|
Status BlobFileCache::GetBlobFileReader(
|
|
uint64_t blob_file_number,
|
|
CacheHandleGuard<BlobFileReader>* blob_file_reader) {
|
|
assert(blob_file_reader);
|
|
assert(blob_file_reader->IsEmpty());
|
|
|
|
const Slice key = GetSlice(&blob_file_number);
|
|
|
|
assert(cache_);
|
|
|
|
Cache::Handle* handle = cache_->Lookup(key);
|
|
if (handle) {
|
|
*blob_file_reader = CacheHandleGuard<BlobFileReader>(cache_, 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 = CacheHandleGuard<BlobFileReader>(cache_, handle);
|
|
return Status::OK();
|
|
}
|
|
|
|
assert(immutable_cf_options_);
|
|
Statistics* const statistics = immutable_cf_options_->statistics;
|
|
|
|
RecordTick(statistics, NO_FILE_OPENS);
|
|
|
|
std::unique_ptr<BlobFileReader> reader;
|
|
|
|
{
|
|
assert(file_options_);
|
|
const Status s = BlobFileReader::Create(
|
|
*immutable_cf_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,
|
|
&DeleteCacheEntry<BlobFileReader>, &handle);
|
|
if (!s.ok()) {
|
|
RecordTick(statistics, NO_FILE_ERRORS);
|
|
return s;
|
|
}
|
|
}
|
|
|
|
reader.release();
|
|
|
|
*blob_file_reader = CacheHandleGuard<BlobFileReader>(cache_, handle);
|
|
|
|
return Status::OK();
|
|
}
|
|
|
|
} // namespace ROCKSDB_NAMESPACE
|