|
|
- #include "gtest/gtest.h"
- #include "leveldb/env.h"
- #include "leveldb/db.h"
-
- using namespace leveldb;
-
- constexpr int value_size = 2048;
- constexpr int data_size = 128 << 20;
-
- Status OpenDB(std::string dbName, DB **db) {
- Options options;
- options.create_if_missing = true;
- return DB::Open(options, dbName, 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 < 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) {
- std::cerr << "open db failed" << std::endl;
- abort();
- }
-
- uint64_t ttl = 20;
-
- InsertData(db, ttl);
-
- ReadOptions readOptions;
- Status status;
- 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;
- status = db->Get(readOptions, key, &value);
- ASSERT_TRUE(status.ok());
- }
-
- Env::Default()->SleepForMicroseconds(ttl * 1000000);
-
- 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_FALSE(status.ok());
- }
-
- delete 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);
-
- leveldb::Range ranges[1];
- ranges[0] = leveldb::Range("-", "A");
- uint64_t sizes[1];
- db->GetApproximateSizes(ranges, 1, sizes);
- ASSERT_GT(sizes[0], 0);
- Env::Default()->SleepForMicroseconds(ttl * 1000000);
-
- db->CompactRange(nullptr, nullptr);
- leveldb::Range ranges1[1];
- ranges1[0] = leveldb::Range("-", "A");
- uint64_t sizes1[1];
- db->GetApproximateSizes(ranges1, 1, sizes1);
- ASSERT_EQ(sizes1[0], 0);
-
- delete db;
- }
-
- // Time-consuming when kNumLevels is 7, so run this test when kNumLevels is set to 3,
- // the codes on branch light_ver is a version with kNumLevels set to 3.
- // In this case level 2 is the last level with files.
- TEST(TestTTL, LastLevelCompaction) {
- DB *db;
-
- if(OpenDB("testdb", &db).ok() == false) {
- std::cerr << "open db failed" << std::endl;
- abort();
- }
-
- uint64_t ttl = 20;
- InsertData(db, ttl);
-
- leveldb::Range ranges[1];
- ranges[0] = leveldb::Range("-", "A");
- uint64_t sizes[1];
- db->GetApproximateSizes(ranges, 1, sizes);
- ASSERT_GT(sizes[0], 0);
-
- std::string last_level_file_num;
- std::string last_level = "2";
- db->GetProperty("leveldb.num-files-at-level" + last_level, &last_level_file_num);
- std::cout << "File nums in last level: " << last_level_file_num << std::endl;
- ASSERT_GT(std::atoi(last_level_file_num.c_str()), 0);
-
- Env::Default()->SleepForMicroseconds(ttl * 1000000);
-
- db->CompactRange(nullptr, nullptr);
- leveldb::Range ranges1[1];
- ranges1[0] = leveldb::Range("-", "A");
- uint64_t sizes1[1];
- db->GetApproximateSizes(ranges1, 1, sizes1);
- ASSERT_EQ(sizes1[0], 0);
-
- 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();
- }
|