From 189f0c27aaecdf17ae7fc1f826a423a28b77984f Mon Sep 17 00:00:00 2001 From: Sagar Vemuri Date: Mon, 25 Jun 2018 13:07:38 -0700 Subject: [PATCH] Make BlockBasedTableIterator compaction-aware (#4048) Summary: Pass in `for_compaction` to `BlockBasedTableIterator` via `BlockBasedTableReader::NewIterator`. In 7103559f49b46b3287973045f741c0679e3e9e44, `for_compaction` was set in `BlockBasedTable::Rep` via `BlockBasedTable::SetupForCompaction`. In hindsight it was not the right decision; it also caused TSAN to complain. Closes https://github.com/facebook/rocksdb/pull/4048 Differential Revision: D8601056 Pulled By: sagar0 fbshipit-source-id: 30127e898c15c38c1080d57710b8c5a6d64a0ab3 --- db/table_cache.cc | 2 +- table/block_based_table_reader.cc | 11 ++++++----- table/block_based_table_reader.h | 11 +++++++---- table/cuckoo_table_reader.cc | 2 +- table/cuckoo_table_reader.h | 3 ++- table/mock_table.cc | 2 +- table/mock_table.h | 3 ++- table/plain_table_reader.cc | 2 +- table/plain_table_reader.h | 3 ++- table/table_reader.h | 3 ++- 10 files changed, 25 insertions(+), 17 deletions(-) diff --git a/db/table_cache.cc b/db/table_cache.cc index c38514cb0..15ad1a5bf 100644 --- a/db/table_cache.cc +++ b/db/table_cache.cc @@ -236,7 +236,7 @@ InternalIterator* TableCache::NewIterator( result = NewEmptyInternalIterator(arena); } else { result = table_reader->NewIterator(options, prefix_extractor, arena, - skip_filters); + skip_filters, for_compaction); } if (create_new_table_reader) { assert(handle == nullptr); diff --git a/table/block_based_table_reader.cc b/table/block_based_table_reader.cc index 2721d4cf2..f7b6cc4d7 100644 --- a/table/block_based_table_reader.cc +++ b/table/block_based_table_reader.cc @@ -1028,7 +1028,6 @@ void BlockBasedTable::SetupForCompaction() { default: assert(false); } - rep_->for_compaction = true; } std::shared_ptr BlockBasedTable::GetTableProperties() @@ -2004,7 +2003,7 @@ void BlockBasedTableIterator::InitDataBlock() { // Automatically prefetch additional data when a range scan (iterator) does // more than 2 sequential IOs. This is enabled only for user reads and when // ReadOptions.readahead_size is 0. - if (!rep->for_compaction && read_options_.readahead_size == 0) { + if (!for_compaction_ && read_options_.readahead_size == 0) { num_file_reads_++; if (num_file_reads_ > 2) { if (!rep->file->use_direct_io() && @@ -2101,7 +2100,7 @@ void BlockBasedTableIterator::FindKeyBackward() { InternalIterator* BlockBasedTable::NewIterator( const ReadOptions& read_options, const SliceTransform* prefix_extractor, - Arena* arena, bool skip_filters) { + Arena* arena, bool skip_filters, bool for_compaction) { bool prefix_extractor_changed = PrefixExtractorChanged(rep_->table_properties.get(), prefix_extractor); const bool kIsNotIndex = false; @@ -2114,7 +2113,8 @@ InternalIterator* BlockBasedTable::NewIterator( rep_->index_type == BlockBasedTableOptions::kHashSearch), !skip_filters && !read_options.total_order_seek && prefix_extractor != nullptr && !prefix_extractor_changed, - prefix_extractor, kIsNotIndex); + prefix_extractor, kIsNotIndex, true /*key_includes_seq*/, + for_compaction); } else { auto* mem = arena->AllocateAligned(sizeof(BlockBasedTableIterator)); return new (mem) BlockBasedTableIterator( @@ -2122,7 +2122,8 @@ InternalIterator* BlockBasedTable::NewIterator( NewIndexIterator(read_options, prefix_extractor_changed), !skip_filters && !read_options.total_order_seek && prefix_extractor != nullptr && !prefix_extractor_changed, - prefix_extractor, kIsNotIndex); + prefix_extractor, kIsNotIndex, true /*key_includes_seq*/, + for_compaction); } } diff --git a/table/block_based_table_reader.h b/table/block_based_table_reader.h index 3cbbbbc87..53ba5e197 100644 --- a/table/block_based_table_reader.h +++ b/table/block_based_table_reader.h @@ -104,7 +104,8 @@ class BlockBasedTable : public TableReader { InternalIterator* NewIterator(const ReadOptions&, const SliceTransform* prefix_extractor, Arena* arena = nullptr, - bool skip_filters = false) override; + bool skip_filters = false, + bool for_compaction = false) override; InternalIterator* NewRangeTombstoneIterator( const ReadOptions& read_options) override; @@ -506,8 +507,6 @@ struct BlockBasedTable::Rep { bool blocks_maybe_compressed = true; bool closed = false; - - bool for_compaction = false; }; class BlockBasedTableIterator : public InternalIterator { @@ -517,7 +516,8 @@ class BlockBasedTableIterator : public InternalIterator { const InternalKeyComparator& icomp, InternalIterator* index_iter, bool check_filter, const SliceTransform* prefix_extractor, bool is_index, - bool key_includes_seq = true) + bool key_includes_seq = true, + bool for_compaction = false) : table_(table), read_options_(read_options), icomp_(icomp), @@ -527,6 +527,7 @@ class BlockBasedTableIterator : public InternalIterator { check_filter_(check_filter), is_index_(is_index), key_includes_seq_(key_includes_seq), + for_compaction_(for_compaction), prefix_extractor_(prefix_extractor) {} ~BlockBasedTableIterator() { delete index_iter_; } @@ -624,6 +625,8 @@ class BlockBasedTableIterator : public InternalIterator { bool is_index_; // If the keys in the blocks over which we iterate include 8 byte sequence bool key_includes_seq_; + // If this iterator is created for compaction + bool for_compaction_; // TODO use block offset instead std::string prev_index_value_; const SliceTransform* prefix_extractor_; diff --git a/table/cuckoo_table_reader.cc b/table/cuckoo_table_reader.cc index 8748954e7..d008ff1a3 100644 --- a/table/cuckoo_table_reader.cc +++ b/table/cuckoo_table_reader.cc @@ -380,7 +380,7 @@ extern InternalIterator* NewErrorInternalIterator(const Status& status, InternalIterator* CuckooTableReader::NewIterator( const ReadOptions& /*read_options*/, const SliceTransform* /* prefix_extractor */, Arena* arena, - bool /*skip_filters*/) { + bool /*skip_filters*/, bool /*for_compaction*/) { if (!status().ok()) { return NewErrorInternalIterator( Status::Corruption("CuckooTableReader status is not okay."), arena); diff --git a/table/cuckoo_table_reader.h b/table/cuckoo_table_reader.h index c0d70174b..f1539f09f 100644 --- a/table/cuckoo_table_reader.h +++ b/table/cuckoo_table_reader.h @@ -49,7 +49,8 @@ class CuckooTableReader: public TableReader { InternalIterator* NewIterator(const ReadOptions&, const SliceTransform* prefix_extractor, Arena* arena = nullptr, - bool skip_filters = false) override; + bool skip_filters = false, + bool for_compaction = false) override; void Prepare(const Slice& target) override; // Report an approximation of how much memory has been used. diff --git a/table/mock_table.cc b/table/mock_table.cc index 32704dbca..54bab73d8 100644 --- a/table/mock_table.cc +++ b/table/mock_table.cc @@ -28,7 +28,7 @@ stl_wrappers::KVMap MakeMockFile( InternalIterator* MockTableReader::NewIterator( const ReadOptions&, const SliceTransform* /* prefix_extractor */, - Arena* /*arena*/, bool /*skip_filters*/) { + Arena* /*arena*/, bool /*skip_filters*/, bool /*for_compaction*/) { return new MockTableIterator(table_); } diff --git a/table/mock_table.h b/table/mock_table.h index a95725b6a..92cf87370 100644 --- a/table/mock_table.h +++ b/table/mock_table.h @@ -41,7 +41,8 @@ class MockTableReader : public TableReader { InternalIterator* NewIterator(const ReadOptions&, const SliceTransform* prefix_extractor, Arena* arena = nullptr, - bool skip_filters = false) override; + bool skip_filters = false, + bool for_compaction = false) override; Status Get(const ReadOptions& readOptions, const Slice& key, GetContext* get_context, const SliceTransform* prefix_extractor, diff --git a/table/plain_table_reader.cc b/table/plain_table_reader.cc index d6ab7f16e..1143eb1cd 100644 --- a/table/plain_table_reader.cc +++ b/table/plain_table_reader.cc @@ -191,7 +191,7 @@ void PlainTableReader::SetupForCompaction() { InternalIterator* PlainTableReader::NewIterator( const ReadOptions& options, const SliceTransform* /* prefix_extractor */, - Arena* arena, bool /*skip_filters*/) { + Arena* arena, bool /*skip_filters*/, bool /*for_compaction*/) { bool use_prefix_seek = !IsTotalOrderMode() && !options.total_order_seek; if (arena == nullptr) { return new PlainTableIterator(this, use_prefix_seek); diff --git a/table/plain_table_reader.h b/table/plain_table_reader.h index 5fdfa4786..8dd774e4a 100644 --- a/table/plain_table_reader.h +++ b/table/plain_table_reader.h @@ -82,7 +82,8 @@ class PlainTableReader: public TableReader { InternalIterator* NewIterator(const ReadOptions&, const SliceTransform* prefix_extractor, Arena* arena = nullptr, - bool skip_filters = false) override; + bool skip_filters = false, + bool for_compaction = false) override; void Prepare(const Slice& target) override; diff --git a/table/table_reader.h b/table/table_reader.h index 43a146b2b..b51b44b67 100644 --- a/table/table_reader.h +++ b/table/table_reader.h @@ -42,7 +42,8 @@ class TableReader { virtual InternalIterator* NewIterator(const ReadOptions&, const SliceTransform* prefix_extractor, Arena* arena = nullptr, - bool skip_filters = false) = 0; + bool skip_filters = false, + bool for_compaction = false) = 0; virtual InternalIterator* NewRangeTombstoneIterator( const ReadOptions& /*read_options*/) {