You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
224 lines
6.4 KiB
224 lines
6.4 KiB
#include "gtest/gtest.h"
|
|
#include "leveldb/env.h"
|
|
#include "leveldb/db.h"
|
|
#include "util/coding.h"
|
|
#include <iostream>
|
|
using namespace leveldb;
|
|
|
|
using Field=std::pair<Slice,Slice>;
|
|
using FieldArray=std::vector<std::pair<Slice, Slice>>;
|
|
|
|
Status OpenDB(std::string dbName, DB **db) {
|
|
Options options;
|
|
options.max_file_size=16*1024;
|
|
options.write_buffer_size=32*1024;
|
|
options.create_if_missing = true;
|
|
return DB::Open(options, dbName, db);
|
|
}
|
|
|
|
TEST(Test, checkIterator) {
|
|
DB *db;
|
|
WriteOptions writeOptions;
|
|
ReadOptions readOptions;
|
|
if(OpenDB("testdb_for_XOY_search", &db).ok() == false) {
|
|
std::cerr << "open db failed" << std::endl;
|
|
abort();
|
|
}
|
|
std::vector<std::string> values;
|
|
for(int i=0;i<5000;i++){
|
|
std::string key=std::to_string(i);
|
|
while(key.size()<4){
|
|
key='0'+key;
|
|
}
|
|
std::string value;
|
|
for(int j=0;j<5000;j++){
|
|
value+=std::to_string(i);
|
|
}
|
|
values.push_back(value);
|
|
Status s=db->Put(writeOptions,key,value);
|
|
assert(s.ok());
|
|
}
|
|
auto iter=db->NewIterator(readOptions);
|
|
iter->SeekToFirst();
|
|
for(int i=0;i<5000;i++){
|
|
assert(iter->Valid());
|
|
auto value=iter->value();
|
|
if(value!=values[i]){
|
|
std::cout<<std::string(value.data(),value.size())<<std::endl;
|
|
}
|
|
assert(values[i]==value);
|
|
iter->Next();
|
|
}
|
|
assert(!iter->Valid());
|
|
iter->SeekToLast();
|
|
for(int i=4999;i>=0;i--){
|
|
assert(iter->Valid());
|
|
auto value=iter->value();
|
|
if(value!=values[i]){
|
|
std::cout<<std::string(value.data(),value.size())<<std::endl;
|
|
}
|
|
assert(values[i]==value);
|
|
iter->Prev();
|
|
}
|
|
assert(!iter->Valid());
|
|
iter->Seek("4990");
|
|
for(int i=4990;i<5000;i++){
|
|
assert(iter->Valid());
|
|
auto value=iter->value();
|
|
if(value!=values[i]){
|
|
std::cout<<std::string(value.data(),value.size())<<std::endl;
|
|
}
|
|
assert(values[i]==value);
|
|
iter->Next();
|
|
}
|
|
assert(!iter->Valid());
|
|
delete iter;
|
|
delete db;
|
|
}
|
|
|
|
// TEST(Test, CheckGetFields) {
|
|
// DB *db;
|
|
// WriteOptions writeOptions;
|
|
// ReadOptions readOptions;
|
|
// if(OpenDB("testdb_for_XOY", &db).ok() == false) {
|
|
// std::cerr << "open db failed" << std::endl;
|
|
// abort();
|
|
// }
|
|
// std::string key1 = "k_1";
|
|
|
|
// FieldArray fields1 = {
|
|
// {"name", "Customer#000000001"},
|
|
// {"address", "IVhzIApeRb"},
|
|
// {"phone", "25-989-741-2988"}
|
|
// };
|
|
|
|
// auto value1=SerializeValue(fields1);
|
|
|
|
// db->Put(WriteOptions(), key1, value1);
|
|
|
|
// std::string value_ret;
|
|
// FieldArray res1;
|
|
|
|
// db->Get(ReadOptions(), key1, &value_ret);
|
|
// DeserializeValue(value_ret, &res1);
|
|
// for(auto pr:res1){
|
|
// std::cout<<std::string(pr.first.data(),pr.first.size())<<" "<<std::string(pr.second.data(),pr.second.size())<<"\n";
|
|
// }
|
|
// ASSERT_TRUE(CompareFieldArray(fields1, res1));
|
|
|
|
// db->Delete(WriteOptions(),key1);
|
|
|
|
// std::cout<<"get serialized value done"<<std::endl;
|
|
// delete db;
|
|
|
|
// }
|
|
|
|
// TEST(Test, CheckSearchKey) {
|
|
// DB *db;
|
|
// ReadOptions readOptions;
|
|
// if(OpenDB("testdb_for_XOY", &db).ok() == false) {
|
|
// std::cerr << "open db failed" << std::endl;
|
|
// abort();
|
|
// }
|
|
// std::vector<std::string> keys;
|
|
// std::vector<std::string> target_keys;
|
|
// for(int i=0;i<10000;i++){
|
|
// std::string key=std::to_string(rand()%10000)+"_"+std::to_string(i);//random for generate nonincreasing keys
|
|
// FieldArray fields={
|
|
// {"name", key},
|
|
// {"address", std::to_string(rand()%7)},
|
|
// {"phone", std::to_string(rand()%114514)}
|
|
// };
|
|
// if(rand()%5==0){
|
|
// fields[0].second="special_key";
|
|
// target_keys.push_back(key);
|
|
// }
|
|
// keys.push_back(key);
|
|
// db->Put(WriteOptions(),key,SerializeValue(fields));
|
|
// }
|
|
// std::sort(target_keys.begin(),target_keys.end());
|
|
// std::vector<std::string> key_res;
|
|
// Get_keys_by_field(db,ReadOptions(),{"name", "special_key"},&key_res);
|
|
// ASSERT_TRUE(CompareKey(key_res, target_keys));
|
|
// std::cout<<"get key by field done"<<std::endl;
|
|
// for(auto s:keys){
|
|
// db->Delete(WriteOptions(),s);
|
|
// }
|
|
// delete db;
|
|
// }
|
|
|
|
TEST(Test, LARGE_DATA_COMPACT_TEST) {
|
|
DB *db;
|
|
WriteOptions writeOptions;
|
|
ReadOptions readOptions;
|
|
if(OpenDB("testdb_for_XOY_large", &db).ok() == false) {
|
|
std::cerr << "open db failed" << std::endl;
|
|
abort();
|
|
}
|
|
std::vector<std::string> values;
|
|
for(int i=0;i<5000;i++){
|
|
std::string key=std::to_string(i);
|
|
std::string value;
|
|
for(int j=0;j<5000;j++){
|
|
value+=std::to_string(i);
|
|
}
|
|
values.push_back(value);
|
|
db->Put(writeOptions,key,value);
|
|
}
|
|
for(int i=0;i<5000;i++){
|
|
std::string key=std::to_string(i);
|
|
std::string value;
|
|
Status s=db->Get(readOptions,key,&value);
|
|
assert(s.ok());
|
|
if(values[i]!=value){
|
|
std::cout<<value.size()<<std::endl;
|
|
assert(0);
|
|
}
|
|
ASSERT_TRUE(values[i]==value);
|
|
}
|
|
delete db;
|
|
}
|
|
|
|
// TEST(Test, Garbage_Collect_TEST) {
|
|
// DB *db;
|
|
// WriteOptions writeOptions;
|
|
// ReadOptions readOptions;
|
|
// if(OpenDB("testdb_for_XOY_large", &db).ok() == false) {
|
|
// std::cerr << "open db failed" << std::endl;
|
|
// abort();
|
|
// }
|
|
// std::vector<std::string> values;
|
|
// for(int i=0;i<5000;i++){
|
|
// std::string key=std::to_string(i);
|
|
// std::string value;
|
|
// for(int j=0;j<1000;j++){
|
|
// value+=std::to_string(i);
|
|
// }
|
|
// values.push_back(value);
|
|
// db->Put(writeOptions,key,value);
|
|
// }
|
|
// std::cout<<"start gc"<<std::endl;
|
|
// db->TEST_GarbageCollect();
|
|
// std::cout<<"finish gc"<<std::endl;
|
|
|
|
// for(int i=0;i<5000;i++){
|
|
// // std::cout<<i<<std::endl;
|
|
// std::string key=std::to_string(i);
|
|
// std::string value;
|
|
// Status s=db->Get(readOptions,key,&value);
|
|
// assert(s.ok());
|
|
// if(values[i]!=value){
|
|
// std::cout<<value.size()<<std::endl;
|
|
// assert(0);
|
|
// }
|
|
// ASSERT_TRUE(values[i]==value);
|
|
// }
|
|
// delete db;
|
|
// }
|
|
|
|
|
|
int main(int argc, char** argv) {
|
|
// All tests currently run with the same read-only file limits.
|
|
testing::InitGoogleTest(&argc, argv);
|
|
return RUN_ALL_TESTS();
|
|
}
|