Summary: BlockCacheLookupContext only contains the caller for now. We will trace block accesses at five places: 1. BlockBasedTable::GetFilter. 2. BlockBasedTable::GetUncompressedDict. 3. BlockBasedTable::MaybeReadAndLoadToCache. (To trace access on data, index, and range deletion block.) 4. BlockBasedTable::Get. (To trace the referenced key and whether the referenced key exists in a fetched data block.) 5. BlockBasedTable::MultiGet. (To trace the referenced key and whether the referenced key exists in a fetched data block.) We create the context at: 1. BlockBasedTable::Get. (kUserGet) 2. BlockBasedTable::MultiGet. (kUserMGet) 3. BlockBasedTable::NewIterator. (either kUserIterator, kCompaction, or external SST ingestion calls this function.) 4. BlockBasedTable::Open. (kPrefetch) 5. Index/Filter::CacheDependencies. (kPrefetch) 6. BlockBasedTable::ApproximateOffsetOf. (kCompaction or kUserApproximateSize). I loaded 1 million key-value pairs into the database and ran the readrandom benchmark with a single thread. I gave the block cache 10 GB to make sure all reads hit the block cache after warmup. The throughput is comparable. Throughput of this PR: 231334 ops/s. Throughput of the master branch: 238428 ops/s. Experiment setup: RocksDB: version 6.2 Date: Mon Jun 10 10:42:51 2019 CPU: 24 * Intel Core Processor (Skylake) CPUCache: 16384 KB Keys: 20 bytes each Values: 100 bytes each (100 bytes after compression) Entries: 1000000 Prefix: 20 bytes Keys per prefix: 0 RawSize: 114.4 MB (estimated) FileSize: 114.4 MB (estimated) Write rate: 0 bytes/second Read rate: 0 ops/second Compression: NoCompression Compression sampling rate: 0 Memtablerep: skip_list Perf Level: 1 Load command: ./db_bench --benchmarks="fillseq" --key_size=20 --prefix_size=20 --keys_per_prefix=0 --value_size=100 --statistics --cache_index_and_filter_blocks --cache_size=10737418240 --disable_auto_compactions=1 --disable_wal=1 --compression_type=none --min_level_to_compress=-1 --compression_ratio=1 --num=1000000 Run command: ./db_bench --benchmarks="readrandom,stats" --use_existing_db --threads=1 --duration=120 --key_size=20 --prefix_size=20 --keys_per_prefix=0 --value_size=100 --statistics --cache_index_and_filter_blocks --cache_size=10737418240 --disable_auto_compactions=1 --disable_wal=1 --compression_type=none --min_level_to_compress=-1 --compression_ratio=1 --num=1000000 --duration=120 TODOs: 1. Create a caller for external SST file ingestion and differentiate the callers for iterator. 2. Integrate tracer to trace block cache accesses. Pull Request resolved: https://github.com/facebook/rocksdb/pull/5421 Differential Revision: D15704258 Pulled By: HaoyuHuang fbshipit-source-id: 4aa8a55f8cb1576ffb367bfa3186a91d8f06d93a
349 lines
14 KiB
C++
349 lines
14 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).
|
|
//
|
|
// Copyright (c) 2012 The LevelDB Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style license that can be
|
|
// found in the LICENSE file. See the AUTHORS file for names of contributors.
|
|
|
|
#include "table/block_based/block_based_filter_block.h"
|
|
|
|
#include "rocksdb/filter_policy.h"
|
|
#include "test_util/testharness.h"
|
|
#include "test_util/testutil.h"
|
|
#include "util/coding.h"
|
|
#include "util/hash.h"
|
|
#include "util/string_util.h"
|
|
|
|
namespace rocksdb {
|
|
|
|
// For testing: emit an array with one hash value per key
|
|
class TestHashFilter : public FilterPolicy {
|
|
public:
|
|
const char* Name() const override { return "TestHashFilter"; }
|
|
|
|
void CreateFilter(const Slice* keys, int n, std::string* dst) const override {
|
|
for (int i = 0; i < n; i++) {
|
|
uint32_t h = Hash(keys[i].data(), keys[i].size(), 1);
|
|
PutFixed32(dst, h);
|
|
}
|
|
}
|
|
|
|
bool KeyMayMatch(const Slice& key, const Slice& filter) const override {
|
|
uint32_t h = Hash(key.data(), key.size(), 1);
|
|
for (unsigned int i = 0; i + 4 <= filter.size(); i += 4) {
|
|
if (h == DecodeFixed32(filter.data() + i)) {
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
};
|
|
|
|
class FilterBlockTest : public testing::Test {
|
|
public:
|
|
TestHashFilter policy_;
|
|
BlockBasedTableOptions table_options_;
|
|
|
|
FilterBlockTest() {
|
|
table_options_.filter_policy.reset(new TestHashFilter());
|
|
}
|
|
};
|
|
|
|
TEST_F(FilterBlockTest, EmptyBuilder) {
|
|
BlockBasedFilterBlockBuilder builder(nullptr, table_options_);
|
|
BlockContents block(builder.Finish());
|
|
ASSERT_EQ("\\x00\\x00\\x00\\x00\\x0b", EscapeString(block.data));
|
|
BlockBasedFilterBlockReader reader(nullptr, table_options_, true,
|
|
std::move(block), nullptr);
|
|
ASSERT_TRUE(reader.KeyMayMatch(
|
|
"foo", /*prefix_extractor=*/nullptr, /*block_offset=*/uint64_t{0},
|
|
/*no_io=*/false, /*const_ikey_ptr=*/nullptr, /*context=*/nullptr));
|
|
ASSERT_TRUE(reader.KeyMayMatch(
|
|
"foo", /*prefix_extractor=*/nullptr, /*block_offset=*/100000,
|
|
/*no_io=*/false, /*const_ikey_ptr=*/nullptr, /*context=*/nullptr));
|
|
}
|
|
|
|
TEST_F(FilterBlockTest, SingleChunk) {
|
|
BlockBasedFilterBlockBuilder builder(nullptr, table_options_);
|
|
ASSERT_EQ(0, builder.NumAdded());
|
|
builder.StartBlock(100);
|
|
builder.Add("foo");
|
|
builder.Add("bar");
|
|
builder.Add("box");
|
|
builder.StartBlock(200);
|
|
builder.Add("box");
|
|
builder.StartBlock(300);
|
|
builder.Add("hello");
|
|
ASSERT_EQ(5, builder.NumAdded());
|
|
BlockContents block(builder.Finish());
|
|
BlockBasedFilterBlockReader reader(nullptr, table_options_, true,
|
|
std::move(block), nullptr);
|
|
ASSERT_TRUE(reader.KeyMayMatch(
|
|
"foo", /*prefix_extractor=*/nullptr, /*block_offset=*/100,
|
|
/*no_io=*/false, /*const_ikey_ptr=*/nullptr, /*context=*/nullptr));
|
|
ASSERT_TRUE(reader.KeyMayMatch(
|
|
"bar", /*prefix_extractor=*/nullptr, /*block_offset=*/100,
|
|
/*no_io=*/false, /*const_ikey_ptr=*/nullptr, /*context=*/nullptr));
|
|
ASSERT_TRUE(reader.KeyMayMatch(
|
|
"box", /*prefix_extractor=*/nullptr, /*block_offset=*/100,
|
|
/*no_io=*/false, /*const_ikey_ptr=*/nullptr, /*context=*/nullptr));
|
|
ASSERT_TRUE(reader.KeyMayMatch(
|
|
"hello", /*prefix_extractor=*/nullptr, /*block_offset=*/100,
|
|
/*no_io=*/false, /*const_ikey_ptr=*/nullptr, /*context=*/nullptr));
|
|
ASSERT_TRUE(reader.KeyMayMatch(
|
|
"foo", /*prefix_extractor=*/nullptr, /*block_offset=*/100,
|
|
/*no_io=*/false, /*const_ikey_ptr=*/nullptr, /*context=*/nullptr));
|
|
ASSERT_TRUE(!reader.KeyMayMatch(
|
|
"missing", /*prefix_extractor=*/nullptr, /*block_offset=*/100,
|
|
/*no_io=*/false, /*const_ikey_ptr=*/nullptr, /*context=*/nullptr));
|
|
ASSERT_TRUE(!reader.KeyMayMatch(
|
|
"other", /*prefix_extractor=*/nullptr, /*block_offset=*/100,
|
|
/*no_io=*/false, /*const_ikey_ptr=*/nullptr, /*context=*/nullptr));
|
|
}
|
|
|
|
TEST_F(FilterBlockTest, MultiChunk) {
|
|
BlockBasedFilterBlockBuilder builder(nullptr, table_options_);
|
|
|
|
// First filter
|
|
builder.StartBlock(0);
|
|
builder.Add("foo");
|
|
builder.StartBlock(2000);
|
|
builder.Add("bar");
|
|
|
|
// Second filter
|
|
builder.StartBlock(3100);
|
|
builder.Add("box");
|
|
|
|
// Third filter is empty
|
|
|
|
// Last filter
|
|
builder.StartBlock(9000);
|
|
builder.Add("box");
|
|
builder.Add("hello");
|
|
|
|
BlockContents block(builder.Finish());
|
|
BlockBasedFilterBlockReader reader(nullptr, table_options_, true,
|
|
std::move(block), nullptr);
|
|
|
|
// Check first filter
|
|
ASSERT_TRUE(reader.KeyMayMatch(
|
|
"foo", /*prefix_extractor=*/nullptr, /*block_offset=*/uint64_t{0},
|
|
/*no_io=*/false, /*const_ikey_ptr=*/nullptr, /*context=*/nullptr));
|
|
ASSERT_TRUE(reader.KeyMayMatch(
|
|
"bar", /*prefix_extractor=*/nullptr, /*block_offset=*/2000,
|
|
/*no_io=*/false, /*const_ikey_ptr=*/nullptr, /*context=*/nullptr));
|
|
ASSERT_TRUE(!reader.KeyMayMatch(
|
|
"box", /*prefix_extractor=*/nullptr, /*block_offset=*/uint64_t{0},
|
|
/*no_io=*/false, /*const_ikey_ptr=*/nullptr, /*context=*/nullptr));
|
|
ASSERT_TRUE(!reader.KeyMayMatch(
|
|
"hello", /*prefix_extractor=*/nullptr, /*block_offset=*/uint64_t{0},
|
|
/*no_io=*/false, /*const_ikey_ptr=*/nullptr, /*context=*/nullptr));
|
|
|
|
// Check second filter
|
|
ASSERT_TRUE(reader.KeyMayMatch(
|
|
"box", /*prefix_extractor=*/nullptr, /*block_offset=*/3100,
|
|
/*no_io=*/false, /*const_ikey_ptr=*/nullptr, /*context=*/nullptr));
|
|
ASSERT_TRUE(!reader.KeyMayMatch(
|
|
"foo", /*prefix_extractor=*/nullptr, /*block_offset=*/3100,
|
|
/*no_io=*/false, /*const_ikey_ptr=*/nullptr, /*context=*/nullptr));
|
|
ASSERT_TRUE(!reader.KeyMayMatch(
|
|
"bar", /*prefix_extractor=*/nullptr, /*block_offset=*/3100,
|
|
/*no_io=*/false, /*const_ikey_ptr=*/nullptr, /*context=*/nullptr));
|
|
ASSERT_TRUE(!reader.KeyMayMatch(
|
|
"hello", /*prefix_extractor=*/nullptr, /*block_offset=*/3100,
|
|
/*no_io=*/false, /*const_ikey_ptr=*/nullptr, /*context=*/nullptr));
|
|
|
|
// Check third filter (empty)
|
|
ASSERT_TRUE(!reader.KeyMayMatch(
|
|
"foo", /*prefix_extractor=*/nullptr, /*block_offset=*/4100,
|
|
/*no_io=*/false, /*const_ikey_ptr=*/nullptr, /*context=*/nullptr));
|
|
ASSERT_TRUE(!reader.KeyMayMatch(
|
|
"bar", /*prefix_extractor=*/nullptr, /*block_offset=*/4100,
|
|
/*no_io=*/false, /*const_ikey_ptr=*/nullptr, /*context=*/nullptr));
|
|
ASSERT_TRUE(!reader.KeyMayMatch(
|
|
"box", /*prefix_extractor=*/nullptr, /*block_offset=*/4100,
|
|
/*no_io=*/false, /*const_ikey_ptr=*/nullptr, /*context=*/nullptr));
|
|
ASSERT_TRUE(!reader.KeyMayMatch(
|
|
"hello", /*prefix_extractor=*/nullptr, /*block_offset=*/4100,
|
|
/*no_io=*/false, /*const_ikey_ptr=*/nullptr, /*context=*/nullptr));
|
|
|
|
// Check last filter
|
|
ASSERT_TRUE(reader.KeyMayMatch(
|
|
"box", /*prefix_extractor=*/nullptr, /*block_offset=*/9000,
|
|
/*no_io=*/false, /*const_ikey_ptr=*/nullptr, /*context=*/nullptr));
|
|
ASSERT_TRUE(reader.KeyMayMatch(
|
|
"hello", /*prefix_extractor=*/nullptr, /*block_offset=*/9000,
|
|
/*no_io=*/false, /*const_ikey_ptr=*/nullptr, /*context=*/nullptr));
|
|
ASSERT_TRUE(!reader.KeyMayMatch(
|
|
"foo", /*prefix_extractor=*/nullptr, /*block_offset=*/9000,
|
|
/*no_io=*/false, /*const_ikey_ptr=*/nullptr, /*context=*/nullptr));
|
|
ASSERT_TRUE(!reader.KeyMayMatch(
|
|
"bar", /*prefix_extractor=*/nullptr, /*block_offset=*/9000,
|
|
/*no_io=*/false, /*const_ikey_ptr=*/nullptr, /*context=*/nullptr));
|
|
}
|
|
|
|
// Test for block based filter block
|
|
// use new interface in FilterPolicy to create filter builder/reader
|
|
class BlockBasedFilterBlockTest : public testing::Test {
|
|
public:
|
|
BlockBasedTableOptions table_options_;
|
|
|
|
BlockBasedFilterBlockTest() {
|
|
table_options_.filter_policy.reset(NewBloomFilterPolicy(10));
|
|
}
|
|
|
|
~BlockBasedFilterBlockTest() override {}
|
|
};
|
|
|
|
TEST_F(BlockBasedFilterBlockTest, BlockBasedEmptyBuilder) {
|
|
FilterBlockBuilder* builder =
|
|
new BlockBasedFilterBlockBuilder(nullptr, table_options_);
|
|
BlockContents block(builder->Finish());
|
|
ASSERT_EQ("\\x00\\x00\\x00\\x00\\x0b", EscapeString(block.data));
|
|
FilterBlockReader* reader = new BlockBasedFilterBlockReader(
|
|
nullptr, table_options_, true, std::move(block), nullptr);
|
|
ASSERT_TRUE(reader->KeyMayMatch(
|
|
"foo", /*prefix_extractor=*/nullptr, /*block_offset=*/uint64_t{0},
|
|
/*no_io=*/false, /*const_ikey_ptr=*/nullptr, /*context=*/nullptr));
|
|
ASSERT_TRUE(reader->KeyMayMatch(
|
|
"foo", /*prefix_extractor=*/nullptr, /*block_offset=*/10000,
|
|
/*no_io=*/false, /*const_ikey_ptr=*/nullptr, /*context=*/nullptr));
|
|
|
|
delete builder;
|
|
delete reader;
|
|
}
|
|
|
|
TEST_F(BlockBasedFilterBlockTest, BlockBasedSingleChunk) {
|
|
FilterBlockBuilder* builder =
|
|
new BlockBasedFilterBlockBuilder(nullptr, table_options_);
|
|
builder->StartBlock(100);
|
|
builder->Add("foo");
|
|
builder->Add("bar");
|
|
builder->Add("box");
|
|
builder->StartBlock(200);
|
|
builder->Add("box");
|
|
builder->StartBlock(300);
|
|
builder->Add("hello");
|
|
BlockContents block(builder->Finish());
|
|
FilterBlockReader* reader = new BlockBasedFilterBlockReader(
|
|
nullptr, table_options_, true, std::move(block), nullptr);
|
|
ASSERT_TRUE(reader->KeyMayMatch(
|
|
"foo", /*prefix_extractor=*/nullptr, /*block_offset=*/100,
|
|
/*no_io=*/false, /*const_ikey_ptr=*/nullptr, /*context=*/nullptr));
|
|
ASSERT_TRUE(reader->KeyMayMatch(
|
|
"bar", /*prefix_extractor=*/nullptr, /*block_offset=*/100,
|
|
/*no_io=*/false, /*const_ikey_ptr=*/nullptr, /*context=*/nullptr));
|
|
ASSERT_TRUE(reader->KeyMayMatch(
|
|
"box", /*prefix_extractor=*/nullptr, /*block_offset=*/100,
|
|
/*no_io=*/false, /*const_ikey_ptr=*/nullptr, /*context=*/nullptr));
|
|
ASSERT_TRUE(reader->KeyMayMatch(
|
|
"hello", /*prefix_extractor=*/nullptr, /*block_offset=*/100,
|
|
/*no_io=*/false, /*const_ikey_ptr=*/nullptr, /*context=*/nullptr));
|
|
ASSERT_TRUE(reader->KeyMayMatch(
|
|
"foo", /*prefix_extractor=*/nullptr, /*block_offset=*/100,
|
|
/*no_io=*/false, /*const_ikey_ptr=*/nullptr, /*context=*/nullptr));
|
|
ASSERT_TRUE(!reader->KeyMayMatch(
|
|
"missing", /*prefix_extractor=*/nullptr, /*block_offset=*/100,
|
|
/*no_io=*/false, /*const_ikey_ptr=*/nullptr, /*context=*/nullptr));
|
|
ASSERT_TRUE(!reader->KeyMayMatch(
|
|
"other", /*prefix_extractor=*/nullptr, /*block_offset=*/100,
|
|
/*no_io=*/false, /*const_ikey_ptr=*/nullptr, /*context=*/nullptr));
|
|
|
|
delete builder;
|
|
delete reader;
|
|
}
|
|
|
|
TEST_F(BlockBasedFilterBlockTest, BlockBasedMultiChunk) {
|
|
FilterBlockBuilder* builder =
|
|
new BlockBasedFilterBlockBuilder(nullptr, table_options_);
|
|
|
|
// First filter
|
|
builder->StartBlock(0);
|
|
builder->Add("foo");
|
|
builder->StartBlock(2000);
|
|
builder->Add("bar");
|
|
|
|
// Second filter
|
|
builder->StartBlock(3100);
|
|
builder->Add("box");
|
|
|
|
// Third filter is empty
|
|
|
|
// Last filter
|
|
builder->StartBlock(9000);
|
|
builder->Add("box");
|
|
builder->Add("hello");
|
|
|
|
BlockContents block(builder->Finish());
|
|
FilterBlockReader* reader = new BlockBasedFilterBlockReader(
|
|
nullptr, table_options_, true, std::move(block), nullptr);
|
|
|
|
// Check first filter
|
|
ASSERT_TRUE(reader->KeyMayMatch(
|
|
"foo", /*prefix_extractor=*/nullptr, /*block_offset=*/uint64_t{0},
|
|
/*no_io=*/false, /*const_ikey_ptr=*/nullptr, /*context=*/nullptr));
|
|
ASSERT_TRUE(reader->KeyMayMatch(
|
|
"bar", /*prefix_extractor=*/nullptr, /*block_offset=*/2000,
|
|
/*no_io=*/false, /*const_ikey_ptr=*/nullptr, /*context=*/nullptr));
|
|
ASSERT_TRUE(!reader->KeyMayMatch(
|
|
"box", /*prefix_extractor=*/nullptr, /*block_offset=*/uint64_t{0},
|
|
/*no_io=*/false, /*const_ikey_ptr=*/nullptr, /*context=*/nullptr));
|
|
ASSERT_TRUE(!reader->KeyMayMatch(
|
|
"hello", /*prefix_extractor=*/nullptr, /*block_offset=*/uint64_t{0},
|
|
/*no_io=*/false, /*const_ikey_ptr=*/nullptr, /*context=*/nullptr));
|
|
|
|
// Check second filter
|
|
ASSERT_TRUE(reader->KeyMayMatch(
|
|
"box", /*prefix_extractor=*/nullptr, /*block_offset=*/3100,
|
|
/*no_io=*/false, /*const_ikey_ptr=*/nullptr, /*context=*/nullptr));
|
|
ASSERT_TRUE(!reader->KeyMayMatch(
|
|
"foo", /*prefix_extractor=*/nullptr, /*block_offset=*/3100,
|
|
/*no_io=*/false, /*const_ikey_ptr=*/nullptr, /*context=*/nullptr));
|
|
ASSERT_TRUE(!reader->KeyMayMatch(
|
|
"bar", /*prefix_extractor=*/nullptr, /*block_offset=*/3100,
|
|
/*no_io=*/false, /*const_ikey_ptr=*/nullptr, /*context=*/nullptr));
|
|
ASSERT_TRUE(!reader->KeyMayMatch(
|
|
"hello", /*prefix_extractor=*/nullptr, /*block_offset=*/3100,
|
|
/*no_io=*/false, /*const_ikey_ptr=*/nullptr, /*context=*/nullptr));
|
|
|
|
// Check third filter (empty)
|
|
ASSERT_TRUE(!reader->KeyMayMatch(
|
|
"foo", /*prefix_extractor=*/nullptr, /*block_offset=*/4100,
|
|
/*no_io=*/false, /*const_ikey_ptr=*/nullptr, /*context=*/nullptr));
|
|
ASSERT_TRUE(!reader->KeyMayMatch(
|
|
"bar", /*prefix_extractor=*/nullptr, /*block_offset=*/4100,
|
|
/*no_io=*/false, /*const_ikey_ptr=*/nullptr, /*context=*/nullptr));
|
|
ASSERT_TRUE(!reader->KeyMayMatch(
|
|
"box", /*prefix_extractor=*/nullptr, /*block_offset=*/4100,
|
|
/*no_io=*/false, /*const_ikey_ptr=*/nullptr, /*context=*/nullptr));
|
|
ASSERT_TRUE(!reader->KeyMayMatch(
|
|
"hello", /*prefix_extractor=*/nullptr, /*block_offset=*/4100,
|
|
/*no_io=*/false, /*const_ikey_ptr=*/nullptr, /*context=*/nullptr));
|
|
|
|
// Check last filter
|
|
ASSERT_TRUE(reader->KeyMayMatch(
|
|
"box", /*prefix_extractor=*/nullptr, /*block_offset=*/9000,
|
|
/*no_io=*/false, /*const_ikey_ptr=*/nullptr, /*context=*/nullptr));
|
|
ASSERT_TRUE(reader->KeyMayMatch(
|
|
"hello", /*prefix_extractor=*/nullptr, /*block_offset=*/9000,
|
|
/*no_io=*/false, /*const_ikey_ptr=*/nullptr, /*context=*/nullptr));
|
|
ASSERT_TRUE(!reader->KeyMayMatch(
|
|
"foo", /*prefix_extractor=*/nullptr, /*block_offset=*/9000,
|
|
/*no_io=*/false, /*const_ikey_ptr=*/nullptr, /*context=*/nullptr));
|
|
ASSERT_TRUE(!reader->KeyMayMatch(
|
|
"bar", /*prefix_extractor=*/nullptr, /*block_offset=*/9000,
|
|
/*no_io=*/false, /*const_ikey_ptr=*/nullptr, /*context=*/nullptr));
|
|
|
|
delete builder;
|
|
delete reader;
|
|
}
|
|
|
|
} // namespace rocksdb
|
|
|
|
int main(int argc, char** argv) {
|
|
::testing::InitGoogleTest(&argc, argv);
|
|
return RUN_ALL_TESTS();
|
|
}
|