Igor Canadi
11 years ago
commit
1e0d47276c
@ -0,0 +1,201 @@ |
|||||||
|
// Copyright (c) 2013, Facebook, Inc. All rights reserved.
|
||||||
|
// This source code is licensed under the BSD-style license found in the
|
||||||
|
// LICENSE file in the root directory of this source tree. An additional grant
|
||||||
|
// of patent rights can be found in the PATENTS file in the same directory.
|
||||||
|
|
||||||
|
#include <cstdio> |
||||||
|
#include <vector> |
||||||
|
#include <memory> |
||||||
|
|
||||||
|
#include "include/rocksdb/db.h" |
||||||
|
#include "include/rocksdb/options.h" |
||||||
|
#include "include/rocksdb/env.h" |
||||||
|
#include "include/rocksdb/slice.h" |
||||||
|
#include "include/rocksdb/status.h" |
||||||
|
#include "include/rocksdb/comparator.h" |
||||||
|
#include "include/rocksdb/table.h" |
||||||
|
#include "include/rocksdb/slice_transform.h" |
||||||
|
|
||||||
|
namespace rocksdb { |
||||||
|
|
||||||
|
class SanityTest { |
||||||
|
public: |
||||||
|
explicit SanityTest(const std::string& path) |
||||||
|
: env_(Env::Default()), path_(path) { |
||||||
|
env_->CreateDirIfMissing(path); |
||||||
|
} |
||||||
|
virtual ~SanityTest() {} |
||||||
|
|
||||||
|
virtual std::string Name() const = 0; |
||||||
|
virtual Options GetOptions() const = 0; |
||||||
|
|
||||||
|
Status Create() { |
||||||
|
Options options = GetOptions(); |
||||||
|
options.create_if_missing = true; |
||||||
|
std::string dbname = path_ + Name(); |
||||||
|
DestroyDB(dbname, options); |
||||||
|
DB* db; |
||||||
|
Status s = DB::Open(options, dbname, &db); |
||||||
|
std::unique_ptr<DB> db_guard(db); |
||||||
|
if (!s.ok()) { |
||||||
|
return s; |
||||||
|
} |
||||||
|
for (int i = 0; i < 1000000; ++i) { |
||||||
|
std::string k = "key" + std::to_string(i); |
||||||
|
std::string v = "value" + std::to_string(i); |
||||||
|
s = db->Put(WriteOptions(), Slice(k), Slice(v)); |
||||||
|
if (!s.ok()) { |
||||||
|
return s; |
||||||
|
} |
||||||
|
} |
||||||
|
return Status::OK(); |
||||||
|
} |
||||||
|
Status Verify() { |
||||||
|
DB* db; |
||||||
|
std::string dbname = path_ + Name(); |
||||||
|
Status s = DB::Open(GetOptions(), dbname, &db); |
||||||
|
std::unique_ptr<DB> db_guard(db); |
||||||
|
if (!s.ok()) { |
||||||
|
return s; |
||||||
|
} |
||||||
|
for (int i = 0; i < 1000000; ++i) { |
||||||
|
std::string k = "key" + std::to_string(i); |
||||||
|
std::string v = "value" + std::to_string(i); |
||||||
|
std::string result; |
||||||
|
s = db->Get(ReadOptions(), Slice(k), &result); |
||||||
|
if (!s.ok()) { |
||||||
|
return s; |
||||||
|
} |
||||||
|
if (result != v) { |
||||||
|
return Status::Corruption("Unexpected value for key " + k); |
||||||
|
} |
||||||
|
} |
||||||
|
return Status::OK(); |
||||||
|
} |
||||||
|
|
||||||
|
private: |
||||||
|
Env* env_; |
||||||
|
std::string const path_; |
||||||
|
}; |
||||||
|
|
||||||
|
class SanityTestBasic : public SanityTest { |
||||||
|
public: |
||||||
|
explicit SanityTestBasic(const std::string& path) : SanityTest(path) {} |
||||||
|
virtual Options GetOptions() const { |
||||||
|
Options options; |
||||||
|
options.create_if_missing = true; |
||||||
|
return options; |
||||||
|
} |
||||||
|
virtual std::string Name() const { return "Basic"; } |
||||||
|
}; |
||||||
|
|
||||||
|
class SanityTestSpecialComparator : public SanityTest { |
||||||
|
public: |
||||||
|
explicit SanityTestSpecialComparator(const std::string& path) |
||||||
|
: SanityTest(path) { |
||||||
|
options_.comparator = new NewComparator(); |
||||||
|
} |
||||||
|
~SanityTestSpecialComparator() { delete options_.comparator; } |
||||||
|
virtual Options GetOptions() const { return options_; } |
||||||
|
virtual std::string Name() const { return "SpecialComparator"; } |
||||||
|
|
||||||
|
private: |
||||||
|
class NewComparator : public Comparator { |
||||||
|
public: |
||||||
|
virtual const char* Name() const { return "rocksdb.NewComparator"; } |
||||||
|
virtual int Compare(const Slice& a, const Slice& b) const { |
||||||
|
return BytewiseComparator()->Compare(a, b); |
||||||
|
} |
||||||
|
virtual void FindShortestSeparator(std::string* s, const Slice& l) const { |
||||||
|
BytewiseComparator()->FindShortestSeparator(s, l); |
||||||
|
} |
||||||
|
virtual void FindShortSuccessor(std::string* key) const { |
||||||
|
BytewiseComparator()->FindShortSuccessor(key); |
||||||
|
} |
||||||
|
}; |
||||||
|
Options options_; |
||||||
|
}; |
||||||
|
|
||||||
|
class SanityTestZlibCompression : public SanityTest { |
||||||
|
public: |
||||||
|
explicit SanityTestZlibCompression(const std::string& path) |
||||||
|
: SanityTest(path) { |
||||||
|
options_.compression = kZlibCompression; |
||||||
|
} |
||||||
|
virtual Options GetOptions() const { return options_; } |
||||||
|
virtual std::string Name() const { return "ZlibCompression"; } |
||||||
|
|
||||||
|
private: |
||||||
|
Options options_; |
||||||
|
}; |
||||||
|
|
||||||
|
class SanityTestPlainTableFactory : public SanityTest { |
||||||
|
public: |
||||||
|
explicit SanityTestPlainTableFactory(const std::string& path) |
||||||
|
: SanityTest(path) { |
||||||
|
options_.table_factory.reset(NewPlainTableFactory()); |
||||||
|
options_.prefix_extractor = NewFixedPrefixTransform(2); |
||||||
|
options_.allow_mmap_reads = true; |
||||||
|
} |
||||||
|
~SanityTestPlainTableFactory() { delete options_.prefix_extractor; } |
||||||
|
virtual Options GetOptions() const { return options_; } |
||||||
|
virtual std::string Name() const { return "PlainTable"; } |
||||||
|
|
||||||
|
private: |
||||||
|
Options options_; |
||||||
|
}; |
||||||
|
|
||||||
|
bool RunSanityTests(const std::string& command, const std::string& path) { |
||||||
|
std::vector<SanityTest*> sanity_tests = { |
||||||
|
new SanityTestBasic(path), |
||||||
|
new SanityTestSpecialComparator(path), |
||||||
|
new SanityTestZlibCompression(path), |
||||||
|
new SanityTestPlainTableFactory(path)}; |
||||||
|
|
||||||
|
if (command == "create") { |
||||||
|
fprintf(stderr, "Creating...\n"); |
||||||
|
} else { |
||||||
|
fprintf(stderr, "Verifying...\n"); |
||||||
|
} |
||||||
|
for (auto sanity_test : sanity_tests) { |
||||||
|
Status s; |
||||||
|
fprintf(stderr, "%s -- ", sanity_test->Name().c_str()); |
||||||
|
if (command == "create") { |
||||||
|
s = sanity_test->Create(); |
||||||
|
} else { |
||||||
|
assert(command == "verify"); |
||||||
|
s = sanity_test->Verify(); |
||||||
|
} |
||||||
|
fprintf(stderr, "%s\n", s.ToString().c_str()); |
||||||
|
if (!s.ok()) { |
||||||
|
fprintf(stderr, "FAIL\n"); |
||||||
|
return false; |
||||||
|
} |
||||||
|
|
||||||
|
delete sanity_test; |
||||||
|
} |
||||||
|
return true; |
||||||
|
} |
||||||
|
|
||||||
|
} // namespace rocksdb
|
||||||
|
|
||||||
|
int main(int argc, char** argv) { |
||||||
|
std::string path, command; |
||||||
|
bool ok = (argc == 3); |
||||||
|
if (ok) { |
||||||
|
path = std::string(argv[1]); |
||||||
|
command = std::string(argv[2]); |
||||||
|
ok = (command == "create" || command == "verify"); |
||||||
|
} |
||||||
|
if (!ok) { |
||||||
|
fprintf(stderr, "Usage: %s <path> [create|verify] \n", argv[0]); |
||||||
|
exit(1); |
||||||
|
} |
||||||
|
if (path.back() != '/') { |
||||||
|
path += "/"; |
||||||
|
} |
||||||
|
|
||||||
|
bool sanity_ok = rocksdb::RunSanityTests(command, path); |
||||||
|
|
||||||
|
return sanity_ok ? 0 : 1; |
||||||
|
} |
||||||
Loading…
Reference in new issue