andreacavalli/rocksdb
0
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

"make format" in some recent commits

Browse Source 
Summary: Run "make format" for some recent commits.

Test Plan: Build and run tests

Reviewers: IslamAbdelRahman

Reviewed By: IslamAbdelRahman

Subscribers: leveldb, dhruba

Differential Revision: https://reviews.facebook.net/D49707
This commit is contained in:
sdong 2015-10-29 15:52:32 -07:00
parent 6388e7f4e2
commit 296c3a1f94
13 changed files with 240 additions and 232 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
db/db_impl.cc
View File

@ -1158,17 +1158,18 @@ Status DBImpl::RecoverLogFiles(const std::vector<uint64_t>& log_numbers,
bool batch_changed = false; bool batch_changed = false;
WalFilter::WalProcessingOption wal_processing_option = WalFilter::WalProcessingOption wal_processing_option =
db_options_.wal_filter->LogRecord(batch, &new_batch, &batch_changed); db_options_.wal_filter->LogRecord(batch, &new_batch,
&batch_changed);
switch (wal_processing_option) { switch (wal_processing_option) {
case WalFilter::WalProcessingOption::kContinueProcessing: case WalFilter::WalProcessingOption::kContinueProcessing:
//do nothing, proceeed normally // do nothing, proceeed normally
break; break;
case WalFilter::WalProcessingOption::kIgnoreCurrentRecord: case WalFilter::WalProcessingOption::kIgnoreCurrentRecord:
//skip current record // skip current record
continue; continue;
case WalFilter::WalProcessingOption::kStopReplay: case WalFilter::WalProcessingOption::kStopReplay:
//skip current record and stop replay // skip current record and stop replay
continue_replay_log = false; continue_replay_log = false;
continue; continue;
case WalFilter::WalProcessingOption::kCorruptedRecord: { case WalFilter::WalProcessingOption::kCorruptedRecord: {
@ -1182,14 +1183,15 @@ Status DBImpl::RecoverLogFiles(const std::vector<uint64_t>& log_numbers,
break; break;
} }
default: { default: {
assert(false); //unhandled case assert(false); // unhandled case
status = Status::NotSupported("Unknown WalProcessingOption returned" status = Status::NotSupported(
" by Wal Filter ", db_options_.wal_filter->Name()); "Unknown WalProcessingOption returned"
" by Wal Filter ",
db_options_.wal_filter->Name());
MaybeIgnoreError(&status); MaybeIgnoreError(&status);
if (!status.ok()) { if (!status.ok()) {
return status; return status;
} } else {
else {
// Ignore the error with current record processing. // Ignore the error with current record processing.
continue; continue;
} }
@ -1203,13 +1205,16 @@ Status DBImpl::RecoverLogFiles(const std::vector<uint64_t>& log_numbers,
int original_count = WriteBatchInternal::Count(&batch); int original_count = WriteBatchInternal::Count(&batch);
if (new_count > original_count) { if (new_count > original_count) {
Log(InfoLogLevel::FATAL_LEVEL, db_options_.info_log, Log(InfoLogLevel::FATAL_LEVEL, db_options_.info_log,
"Recovering log #%" PRIu64 " mode %d log filter %s returned " "Recovering log #%" PRIu64
" mode %d log filter %s returned "
"more records (%d) than original (%d) which is not allowed. " "more records (%d) than original (%d) which is not allowed. "
"Aborting recovery.", "Aborting recovery.",
log_number, db_options_.wal_recovery_mode, log_number, db_options_.wal_recovery_mode,
db_options_.wal_filter->Name(), new_count, original_count); db_options_.wal_filter->Name(), new_count, original_count);
status = Status::NotSupported("More than original # of records " status = Status::NotSupported(
"returned by Wal Filter ", db_options_.wal_filter->Name()); "More than original # of records "
"returned by Wal Filter ",
db_options_.wal_filter->Name());
return status; return status;
} }
// Set the same sequence number in the new_batch // Set the same sequence number in the new_batch
@ -1219,7 +1224,7 @@ Status DBImpl::RecoverLogFiles(const std::vector<uint64_t>& log_numbers,
batch = new_batch; batch = new_batch;
} }
} }
#endif //ROCKSDB_LITE #endif // ROCKSDB_LITE
// If column family was not found, it might mean that the WAL write // If column family was not found, it might mean that the WAL write
// batch references to the column family that was dropped after the // batch references to the column family that was dropped after the
@ -4172,8 +4177,7 @@ Status DBImpl::SwitchMemtable(ColumnFamilyData* cfd, WriteContext* context) {
mutable_cf_options.write_buffer_size); mutable_cf_options.write_buffer_size);
unique_ptr<WritableFileWriter> file_writer( unique_ptr<WritableFileWriter> file_writer(
new WritableFileWriter(std::move(lfile), opt_env_opt)); new WritableFileWriter(std::move(lfile), opt_env_opt));
new_log = new log::Writer(std::move(file_writer), new_log = new log::Writer(std::move(file_writer), new_log_number,
new_log_number,
db_options_.recycle_log_file_num > 0); db_options_.recycle_log_file_num > 0);
} }
} }
@ -4815,8 +4819,7 @@ Status DB::Open(const DBOptions& db_options, const std::string& dbname,
new WritableFileWriter(std::move(lfile), opt_env_options)); new WritableFileWriter(std::move(lfile), opt_env_options));
impl->logs_.emplace_back( impl->logs_.emplace_back(
new_log_number, new_log_number,
new log::Writer(std::move(file_writer), new log::Writer(std::move(file_writer), new_log_number,
new_log_number,
impl->db_options_.recycle_log_file_num > 0)); impl->db_options_.recycle_log_file_num > 0));
// set column family handles // set column family handles

