feat: expire time for key

Author: ZYunfeii

client/client_main.cc

@@ -1,49 +1,72 @@
 #include <iostream>
 #include "kvclient.h"
 
-int main(int argc, char** argv) {
-    KVClient* kvclient = new KVClient("127.0.0.1", 6789);
-    int res = kvclient->setKV("yunfei", "23", MiniKV_STRING);
+void testExpire(KVClient* c) {
+    int res = c->setKV("yunfei", "23", MiniKV_STRING);
     if (res == MiniKV_SET_SUCCESS) {
         fprintf(stdout, "set key success!\n");
     } else {
         fprintf(stderr, "set key failed!\n");
         exit(1);
     }
-    GetRes ans = kvclient->getK("yunfei");
-    if (ans.encoding == MiniKV_GET_SUCCESS) {
-        fprintf(stdout, "get key success!\n");
+    GetRes ans = c->getK("yunfei");
+    std::cout << ans.data[0] << std::endl;
+    res = c->setExpires("yunfei", 1000);
+    if (res == MiniKV_SET_EXPIRE_SUCCESS) {
+        fprintf(stdout, "set expire success!\n");
     } else {
-        fprintf(stderr, "get key failed!\n");
+        fprintf(stderr, "set expire failed!\n");
         exit(1);
     }
-    for (int i = 0; i < ans.data.size(); ++i) {
-        std::cout << ans.data[i] << " ";
-    }std::cout << std::endl;
+    sleep(3);
+    ans = c->getK("yunfei");
+    std::cout << (ans.data.empty() ? "expired" : "expire doesn't work") << std::endl;
+}
 
-    res = kvclient->setKV("qjx", "23", MiniKV_LIST);
-    if (res == MiniKV_SET_SUCCESS) {
-        fprintf(stdout, "set key success!\n");
-    } else {
-        fprintf(stderr, "set key failed!\n");
-        exit(1);
-    }
-    res = kvclient->setKV("qjx", "24", MiniKV_LIST);
-    if (res == MiniKV_SET_SUCCESS) {
-        fprintf(stdout, "set key success!\n");
-    } else {
-        fprintf(stderr, "set key failed!\n");
-        exit(1);
-    }
-    ans = kvclient->getK("qjx");
-    if (ans.encoding == MiniKV_GET_SUCCESS) {
-        fprintf(stdout, "get key success!\n");
-    } else {
-        fprintf(stderr, "get key failed!\n");
-        exit(1);
-    }
-    for (int i = 0; i < ans.data.size(); ++i) {
-        std::cout << ans.data[i] << " ";
-    } std::cout << std::endl;
+int main(int argc, char** argv) {
+    KVClient* kvclient = new KVClient("127.0.0.1", 6789);
+    testExpire(kvclient);
+    // int res = kvclient->setKV("yunfei", "23", MiniKV_STRING);
+    // if (res == MiniKV_SET_SUCCESS) {
+    //     fprintf(stdout, "set key success!\n");
+    // } else {
+    //     fprintf(stderr, "set key failed!\n");
+    //     exit(1);
+    // }
+    // GetRes ans = kvclient->getK("yunfei");
+    // if (ans.encoding == MiniKV_GET_SUCCESS) {
+    //     fprintf(stdout, "get key success!\n");
+    // } else {
+    //     fprintf(stderr, "get key failed!\n");
+    //     exit(1);
+    // }
+    // for (int i = 0; i < ans.data.size(); ++i) {
+    //     std::cout << ans.data[i] << " ";
+    // }std::cout << std::endl;
+
+    // res = kvclient->setKV("qjx", "23", MiniKV_LIST);
+    // if (res == MiniKV_SET_SUCCESS) {
+    //     fprintf(stdout, "set key success!\n");
+    // } else {
+    //     fprintf(stderr, "set key failed!\n");
+    //     exit(1);
+    // }
+    // res = kvclient->setKV("qjx", "24", MiniKV_LIST);
+    // if (res == MiniKV_SET_SUCCESS) {
+    //     fprintf(stdout, "set key success!\n");
+    // } else {
+    //     fprintf(stderr, "set key failed!\n");
+    //     exit(1);
+    // }
+    // ans = kvclient->getK("qjx");
+    // if (ans.encoding == MiniKV_GET_SUCCESS) {
+    //     fprintf(stdout, "get key success!\n");
+    // } else {
+    //     fprintf(stderr, "get key failed!\n");
+    //     exit(1);
+    // }
+    // for (int i = 0; i < ans.data.size(); ++i) {
+    //     std::cout << ans.data[i] << " ";
+    // } std::cout << std::endl;
     return 0;
 }

client/kvclient.cc

@@ -55,4 +55,21 @@ GetRes KVClient::getK(std::string key) {
     }
     ans.data = data;
     return ans;
+}
+
+int KVClient::setExpires(std::string key, uint64_t millisecond) {
+    auto now = std::chrono::system_clock::now(); 
+    auto timestamp = std::chrono::duration_cast<std::chrono::milliseconds>(now.time_since_epoch()).count();
+    uint64_t expires = timestamp + millisecond;
+    kv::ReqExpire req;
+    kv::SetExpireResponse res;
+    req.set_key(key);
+    req.set_expires(expires);
+    grpc::ClientContext context;
+    grpc::Status status = stub_->SetExpire(&context, req, &res);
+    if (!status.ok() || res.flag() == false) {
+        // TODO: log
+        return MiniKV_SET_EXPIRE_FAIL;
+    }
+    return MiniKV_SET_EXPIRE_SUCCESS;
 }

client/kvclient.h

@@ -17,6 +17,7 @@ class KVClient {
 
     int setKV(std::string key, std::string val, uint32_t encoding);
     int delK(std::string key);
+    int setExpires(std::string key, uint64_t millisecond);
     GetRes getK(std::string key);
 private:
     std::unique_ptr<kv::KVServer::Stub> stub_;