2016-02-10 00:12:00 +01:00
|
|
|
// Copyright (c) 2011-present, Facebook, Inc. All rights reserved.
|
2017-07-16 01:03:42 +02:00
|
|
|
// 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).
|
2014-10-29 01:52:32 +01:00
|
|
|
|
2015-09-02 22:58:22 +02:00
|
|
|
#include "table/mock_table.h"
|
|
|
|
|
2014-10-29 01:52:32 +01:00
|
|
|
#include "db/dbformat.h"
|
|
|
|
#include "port/port.h"
|
2015-08-08 06:59:51 +02:00
|
|
|
#include "rocksdb/table_properties.h"
|
|
|
|
#include "table/get_context.h"
|
2014-10-29 01:52:32 +01:00
|
|
|
#include "util/coding.h"
|
Move rate_limiter, write buffering, most perf context instrumentation and most random kill out of Env
Summary: We want to keep Env a think layer for better portability. Less platform dependent codes should be moved out of Env. In this patch, I create a wrapper of file readers and writers, and put rate limiting, write buffering, as well as most perf context instrumentation and random kill out of Env. It will make it easier to maintain multiple Env in the future.
Test Plan: Run all existing unit tests.
Reviewers: anthony, kradhakrishnan, IslamAbdelRahman, yhchiang, igor
Reviewed By: igor
Subscribers: leveldb, dhruba
Differential Revision: https://reviews.facebook.net/D42321
2015-07-18 01:16:11 +02:00
|
|
|
#include "util/file_reader_writer.h"
|
2014-10-29 01:52:32 +01:00
|
|
|
|
|
|
|
namespace rocksdb {
|
2014-11-14 20:35:48 +01:00
|
|
|
namespace mock {
|
2014-10-29 01:52:32 +01:00
|
|
|
|
2015-09-02 22:58:22 +02:00
|
|
|
namespace {
|
|
|
|
|
|
|
|
const InternalKeyComparator icmp_(BytewiseComparator());
|
|
|
|
|
|
|
|
} // namespace
|
|
|
|
|
|
|
|
stl_wrappers::KVMap MakeMockFile(
|
|
|
|
std::initializer_list<std::pair<const std::string, std::string>> l) {
|
|
|
|
return stl_wrappers::KVMap(l, stl_wrappers::LessOfComparator(&icmp_));
|
|
|
|
}
|
|
|
|
|
2018-05-21 23:33:55 +02:00
|
|
|
InternalIterator* MockTableReader::NewIterator(
|
|
|
|
const ReadOptions&, const SliceTransform* /* prefix_extractor */,
|
2018-06-25 22:07:38 +02:00
|
|
|
Arena* /*arena*/, bool /*skip_filters*/, bool /*for_compaction*/) {
|
2014-10-29 01:52:32 +01:00
|
|
|
return new MockTableIterator(table_);
|
|
|
|
}
|
|
|
|
|
|
|
|
Status MockTableReader::Get(const ReadOptions&, const Slice& key,
|
2018-05-21 23:33:55 +02:00
|
|
|
GetContext* get_context,
|
|
|
|
const SliceTransform* /*prefix_extractor*/,
|
|
|
|
bool /*skip_filters*/) {
|
2014-10-29 01:52:32 +01:00
|
|
|
std::unique_ptr<MockTableIterator> iter(new MockTableIterator(table_));
|
|
|
|
for (iter->Seek(key); iter->Valid(); iter->Next()) {
|
|
|
|
ParsedInternalKey parsed_key;
|
|
|
|
if (!ParseInternalKey(iter->key(), &parsed_key)) {
|
|
|
|
return Status::Corruption(Slice());
|
|
|
|
}
|
|
|
|
|
2018-04-06 00:54:24 +02:00
|
|
|
bool dont_care __attribute__((__unused__));
|
|
|
|
if (!get_context->SaveValue(parsed_key, iter->value(), &dont_care)) {
|
2014-10-29 01:52:32 +01:00
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return Status::OK();
|
|
|
|
}
|
|
|
|
|
|
|
|
std::shared_ptr<const TableProperties> MockTableReader::GetTableProperties()
|
|
|
|
const {
|
|
|
|
return std::shared_ptr<const TableProperties>(new TableProperties());
|
|
|
|
}
|
|
|
|
|
|
|
|
MockTableFactory::MockTableFactory() : next_id_(1) {}
|
|
|
|
|
|
|
|
Status MockTableFactory::NewTableReader(
|
2018-03-05 22:08:17 +01:00
|
|
|
const TableReaderOptions& /*table_reader_options*/,
|
|
|
|
unique_ptr<RandomAccessFileReader>&& file, uint64_t /*file_size*/,
|
2016-07-20 20:23:31 +02:00
|
|
|
unique_ptr<TableReader>* table_reader,
|
2018-03-05 22:08:17 +01:00
|
|
|
bool /*prefetch_index_and_filter_in_cache*/) const {
|
2014-10-29 01:52:32 +01:00
|
|
|
uint32_t id = GetIDFromFile(file.get());
|
|
|
|
|
|
|
|
MutexLock lock_guard(&file_system_.mutex);
|
|
|
|
|
|
|
|
auto it = file_system_.files.find(id);
|
|
|
|
if (it == file_system_.files.end()) {
|
|
|
|
return Status::IOError("Mock file not found");
|
|
|
|
}
|
|
|
|
|
|
|
|
table_reader->reset(new MockTableReader(it->second));
|
|
|
|
|
|
|
|
return Status::OK();
|
|
|
|
}
|
|
|
|
|
|
|
|
TableBuilder* MockTableFactory::NewTableBuilder(
|
2018-03-05 22:08:17 +01:00
|
|
|
const TableBuilderOptions& /*table_builder_options*/,
|
|
|
|
uint32_t /*column_family_id*/, WritableFileWriter* file) const {
|
2015-10-28 18:53:14 +01:00
|
|
|
uint32_t id = GetAndWriteNextID(file);
|
2014-10-29 01:52:32 +01:00
|
|
|
|
2015-04-14 10:55:19 +02:00
|
|
|
return new MockTableBuilder(id, &file_system_);
|
2014-10-29 01:52:32 +01:00
|
|
|
}
|
|
|
|
|
2014-11-14 20:35:48 +01:00
|
|
|
Status MockTableFactory::CreateMockTable(Env* env, const std::string& fname,
|
2015-09-02 22:58:22 +02:00
|
|
|
stl_wrappers::KVMap file_contents) {
|
2014-11-14 20:35:48 +01:00
|
|
|
std::unique_ptr<WritableFile> file;
|
|
|
|
auto s = env->NewWritableFile(fname, &file, EnvOptions());
|
|
|
|
if (!s.ok()) {
|
|
|
|
return s;
|
|
|
|
}
|
|
|
|
|
2018-08-23 19:04:10 +02:00
|
|
|
WritableFileWriter file_writer(std::move(file), fname, EnvOptions());
|
2015-10-28 18:53:14 +01:00
|
|
|
|
|
|
|
uint32_t id = GetAndWriteNextID(&file_writer);
|
2014-11-14 20:35:48 +01:00
|
|
|
file_system_.files.insert({id, std::move(file_contents)});
|
|
|
|
return Status::OK();
|
|
|
|
}
|
|
|
|
|
2015-10-28 18:53:14 +01:00
|
|
|
uint32_t MockTableFactory::GetAndWriteNextID(WritableFileWriter* file) const {
|
2014-10-29 01:52:32 +01:00
|
|
|
uint32_t next_id = next_id_.fetch_add(1);
|
|
|
|
char buf[4];
|
|
|
|
EncodeFixed32(buf, next_id);
|
|
|
|
file->Append(Slice(buf, 4));
|
|
|
|
return next_id;
|
|
|
|
}
|
|
|
|
|
Move rate_limiter, write buffering, most perf context instrumentation and most random kill out of Env
Summary: We want to keep Env a think layer for better portability. Less platform dependent codes should be moved out of Env. In this patch, I create a wrapper of file readers and writers, and put rate limiting, write buffering, as well as most perf context instrumentation and random kill out of Env. It will make it easier to maintain multiple Env in the future.
Test Plan: Run all existing unit tests.
Reviewers: anthony, kradhakrishnan, IslamAbdelRahman, yhchiang, igor
Reviewed By: igor
Subscribers: leveldb, dhruba
Differential Revision: https://reviews.facebook.net/D42321
2015-07-18 01:16:11 +02:00
|
|
|
uint32_t MockTableFactory::GetIDFromFile(RandomAccessFileReader* file) const {
|
2014-10-29 01:52:32 +01:00
|
|
|
char buf[4];
|
|
|
|
Slice result;
|
|
|
|
file->Read(0, 4, &result, buf);
|
|
|
|
assert(result.size() == 4);
|
|
|
|
return DecodeFixed32(buf);
|
|
|
|
}
|
|
|
|
|
2015-09-02 22:58:22 +02:00
|
|
|
void MockTableFactory::AssertSingleFile(
|
|
|
|
const stl_wrappers::KVMap& file_contents) {
|
2014-10-29 01:52:32 +01:00
|
|
|
ASSERT_EQ(file_system_.files.size(), 1U);
|
2016-11-01 04:35:54 +01:00
|
|
|
ASSERT_EQ(file_contents, file_system_.files.begin()->second);
|
2014-10-29 01:52:32 +01:00
|
|
|
}
|
|
|
|
|
2015-09-02 22:58:22 +02:00
|
|
|
void MockTableFactory::AssertLatestFile(
|
|
|
|
const stl_wrappers::KVMap& file_contents) {
|
2014-11-14 20:35:48 +01:00
|
|
|
ASSERT_GE(file_system_.files.size(), 1U);
|
|
|
|
auto latest = file_system_.files.end();
|
|
|
|
--latest;
|
2015-09-02 22:58:22 +02:00
|
|
|
|
|
|
|
if (file_contents != latest->second) {
|
|
|
|
std::cout << "Wrong content! Content of latest file:" << std::endl;
|
|
|
|
for (const auto& kv : latest->second) {
|
|
|
|
ParsedInternalKey ikey;
|
|
|
|
std::string key, value;
|
|
|
|
std::tie(key, value) = kv;
|
|
|
|
ParseInternalKey(Slice(key), &ikey);
|
|
|
|
std::cout << ikey.DebugString(false) << " -> " << value << std::endl;
|
|
|
|
}
|
2017-07-17 06:23:33 +02:00
|
|
|
FAIL();
|
2015-09-02 22:58:22 +02:00
|
|
|
}
|
2014-11-14 20:35:48 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
} // namespace mock
|
2014-10-29 01:52:32 +01:00
|
|
|
} // namespace rocksdb
|