127
db/db_test.cc
View File

@ -5073,8 +5073,8 @@ class RecoveryTestHelper {
ASSERT_OK(db_options.env->NewWritableFile(fname, &file, env_options)); ASSERT_OK(db_options.env->NewWritableFile(fname, &file, env_options));
unique_ptr<WritableFileWriter> file_writer( unique_ptr<WritableFileWriter> file_writer(
new WritableFileWriter(std::move(file), env_options)); new WritableFileWriter(std::move(file), env_options));
current_log_writer.reset(new log::Writer( current_log_writer.reset(
std::move(file_writer), current_log_number, new log::Writer(std::move(file_writer), current_log_number,
db_options.recycle_log_file_num > 0)); db_options.recycle_log_file_num > 0));
for (int i = 0; i < kKeysPerWALFile; i++) { for (int i = 0; i < kKeysPerWALFile; i++) {
@ -9891,15 +9891,13 @@ TEST_F(DBTest, PauseBackgroundWorkTest) {
#ifndef ROCKSDB_LITE #ifndef ROCKSDB_LITE
namespace { namespace {
void ValidateKeyExistence(DB* db, void ValidateKeyExistence(DB* db, const std::vector<Slice>& keys_must_exist,
const std::vector<Slice>& keys_must_exist,
const std::vector<Slice>& keys_must_not_exist) { const std::vector<Slice>& keys_must_not_exist) {
// Ensure that expected keys exist // Ensure that expected keys exist
std::vector<std::string> values; std::vector<std::string> values;
if (keys_must_exist.size() > 0) { if (keys_must_exist.size() > 0) {
std::vector<Status> status_list = db->MultiGet(ReadOptions(), std::vector<Status> status_list =
keys_must_exist, db->MultiGet(ReadOptions(), keys_must_exist, &values);
&values);
for (size_t i = 0; i < keys_must_exist.size(); i++) { for (size_t i = 0; i < keys_must_exist.size(); i++) {
ASSERT_OK(status_list[i]); ASSERT_OK(status_list[i]);
} }
@ -9907,16 +9905,15 @@ namespace {
// Ensure that given keys don't exist // Ensure that given keys don't exist
if (keys_must_not_exist.size() > 0) { if (keys_must_not_exist.size() > 0) {
std::vector<Status> status_list = db->MultiGet(ReadOptions(), std::vector<Status> status_list =
keys_must_not_exist, db->MultiGet(ReadOptions(), keys_must_not_exist, &values);
&values);
for (size_t i = 0; i < keys_must_not_exist.size(); i++) { for (size_t i = 0; i < keys_must_not_exist.size(); i++) {
ASSERT_TRUE(status_list[i].IsNotFound()); ASSERT_TRUE(status_list[i].IsNotFound());
} }
} }
} }
} //namespace } // namespace
TEST_F(DBTest, WalFilterTest) { TEST_F(DBTest, WalFilterTest) {
class TestWalFilter : public WalFilter { class TestWalFilter : public WalFilter {
@ -9929,21 +9926,22 @@ TEST_F(DBTest, WalFilterTest) {
size_t apply_option_at_record_index_; size_t apply_option_at_record_index_;
// Current record index, incremented with each record encountered. // Current record index, incremented with each record encountered.
size_t current_record_index_; size_t current_record_index_;
public: public:
TestWalFilter(WalFilter::WalProcessingOption wal_processing_option, TestWalFilter(WalFilter::WalProcessingOption wal_processing_option,
size_t apply_option_for_record_index) : size_t apply_option_for_record_index)
wal_processing_option_(wal_processing_option), : wal_processing_option_(wal_processing_option),
apply_option_at_record_index_(apply_option_for_record_index), apply_option_at_record_index_(apply_option_for_record_index),
current_record_index_(0) { } current_record_index_(0) {}
virtual WalProcessingOption LogRecord(const WriteBatch & batch, virtual WalProcessingOption LogRecord(const WriteBatch& batch,
WriteBatch* new_batch, bool* batch_changed) const override { WriteBatch* new_batch,
bool* batch_changed) const override {
WalFilter::WalProcessingOption option_to_return; WalFilter::WalProcessingOption option_to_return;
if (current_record_index_ == apply_option_at_record_index_) { if (current_record_index_ == apply_option_at_record_index_) {
option_to_return = wal_processing_option_; option_to_return = wal_processing_option_;
} } else {
else {
option_to_return = WalProcessingOption::kContinueProcessing; option_to_return = WalProcessingOption::kContinueProcessing;
} }
@ -9955,9 +9953,7 @@ TEST_F(DBTest, WalFilterTest) {
return option_to_return; return option_to_return;
} }
virtual const char* Name() const override { virtual const char* Name() const override { return "TestWalFilter"; }
return "TestWalFilter";
}
}; };
// Create 3 batches with two keys each // Create 3 batches with two keys each
@ -9972,11 +9968,12 @@ TEST_F(DBTest, WalFilterTest) {
// Test with all WAL processing options // Test with all WAL processing options
for (int option = 0; for (int option = 0;
option < static_cast<int>(WalFilter::WalProcessingOption::kWalProcessingOptionMax); option < static_cast<int>(
WalFilter::WalProcessingOption::kWalProcessingOptionMax);
option++) { option++) {
Options options = OptionsForLogIterTest(); Options options = OptionsForLogIterTest();
DestroyAndReopen(options); DestroyAndReopen(options);
CreateAndReopenWithCF({ "pikachu" }, options); CreateAndReopenWithCF({"pikachu"}, options);
// Write given keys in given batches // Write given keys in given batches
for (size_t i = 0; i < batch_keys.size(); i++) { for (size_t i = 0; i < batch_keys.size(); i++) {
@ -9999,15 +9996,15 @@ TEST_F(DBTest, WalFilterTest) {
// Reopen database with option to use WAL filter // Reopen database with option to use WAL filter
options = OptionsForLogIterTest(); options = OptionsForLogIterTest();
options.wal_filter = &test_wal_filter; options.wal_filter = &test_wal_filter;
Status status = TryReopenWithColumnFamilies({ "default", "pikachu" }, Status status =
options); TryReopenWithColumnFamilies({"default", "pikachu"}, options);
if (wal_processing_option == if (wal_processing_option ==
WalFilter::WalProcessingOption::kCorruptedRecord) { WalFilter::WalProcessingOption::kCorruptedRecord) {
assert(!status.ok()); assert(!status.ok());
// In case of corruption we can turn off paranoid_checks to reopen // In case of corruption we can turn off paranoid_checks to reopen
// databse // databse
options.paranoid_checks = false; options.paranoid_checks = false;
ReopenWithColumnFamilies({ "default", "pikachu" }, options); ReopenWithColumnFamilies({"default", "pikachu"}, options);
} else { } else {
assert(status.ok()); assert(status.ok());
} }
@ -10020,7 +10017,7 @@ TEST_F(DBTest, WalFilterTest) {
case WalFilter::WalProcessingOption::kCorruptedRecord: case WalFilter::WalProcessingOption::kCorruptedRecord:
case WalFilter::WalProcessingOption::kContinueProcessing: { case WalFilter::WalProcessingOption::kContinueProcessing: {
fprintf(stderr, "Testing with complete WAL processing\n"); fprintf(stderr, "Testing with complete WAL processing\n");
//we expect all records to be processed // we expect all records to be processed
for (size_t i = 0; i < batch_keys.size(); i++) { for (size_t i = 0; i < batch_keys.size(); i++) {
for (size_t j = 0; j < batch_keys[i].size(); j++) { for (size_t j = 0; j < batch_keys[i].size(); j++) {
keys_must_exist.push_back(Slice(batch_keys[i][j])); keys_must_exist.push_back(Slice(batch_keys[i][j]));
@ -10029,7 +10026,8 @@ TEST_F(DBTest, WalFilterTest) {
break; break;
} }
case WalFilter::WalProcessingOption::kIgnoreCurrentRecord: { case WalFilter::WalProcessingOption::kIgnoreCurrentRecord: {
fprintf(stderr, "Testing with ignoring record %" ROCKSDB_PRIszt " only\n", fprintf(stderr,
"Testing with ignoring record %" ROCKSDB_PRIszt " only\n",
apply_option_for_record_index); apply_option_for_record_index);
// We expect the record with apply_option_for_record_index to be not // We expect the record with apply_option_for_record_index to be not
// found. // found.
@ -10037,8 +10035,7 @@ TEST_F(DBTest, WalFilterTest) {
for (size_t j = 0; j < batch_keys[i].size(); j++) { for (size_t j = 0; j < batch_keys[i].size(); j++) {
if (i == apply_option_for_record_index) { if (i == apply_option_for_record_index) {
keys_must_not_exist.push_back(Slice(batch_keys[i][j])); keys_must_not_exist.push_back(Slice(batch_keys[i][j]));
} } else {
else {
keys_must_exist.push_back(Slice(batch_keys[i][j])); keys_must_exist.push_back(Slice(batch_keys[i][j]));
} }
} }
@ -10046,7 +10043,9 @@ TEST_F(DBTest, WalFilterTest) {
break; break;
} }
case WalFilter::WalProcessingOption::kStopReplay: { case WalFilter::WalProcessingOption::kStopReplay: {
fprintf(stderr, "Testing with stopping replay from record %" ROCKSDB_PRIszt "\n", fprintf(stderr,
"Testing with stopping replay from record %" ROCKSDB_PRIszt
"\n",
apply_option_for_record_index); apply_option_for_record_index);
// We expect records beyond apply_option_for_record_index to be not // We expect records beyond apply_option_for_record_index to be not
// found. // found.
@ -10054,8 +10053,7 @@ TEST_F(DBTest, WalFilterTest) {
for (size_t j = 0; j < batch_keys[i].size(); j++) { for (size_t j = 0; j < batch_keys[i].size(); j++) {
if (i >= apply_option_for_record_index) { if (i >= apply_option_for_record_index) {
keys_must_not_exist.push_back(Slice(batch_keys[i][j])); keys_must_not_exist.push_back(Slice(batch_keys[i][j]));
} } else {
else {
keys_must_exist.push_back(Slice(batch_keys[i][j])); keys_must_exist.push_back(Slice(batch_keys[i][j]));
} }
} }
@ -10063,7 +10061,7 @@ TEST_F(DBTest, WalFilterTest) {
break; break;
} }
default: default:
assert(false); //unhandled case assert(false); // unhandled case
} }
bool checked_after_reopen = false; bool checked_after_reopen = false;
@ -10077,11 +10075,11 @@ TEST_F(DBTest, WalFilterTest) {
break; break;
} }
//reopen database again to make sure previous log(s) are not used // reopen database again to make sure previous log(s) are not used
//(even if they were skipped) //(even if they were skipped)
//reopn database with option to use WAL filter // reopn database with option to use WAL filter
options = OptionsForLogIterTest(); options = OptionsForLogIterTest();
ReopenWithColumnFamilies({ "default", "pikachu" }, options); ReopenWithColumnFamilies({"default", "pikachu"}, options);
checked_after_reopen = true; checked_after_reopen = true;
} }
@ -10097,12 +10095,13 @@ TEST_F(DBTest, WalFilterTestWithChangeBatch) {
size_t num_keys_to_add_in_new_batch_; size_t num_keys_to_add_in_new_batch_;
// Number of keys added to new batch // Number of keys added to new batch
size_t num_keys_added_; size_t num_keys_added_;
public: public:
ChangeBatchHandler(WriteBatch* new_write_batch, ChangeBatchHandler(WriteBatch* new_write_batch,
size_t num_keys_to_add_in_new_batch) : size_t num_keys_to_add_in_new_batch)
new_write_batch_(new_write_batch), : new_write_batch_(new_write_batch),
num_keys_to_add_in_new_batch_(num_keys_to_add_in_new_batch), num_keys_to_add_in_new_batch_(num_keys_to_add_in_new_batch),
num_keys_added_(0){ } num_keys_added_(0) {}
virtual void Put(const Slice& key, const Slice& value) override { virtual void Put(const Slice& key, const Slice& value) override {
if (num_keys_added_ < num_keys_to_add_in_new_batch_) { if (num_keys_added_ < num_keys_to_add_in_new_batch_) {
new_write_batch_->Put(key, value); new_write_batch_->Put(key, value);
@ -10119,17 +10118,17 @@ TEST_F(DBTest, WalFilterTestWithChangeBatch) {
size_t num_keys_to_add_in_new_batch_; size_t num_keys_to_add_in_new_batch_;
// Current record index, incremented with each record encountered. // Current record index, incremented with each record encountered.
size_t current_record_index_; size_t current_record_index_;
public: public:
TestWalFilterWithChangeBatch( TestWalFilterWithChangeBatch(size_t change_records_from_index,
size_t change_records_from_index, size_t num_keys_to_add_in_new_batch)
size_t num_keys_to_add_in_new_batch) : : change_records_from_index_(change_records_from_index),
change_records_from_index_(change_records_from_index),
num_keys_to_add_in_new_batch_(num_keys_to_add_in_new_batch), num_keys_to_add_in_new_batch_(num_keys_to_add_in_new_batch),
current_record_index_(0) { } current_record_index_(0) {}
virtual WalProcessingOption LogRecord(const WriteBatch & batch,
WriteBatch* new_batch, bool* batch_changed) const override {
virtual WalProcessingOption LogRecord(const WriteBatch& batch,
WriteBatch* new_batch,
bool* batch_changed) const override {
if (current_record_index_ >= change_records_from_index_) { if (current_record_index_ >= change_records_from_index_) {
ChangeBatchHandler handler(new_batch, num_keys_to_add_in_new_batch_); ChangeBatchHandler handler(new_batch, num_keys_to_add_in_new_batch_);
batch.Iterate(&handler); batch.Iterate(&handler);
@ -10139,7 +10138,8 @@ TEST_F(DBTest, WalFilterTestWithChangeBatch) {
// Filter is passed as a const object for RocksDB to not modify the // Filter is passed as a const object for RocksDB to not modify the
// object, however we modify it for our own purpose here and hence // object, however we modify it for our own purpose here and hence
// cast the constness away. // cast the constness away.
(const_cast<TestWalFilterWithChangeBatch*>(this)->current_record_index_)++; (const_cast<TestWalFilterWithChangeBatch*>(this)
->current_record_index_)++;
return WalProcessingOption::kContinueProcessing; return WalProcessingOption::kContinueProcessing;
} }
@ -10160,7 +10160,7 @@ TEST_F(DBTest, WalFilterTestWithChangeBatch) {
Options options = OptionsForLogIterTest(); Options options = OptionsForLogIterTest();
DestroyAndReopen(options); DestroyAndReopen(options);
CreateAndReopenWithCF({ "pikachu" }, options); CreateAndReopenWithCF({"pikachu"}, options);
// Write given keys in given batches // Write given keys in given batches
for (size_t i = 0; i < batch_keys.size(); i++) { for (size_t i = 0; i < batch_keys.size(); i++) {
@ -10181,7 +10181,7 @@ TEST_F(DBTest, WalFilterTestWithChangeBatch) {
// Reopen database with option to use WAL filter // Reopen database with option to use WAL filter
options = OptionsForLogIterTest(); options = OptionsForLogIterTest();
options.wal_filter = &test_wal_filter_with_change_batch; options.wal_filter = &test_wal_filter_with_change_batch;
ReopenWithColumnFamilies({ "default", "pikachu" }, options); ReopenWithColumnFamilies({"default", "pikachu"}, options);
// Ensure that all keys exist before change_records_from_index_ // Ensure that all keys exist before change_records_from_index_
// And after that index only single key exists // And after that index only single key exists
@ -10193,8 +10193,7 @@ TEST_F(DBTest, WalFilterTestWithChangeBatch) {
for (size_t j = 0; j < batch_keys[i].size(); j++) { for (size_t j = 0; j < batch_keys[i].size(); j++) {
if (i >= change_records_from_index && j >= num_keys_to_add_in_new_batch) { if (i >= change_records_from_index && j >= num_keys_to_add_in_new_batch) {
keys_must_not_exist.push_back(Slice(batch_keys[i][j])); keys_must_not_exist.push_back(Slice(batch_keys[i][j]));
} } else {
else {
keys_must_exist.push_back(Slice(batch_keys[i][j])); keys_must_exist.push_back(Slice(batch_keys[i][j]));
} }
} }
@ -10211,11 +10210,11 @@ TEST_F(DBTest, WalFilterTestWithChangeBatch) {
break; break;
} }
//reopen database again to make sure previous log(s) are not used // reopen database again to make sure previous log(s) are not used
//(even if they were skipped) //(even if they were skipped)
//reopn database with option to use WAL filter // reopn database with option to use WAL filter
options = OptionsForLogIterTest(); options = OptionsForLogIterTest();
ReopenWithColumnFamilies({ "default", "pikachu" }, options); ReopenWithColumnFamilies({"default", "pikachu"}, options);
checked_after_reopen = true; checked_after_reopen = true;
} }
@ -10224,8 +10223,9 @@ TEST_F(DBTest, WalFilterTestWithChangeBatch) {
TEST_F(DBTest, WalFilterTestWithChangeBatchExtraKeys) { TEST_F(DBTest, WalFilterTestWithChangeBatchExtraKeys) {
class TestWalFilterWithChangeBatchAddExtraKeys : public WalFilter { class TestWalFilterWithChangeBatchAddExtraKeys : public WalFilter {
public: public:
virtual WalProcessingOption LogRecord(const WriteBatch & batch, virtual WalProcessingOption LogRecord(const WriteBatch& batch,
WriteBatch* new_batch, bool* batch_changed) const override { WriteBatch* new_batch,
bool* batch_changed) const override {
*new_batch = batch; *new_batch = batch;
new_batch->Put("key_extra", "value_extra"); new_batch->Put("key_extra", "value_extra");
*batch_changed = true; *batch_changed = true;
@ -10248,7 +10248,7 @@ TEST_F(DBTest, WalFilterTestWithChangeBatchExtraKeys) {
Options options = OptionsForLogIterTest(); Options options = OptionsForLogIterTest();
DestroyAndReopen(options); DestroyAndReopen(options);
CreateAndReopenWithCF({ "pikachu" }, options); CreateAndReopenWithCF({"pikachu"}, options);
// Write given keys in given batches // Write given keys in given batches
for (size_t i = 0; i < batch_keys.size(); i++) { for (size_t i = 0; i < batch_keys.size(); i++) {
@ -10265,17 +10265,16 @@ TEST_F(DBTest, WalFilterTestWithChangeBatchExtraKeys) {
// Reopen database with option to use WAL filter // Reopen database with option to use WAL filter
options = OptionsForLogIterTest(); options = OptionsForLogIterTest();
options.wal_filter = &test_wal_filter_extra_keys; options.wal_filter = &test_wal_filter_extra_keys;
Status status = Status status = TryReopenWithColumnFamilies({"default", "pikachu"}, options);
TryReopenWithColumnFamilies({ "default", "pikachu" }, options);
ASSERT_TRUE(status.IsNotSupported()); ASSERT_TRUE(status.IsNotSupported());
// Reopen without filter, now reopen should succeed - previous // Reopen without filter, now reopen should succeed - previous
// attempt to open must not have altered the db. // attempt to open must not have altered the db.
options = OptionsForLogIterTest(); options = OptionsForLogIterTest();
ReopenWithColumnFamilies({ "default", "pikachu" }, options); ReopenWithColumnFamilies({"default", "pikachu"}, options);
std::vector<Slice> keys_must_exist; std::vector<Slice> keys_must_exist;
std::vector<Slice> keys_must_not_exist; //empty vector std::vector<Slice> keys_must_not_exist; // empty vector
for (size_t i = 0; i < batch_keys.size(); i++) { for (size_t i = 0; i < batch_keys.size(); i++) {
for (size_t j = 0; j < batch_keys[i].size(); j++) { for (size_t j = 0; j < batch_keys[i].size(); j++) {

6
include/rocksdb/db.h
View File

@ -503,7 +503,8 @@ class DB {
return CompactRange(options, DefaultColumnFamily(), begin, end); return CompactRange(options, DefaultColumnFamily(), begin, end);
} }
virtual Status SetOptions(ColumnFamilyHandle* /*column_family*/, virtual Status SetOptions(
ColumnFamilyHandle* /*column_family*/,
const std::unordered_map<std::string, std::string>& /*new_options*/) { const std::unordered_map<std::string, std::string>& /*new_options*/) {
return Status::NotSupported("Not implemented"); return Status::NotSupported("Not implemented");
} }
@ -663,8 +664,7 @@ class DB {
// //
// If cf_name is not specified, then the metadata of the default // If cf_name is not specified, then the metadata of the default
// column family will be returned. // column family will be returned.
virtual void GetColumnFamilyMetaData( virtual void GetColumnFamilyMetaData(ColumnFamilyHandle* /*column_family*/,
ColumnFamilyHandle* /*column_family*/,
ColumnFamilyMetaData* /*metadata*/) {} ColumnFamilyMetaData* /*metadata*/) {}
// Get the metadata of the default column family. // Get the metadata of the default column family.

3
include/rocksdb/env.h
View File

@ -416,8 +416,7 @@ class RandomAccessFile {
// For cases when read-ahead is implemented in the platform dependent // For cases when read-ahead is implemented in the platform dependent
// layer // layer
virtual void EnableReadAhead() { virtual void EnableReadAhead() {}
}
// Tries to get an unique ID for this file that will be the same each time // Tries to get an unique ID for this file that will be the same each time
// the file is opened (and will stay the same while the file is open). // the file is opened (and will stay the same while the file is open).

2
include/rocksdb/options.h
View File

@ -1163,7 +1163,7 @@ struct DBOptions {
// The filter is invoked at startup and is invoked from a single-thread // The filter is invoked at startup and is invoked from a single-thread
// currently. // currently.
const WalFilter* wal_filter; const WalFilter* wal_filter;
#endif //ROCKSDB_LITE #endif // ROCKSDB_LITE
}; };
// Options to control the behavior of a database (passed to DB::Open) // Options to control the behavior of a database (passed to DB::Open)

7
include/rocksdb/wal_filter.h
View File

@ -13,7 +13,7 @@ class WriteBatch;
// records or modify their processing on recovery. // records or modify their processing on recovery.
// Please see the details below. // Please see the details below.
class WalFilter { class WalFilter {
public: public:
enum class WalProcessingOption { enum class WalProcessingOption {
// Continue processing as usual // Continue processing as usual
kContinueProcessing = 0, kContinueProcessing = 0,
@ -28,7 +28,7 @@ public:
kWalProcessingOptionMax = 4 kWalProcessingOptionMax = 4
}; };
virtual ~WalFilter() { }; virtual ~WalFilter() {}
// LogRecord is invoked for each log record encountered for all the logs // LogRecord is invoked for each log record encountered for all the logs
// during replay on logs on recovery. This method can be used to: // during replay on logs on recovery. This method can be used to:
@ -55,7 +55,8 @@ public:
// Please see WalProcessingOption enum above for // Please see WalProcessingOption enum above for
// details. // details.
virtual WalProcessingOption LogRecord(const WriteBatch& batch, virtual WalProcessingOption LogRecord(const WriteBatch& batch,
WriteBatch* new_batch, bool* batch_changed) const = 0; WriteBatch* new_batch,
bool* batch_changed) const = 0;
// Returns a name that identifies this WAL filter. // Returns a name that identifies this WAL filter.
// The name will be printed to LOG file on start up for diagnosis. // The name will be printed to LOG file on start up for diagnosis.

58
port/win/env_win.cc
View File

@ -697,30 +697,35 @@ class WinRandomAccessFile : public RandomAccessFile {
buffered_start_; // file offset set that is currently buffered buffered_start_; // file offset set that is currently buffered
/* /*
* The function reads a requested amount of bytes into the specified aligned buffer * The function reads a requested amount of bytes into the specified aligned
* Upon success the function sets the length of the buffer to the amount of bytes actually * buffer Upon success the function sets the length of the buffer to the
* read even though it might be less than actually requested. * amount of bytes actually read even though it might be less than actually
* It then copies the amount of bytes requested by the user (left) to the user supplied * requested. It then copies the amount of bytes requested by the user (left)
* buffer (dest) and reduces left by the amount of bytes copied to the user buffer * to the user supplied buffer (dest) and reduces left by the amount of bytes
* copied to the user buffer
* *
* @user_offset [in] - offset on disk where the read was requested by the user * @user_offset [in] - offset on disk where the read was requested by the user
* @first_page_start [in] - actual page aligned disk offset that we want to read from * @first_page_start [in] - actual page aligned disk offset that we want to
* @bytes_to_read [in] - total amount of bytes that will be read from disk which is generally * read from
* greater or equal to the amount that the user has requested due to the * @bytes_to_read [in] - total amount of bytes that will be read from disk
* either alignment requirements or read_ahead in effect. * which is generally greater or equal to the amount
* @left [in/out] total amount of bytes that needs to be copied to the user buffer. It is reduced * that the user has requested due to the
* by the amount of bytes that actually copied * either alignment requirements or read_ahead in
* effect.
* @left [in/out] total amount of bytes that needs to be copied to the user
* buffer. It is reduced by the amount of bytes that actually
* copied
* @buffer - buffer to use * @buffer - buffer to use
* @dest - user supplied buffer * @dest - user supplied buffer
*/ */
SSIZE_T ReadIntoBuffer(uint64_t user_offset, uint64_t first_page_start, SSIZE_T ReadIntoBuffer(uint64_t user_offset, uint64_t first_page_start,
size_t bytes_to_read, size_t& left, AlignedBuffer& buffer, char* dest) const { size_t bytes_to_read, size_t& left,
AlignedBuffer& buffer, char* dest) const {
assert(buffer.CurrentSize() == 0); assert(buffer.CurrentSize() == 0);
assert(buffer.Capacity() >= bytes_to_read); assert(buffer.Capacity() >= bytes_to_read);
SSIZE_T read = pread(hFile_, buffer.Destination(), bytes_to_read, SSIZE_T read =
first_page_start); pread(hFile_, buffer.Destination(), bytes_to_read, first_page_start);
if (read > 0) { if (read > 0) {
buffer.Size(read); buffer.Size(read);
@ -739,8 +744,8 @@ class WinRandomAccessFile : public RandomAccessFile {
} }
SSIZE_T ReadIntoOneShotBuffer(uint64_t user_offset, uint64_t first_page_start, SSIZE_T ReadIntoOneShotBuffer(uint64_t user_offset, uint64_t first_page_start,
size_t bytes_to_read, size_t& left, char* dest) const { size_t bytes_to_read, size_t& left,
char* dest) const {
AlignedBuffer bigBuffer; AlignedBuffer bigBuffer;
bigBuffer.Alignment(buffer_.Alignment()); bigBuffer.Alignment(buffer_.Alignment());
bigBuffer.AllocateNewBuffer(bytes_to_read); bigBuffer.AllocateNewBuffer(bytes_to_read);
@ -749,9 +754,10 @@ class WinRandomAccessFile : public RandomAccessFile {
bigBuffer, dest); bigBuffer, dest);
} }
SSIZE_T ReadIntoInstanceBuffer(uint64_t user_offset, uint64_t first_page_start, SSIZE_T ReadIntoInstanceBuffer(uint64_t user_offset,
size_t bytes_to_read, size_t& left, char* dest) const { uint64_t first_page_start,
size_t bytes_to_read, size_t& left,
char* dest) const {
SSIZE_T read = ReadIntoBuffer(user_offset, first_page_start, bytes_to_read, SSIZE_T read = ReadIntoBuffer(user_offset, first_page_start, bytes_to_read,
left, buffer_, dest); left, buffer_, dest);
@ -789,9 +795,7 @@ class WinRandomAccessFile : public RandomAccessFile {
} }
} }
virtual void EnableReadAhead() override { virtual void EnableReadAhead() override { this->Hint(SEQUENTIAL); }
this->Hint(SEQUENTIAL);
}
virtual Status Read(uint64_t offset, size_t n, Slice* result, virtual Status Read(uint64_t offset, size_t n, Slice* result,
char* scratch) const override { char* scratch) const override {
@ -877,9 +881,7 @@ class WinRandomAccessFile : public RandomAccessFile {
} }
virtual void Hint(AccessPattern pattern) override { virtual void Hint(AccessPattern pattern) override {
if (pattern == SEQUENTIAL && !use_os_buffer_ &&
if (pattern == SEQUENTIAL &&
!use_os_buffer_ &&
compaction_readahead_size_ > 0) { compaction_readahead_size_ > 0) {
std::lock_guard<std::mutex> lg(buffer_mut_); std::lock_guard<std::mutex> lg(buffer_mut_);
if (!read_ahead_) { if (!read_ahead_) {
@ -888,12 +890,12 @@ class WinRandomAccessFile : public RandomAccessFile {
// - one for memory alignment which added implicitly by AlignedBuffer // - one for memory alignment which added implicitly by AlignedBuffer
// - We add one more alignment because we will read one alignment more // - We add one more alignment because we will read one alignment more
// from disk // from disk
buffer_.AllocateNewBuffer(compaction_readahead_size_ + buffer_.Alignment()); buffer_.AllocateNewBuffer(compaction_readahead_size_ +
buffer_.Alignment());
} }
} }
} }
virtual Status InvalidateCache(size_t offset, size_t length) override { virtual Status InvalidateCache(size_t offset, size_t length) override {
return Status::OK(); return Status::OK();
} }

3
util/env.cc
View File

@ -293,7 +293,8 @@ void AssignEnvOptions(EnvOptions* env_options, const DBOptions& options) {
env_options->set_fd_cloexec = options.is_fd_close_on_exec; env_options->set_fd_cloexec = options.is_fd_close_on_exec;
env_options->bytes_per_sync = options.bytes_per_sync; env_options->bytes_per_sync = options.bytes_per_sync;
env_options->compaction_readahead_size = options.compaction_readahead_size; env_options->compaction_readahead_size = options.compaction_readahead_size;
env_options->random_access_max_buffer_size = options.random_access_max_buffer_size; env_options->random_access_max_buffer_size =
options.random_access_max_buffer_size;
env_options->rate_limiter = options.rate_limiter.get(); env_options->rate_limiter = options.rate_limiter.get();
env_options->allow_fallocate = options.allow_fallocate; env_options->allow_fallocate = options.allow_fallocate;
} }

13
util/options.cc
View File

@ -260,9 +260,10 @@ DBOptions::DBOptions()
skip_stats_update_on_db_open(false), skip_stats_update_on_db_open(false),
wal_recovery_mode(WALRecoveryMode::kTolerateCorruptedTailRecords) wal_recovery_mode(WALRecoveryMode::kTolerateCorruptedTailRecords)
#ifndef ROCKSDB_LITE #ifndef ROCKSDB_LITE
, wal_filter(nullptr) ,
wal_filter(nullptr)
#endif // ROCKSDB_LITE #endif // ROCKSDB_LITE
{ {
} }
DBOptions::DBOptions(const Options& options) DBOptions::DBOptions(const Options& options)
@ -322,9 +323,10 @@ DBOptions::DBOptions(const Options& options)
wal_recovery_mode(options.wal_recovery_mode), wal_recovery_mode(options.wal_recovery_mode),
row_cache(options.row_cache) row_cache(options.row_cache)
#ifndef ROCKSDB_LITE #ifndef ROCKSDB_LITE
, wal_filter(options.wal_filter) ,
wal_filter(options.wal_filter)
#endif // ROCKSDB_LITE #endif // ROCKSDB_LITE
{ {
} }
static const char* const access_hints[] = { static const char* const access_hints[] = {
@ -405,7 +407,8 @@ void DBOptions::Dump(Logger* log) const {
" Options.compaction_readahead_size: %" ROCKSDB_PRIszt " Options.compaction_readahead_size: %" ROCKSDB_PRIszt
"d", "d",
compaction_readahead_size); compaction_readahead_size);
Header(log, Header(
log,
" Options.random_access_max_buffer_size: %" ROCKSDB_PRIszt " Options.random_access_max_buffer_size: %" ROCKSDB_PRIszt
"d", "d",
random_access_max_buffer_size); random_access_max_buffer_size);

4
util/options_helper.h
View File

@ -181,8 +181,8 @@ static std::unordered_map<std::string, OptionTypeInfo> db_options_type_info = {
{offsetof(struct DBOptions, compaction_readahead_size), OptionType::kSizeT, {offsetof(struct DBOptions, compaction_readahead_size), OptionType::kSizeT,
OptionVerificationType::kNormal}}, OptionVerificationType::kNormal}},
{"random_access_max_buffer_size", {"random_access_max_buffer_size",
{ offsetof(struct DBOptions, random_access_max_buffer_size), OptionType::kSizeT, {offsetof(struct DBOptions, random_access_max_buffer_size),
OptionVerificationType::kNormal}}, OptionType::kSizeT, OptionVerificationType::kNormal}},
{"use_adaptive_mutex", {"use_adaptive_mutex",
{offsetof(struct DBOptions, use_adaptive_mutex), OptionType::kBoolean, {offsetof(struct DBOptions, use_adaptive_mutex), OptionType::kBoolean,
OptionVerificationType::kNormal}}, OptionVerificationType::kNormal}},

2
util/options_test.cc
View File

@ -339,7 +339,7 @@ TEST_F(OptionsTest, GetOptionsFromMapTest) {
{"use_adaptive_mutex", "false"}, {"use_adaptive_mutex", "false"},
{"new_table_reader_for_compaction_inputs", "true"}, {"new_table_reader_for_compaction_inputs", "true"},
{"compaction_readahead_size", "100"}, {"compaction_readahead_size", "100"},
{"random_access_max_buffer_size", "3145728" }, {"random_access_max_buffer_size", "3145728"},
{"bytes_per_sync", "47"}, {"bytes_per_sync", "47"},
{"wal_bytes_per_sync", "48"}, {"wal_bytes_per_sync", "48"},
}; };