mirror of
https://github.com/facebook/rocksdb.git
synced 2024-12-03 14:52:53 +00:00
269478ee46
Summary: This PR implements support for a three tier cache - primary block cache, compressed secondary cache, and a nvm (local flash) secondary cache. This allows more effective utilization of the nvm cache, and minimizes the number of reads from local flash by caching compressed blocks in the compressed secondary cache. The basic design is as follows - 1. A new secondary cache implementation, ```TieredSecondaryCache```, is introduced. It keeps the compressed and nvm secondary caches and manages the movement of blocks between them and the primary block cache. To setup a three tier cache, we allocate a ```CacheWithSecondaryAdapter```, with a ```TieredSecondaryCache``` instance as the secondary cache. 2. The table reader passes both the uncompressed and compressed block to ```FullTypedCacheInterface::InsertFull```, allowing the block cache to optionally store the compressed block. 3. When there's a miss, the block object is constructed and inserted in the primary cache, and the compressed block is inserted into the nvm cache by calling ```InsertSaved```. This avoids the overhead of recompressing the block, as well as avoiding putting more memory pressure on the compressed secondary cache. 4. When there's a hit in the nvm cache, we attempt to insert the block in the compressed secondary cache and the primary cache, subject to the admission policy of those caches (i.e admit on second access). Blocks/items evicted from any tier are simply discarded. We can easily implement additional admission policies if desired. Todo (In a subsequent PR): 1. Add to db_bench and run benchmarks 2. Add to db_stress Pull Request resolved: https://github.com/facebook/rocksdb/pull/11812 Reviewed By: pdillinger Differential Revision: D49461842 Pulled By: anand1976 fbshipit-source-id: b40ac1330ef7cd8c12efa0a3ca75128e602e3a0b
109 lines
4.7 KiB
C++
109 lines
4.7 KiB
C++
// Copyright (c) Meta Platforms, Inc. and affiliates.
|
|
// 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 "table/block_based/block_cache.h"
|
|
|
|
#include "table/block_based/block_based_table_reader.h"
|
|
|
|
namespace ROCKSDB_NAMESPACE {
|
|
|
|
void BlockCreateContext::Create(std::unique_ptr<Block_kData>* parsed_out,
|
|
BlockContents&& block) {
|
|
parsed_out->reset(new Block_kData(
|
|
std::move(block), table_options->read_amp_bytes_per_bit, statistics));
|
|
parsed_out->get()->InitializeDataBlockProtectionInfo(protection_bytes_per_key,
|
|
raw_ucmp);
|
|
}
|
|
void BlockCreateContext::Create(std::unique_ptr<Block_kIndex>* parsed_out,
|
|
BlockContents&& block) {
|
|
parsed_out->reset(new Block_kIndex(std::move(block),
|
|
/*read_amp_bytes_per_bit*/ 0, statistics));
|
|
parsed_out->get()->InitializeIndexBlockProtectionInfo(
|
|
protection_bytes_per_key, raw_ucmp, index_value_is_full,
|
|
index_has_first_key);
|
|
}
|
|
void BlockCreateContext::Create(
|
|
std::unique_ptr<Block_kFilterPartitionIndex>* parsed_out,
|
|
BlockContents&& block) {
|
|
parsed_out->reset(new Block_kFilterPartitionIndex(
|
|
std::move(block), /*read_amp_bytes_per_bit*/ 0, statistics));
|
|
parsed_out->get()->InitializeIndexBlockProtectionInfo(
|
|
protection_bytes_per_key, raw_ucmp, index_value_is_full,
|
|
index_has_first_key);
|
|
}
|
|
void BlockCreateContext::Create(
|
|
std::unique_ptr<Block_kRangeDeletion>* parsed_out, BlockContents&& block) {
|
|
parsed_out->reset(new Block_kRangeDeletion(
|
|
std::move(block), /*read_amp_bytes_per_bit*/ 0, statistics));
|
|
}
|
|
void BlockCreateContext::Create(std::unique_ptr<Block_kMetaIndex>* parsed_out,
|
|
BlockContents&& block) {
|
|
parsed_out->reset(new Block_kMetaIndex(
|
|
std::move(block), /*read_amp_bytes_per_bit*/ 0, statistics));
|
|
parsed_out->get()->InitializeMetaIndexBlockProtectionInfo(
|
|
protection_bytes_per_key);
|
|
}
|
|
|
|
void BlockCreateContext::Create(
|
|
std::unique_ptr<ParsedFullFilterBlock>* parsed_out, BlockContents&& block) {
|
|
parsed_out->reset(new ParsedFullFilterBlock(
|
|
table_options->filter_policy.get(), std::move(block)));
|
|
}
|
|
|
|
void BlockCreateContext::Create(std::unique_ptr<UncompressionDict>* parsed_out,
|
|
BlockContents&& block) {
|
|
parsed_out->reset(new UncompressionDict(
|
|
block.data, std::move(block.allocation), using_zstd));
|
|
}
|
|
|
|
namespace {
|
|
// For getting SecondaryCache-compatible helpers from a BlockType. This is
|
|
// useful for accessing block cache in untyped contexts, such as for generic
|
|
// cache warming in table builder.
|
|
const std::array<const Cache::CacheItemHelper*,
|
|
static_cast<unsigned>(BlockType::kInvalid) + 1>
|
|
kCacheItemFullHelperForBlockType{{
|
|
BlockCacheInterface<Block_kData>::GetFullHelper(),
|
|
BlockCacheInterface<ParsedFullFilterBlock>::GetFullHelper(),
|
|
BlockCacheInterface<Block_kFilterPartitionIndex>::GetFullHelper(),
|
|
nullptr, // kProperties
|
|
BlockCacheInterface<UncompressionDict>::GetFullHelper(),
|
|
BlockCacheInterface<Block_kRangeDeletion>::GetFullHelper(),
|
|
nullptr, // kHashIndexPrefixes
|
|
nullptr, // kHashIndexMetadata
|
|
nullptr, // kMetaIndex (not yet stored in block cache)
|
|
BlockCacheInterface<Block_kIndex>::GetFullHelper(),
|
|
nullptr, // kInvalid
|
|
}};
|
|
|
|
// For getting basic helpers from a BlockType (no SecondaryCache support)
|
|
const std::array<const Cache::CacheItemHelper*,
|
|
static_cast<unsigned>(BlockType::kInvalid) + 1>
|
|
kCacheItemBasicHelperForBlockType{{
|
|
BlockCacheInterface<Block_kData>::GetBasicHelper(),
|
|
BlockCacheInterface<ParsedFullFilterBlock>::GetBasicHelper(),
|
|
BlockCacheInterface<Block_kFilterPartitionIndex>::GetBasicHelper(),
|
|
nullptr, // kProperties
|
|
BlockCacheInterface<UncompressionDict>::GetBasicHelper(),
|
|
BlockCacheInterface<Block_kRangeDeletion>::GetBasicHelper(),
|
|
nullptr, // kHashIndexPrefixes
|
|
nullptr, // kHashIndexMetadata
|
|
nullptr, // kMetaIndex (not yet stored in block cache)
|
|
BlockCacheInterface<Block_kIndex>::GetBasicHelper(),
|
|
nullptr, // kInvalid
|
|
}};
|
|
} // namespace
|
|
|
|
const Cache::CacheItemHelper* GetCacheItemHelper(
|
|
BlockType block_type, CacheTier lowest_used_cache_tier) {
|
|
if (lowest_used_cache_tier > CacheTier::kVolatileTier) {
|
|
return kCacheItemFullHelperForBlockType[static_cast<unsigned>(block_type)];
|
|
} else {
|
|
return kCacheItemBasicHelperForBlockType[static_cast<unsigned>(block_type)];
|
|
}
|
|
}
|
|
|
|
} // namespace ROCKSDB_NAMESPACE
|