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<unsigned int>(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) {