diff --git a/test/ttl_test.cc b/test/ttl_test.cc index 6f9bb92..018258d 100644 --- a/test/ttl_test.cc +++ b/test/ttl_test.cc @@ -1,11 +1,9 @@ #include "gtest/gtest.h" - #include "leveldb/env.h" #include "leveldb/db.h" - using namespace leveldb; constexpr int value_size = 2048; @@ -20,15 +18,30 @@ Status OpenDB(std::string dbName, DB **db) { void InsertData(DB *db, uint64_t ttl/* second */) { WriteOptions writeOptions; int key_num = data_size / value_size; + srand(0); - for (int i = 0; i < 100000; i++) { - int key_ = i; + for (int i = 0; i < key_num; i++) { + int key_ = rand() % key_num+1; std::string key = std::to_string(key_); std::string value(value_size, 'a'); db->Put(writeOptions, key, value, ttl); } } +void GetData(DB *db, int size = (1 << 30)) { + ReadOptions readOptions; + int key_num = data_size / value_size; + + // 点查 + srand(0); + for (int i = 0; i < 100; i++) { + int key_ = rand() % key_num+1; + std::string key = std::to_string(key_); + std::string value; + db->Get(readOptions, key, &value); + } +} + TEST(TestTTL, ReadTTL) { DB *db; if(OpenDB("testdb", &db).ok() == false) { @@ -36,63 +49,121 @@ TEST(TestTTL, ReadTTL) { abort(); } - uint64_t ttl = 10; + uint64_t ttl = 20; InsertData(db, ttl); ReadOptions readOptions; Status status; int key_num = data_size / value_size; - srand(static_cast(time(0))); - for (int i = 0; i < 100000; i++) { - int key_ = i; + srand(0); + for (int i = 0; i < 100; i++) { + int key_ = rand() % key_num+1; std::string key = std::to_string(key_); std::string value; status = db->Get(readOptions, key, &value); ASSERT_TRUE(status.ok()); } - Env::Default()->SleepForMicroseconds((ttl+5) * 1000000); + Env::Default()->SleepForMicroseconds((ttl+1) * 1000000); - for (int i = 0; i < 100000; i++) { - int key_ = i; + for (int i = 0; i < 100; i++) { + int key_ = rand() % key_num+1; std::string key = std::to_string(key_); std::string value; status = db->Get(readOptions, key, &value); - if(status.ok()){ - std::cout<<"!!"; - } ASSERT_FALSE(status.ok()); } + + delete db; } -// TEST(TestTTL, CompactionTTL) { -// DB *db; +TEST(TestTTL, CompactionTTL) { + DB *db; + + if(OpenDB("testdb", &db).ok() == false) { + std::cerr << "open db failed" << std::endl; + abort(); + } + + uint64_t ttl = 20; + InsertData(db, ttl); -// if(OpenDB("testdb", &db).ok() == false) { -// std::cerr << "open db failed" << std::endl; -// abort(); -// } + leveldb::Range ranges[1]; + ranges[0] = leveldb::Range("-", "A"); + uint64_t sizes[1]; + db->GetApproximateSizes(ranges, 1, sizes); + ASSERT_GT(sizes[0], 0); -// uint64_t ttl = 20; -// InsertData(db, ttl); + Env::Default()->SleepForMicroseconds((ttl+1) * 1000000); -// leveldb::Range ranges[1]; -// ranges[0] = leveldb::Range("-", "A"); -// uint64_t sizes[1]; -// db->GetApproximateSizes(ranges, 1, sizes); -// ASSERT_GT(sizes[0], 0); + db->CompactRange(nullptr, nullptr); -// Env::Default()->SleepForMicroseconds(ttl * 1000000); + leveldb::Range ranges_1[1]; + ranges[0] = leveldb::Range("-", "A"); + uint64_t sizes_1[1]; + db->GetApproximateSizes(ranges_1, 1, sizes_1); + ASSERT_EQ(sizes_1[0], 0); -// db->CompactRange(nullptr, nullptr); + delete db; +} + + +TEST(TestTTL, OurTTL) { + DB *db; + WriteOptions writeOptions; + ReadOptions readOptions; + if(OpenDB("testdb_for_XOY", &db).ok() == false) { + std::cerr << "open db failed" << std::endl; + abort(); + } + for (int i = 0; i < 10000; i++) { + std::string key = std::to_string(i); + std::string value = std::to_string(i); + db->Put(writeOptions, key, value); + } + for (int i = 0; i < 10000; i++) { + std::string key = std::to_string(i); + std::string value = std::to_string(i*2); + db->Put(writeOptions, key, value, 30); + } + for (int i = 0; i < 10000; i++) { + std::string key = std::to_string(i); + std::string value = std::to_string(i*3); + db->Put(writeOptions, key, value, 15); + } + + for (int i = 0; i < 10000; i++) { + std::string key = std::to_string(i); + std::string value; + Status status = db->Get(readOptions, key, &value); + ASSERT_TRUE(status.ok()); + ASSERT_TRUE(value==std::to_string(i*3)); + } + + Env::Default()->SleepForMicroseconds((15+1) * 1000000); + + for (int i = 0; i < 10000; i++) { + std::string key = std::to_string(i); + std::string value; + Status status = db->Get(readOptions, key, &value); + ASSERT_TRUE(status.ok()); + ASSERT_TRUE(value==std::to_string(i*2)); + } + + Env::Default()->SleepForMicroseconds((15+1) * 1000000); + + for (int i = 0; i < 10000; i++) { + std::string key = std::to_string(i); + std::string value; + Status status = db->Get(readOptions, key, &value); + ASSERT_TRUE(status.ok()); + ASSERT_TRUE(value==std::to_string(i)); + } + + delete db; +} -// leveldb::Range ranges[1]; -// ranges[0] = leveldb::Range("-", "A"); -// uint64_t sizes[1]; -// db->GetApproximateSizes(ranges, 1, sizes); -// ASSERT_EQ(sizes[0], 0); -// } int main(int argc, char** argv) {