✨ feat: add command keys

camargo2019 · Sep 9, 2024 · 41607c1 · 41607c1
1 parent bbe4968
commit 41607c1
Show file tree

Hide file tree

Showing 5 changed files with 34 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -85,14 +85,15 @@ with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as sock:
 
 ## 📚 Comandos Suportados
 
-| Comando                | Descrição                                                |
-|------------------------|----------------------------------------------------------|
+| Comando                | Descrição                                                 |
+|------------------------|-----------------------------------------------------------|
 | `SET key value`         | Define um valor para a chave                             |
 | `SET key value expire`  | Define um valor com tempo de expiração                   |
 | `GET key`               | Retorna o valor associado à chave                        |
 | `DEL key`               | Remove uma chave                                         |
 | `AUTH user password`    | Autentica o usuário com a senha                          |
 | `USE database`          | Especifica qual banco de dados será utilizado            |
+| `KEYS`                  | Lista todas as chaves registradas                        |
 
 ## 🌟 Contribuindo
 

diff --git a/core/cache/cache.cpp b/core/cache/cache.cpp
@@ -90,6 +90,16 @@ bool Cache::del(std::string db, std::string key){
     return false;
 }
 
+std::vector<std::string> Cache::keys(std::string db){
+    std::vector<std::string> keys;
+
+    for (const auto& row: cache_[db]){
+        keys.push_back(row.first);
+    }
+
+    return keys;
+}
+
 void Cache::save(){
     std::ofstream file("data/databases.dat");
 

diff --git a/core/cache/cache.h b/core/cache/cache.h
@@ -52,6 +52,7 @@
             std::string get(std::string db, std::string key);
             bool set(std::string db, std::string key, std::string value, int expire = -1);
             bool del(std::string db, std::string key);
+            std::vector<std::string> keys (std::string db);
             void save();
 
         private:

diff --git a/core/manager/manager.cpp b/core/manager/manager.cpp
@@ -92,6 +92,10 @@ void Manager::invokeAction(){
         invokeDel(args);
     }
 
+    if (std::find(commands.keys.begin(), commands.keys.end(), command) != commands.keys.end()) {
+        invokeKeys(args);
+    }
+
     data_.clear();
 }
 
@@ -161,7 +165,7 @@ void Manager::invokeUse(std::vector<std::string> args){
 }
 
 void Manager::invokeAuth(std::vector<std::string> args){
-    if (!user.user.empty() && !user.db.empty()) return result(std::string("ERROR: you are not authenticated"));
+    if (!user.user.empty() && !user.db.empty()) return result(std::string("ERROR: you are authenticated"));
     if (args[0].empty() || args[1].empty()) return result(std::string("ERROR: use AUTH USER PASS"));
 
     for (const auto& auth : ConfigConn.auth.basic) {
@@ -176,6 +180,19 @@ void Manager::invokeAuth(std::vector<std::string> args){
     result(std::string("ERROR: failed to authenticate"));
 }
 
+void Manager::invokeKeys(std::vector<std::string> args){
+    if (user.user.empty() && user.db.empty()) return result(std::string("ERROR: you need to authenticate"));
+    if (user.db == "*") return result(std::string("ERROR: use the `USE DB` command to choose the name of the database instance"));
+
+    std::string messageKeys;
+    std::vector<std::string> keys = cache_.keys(user.db);
+    for (const std::string& key : keys) {
+        messageKeys += key + "\n";
+    }
+
+    return result(std::string("SUCCESS: \r\n" + messageKeys));
+}
+
 void Manager::scheduleSave(){
     if (!saveRunning_){
         saveRunning_ = true;

diff --git a/core/manager/manager.h b/core/manager/manager.h
@@ -40,6 +40,7 @@
         std::vector<std::string> get = {"GET", "get"};
         std::vector<std::string> auth = {"AUTH", "auth"};
         std::vector<std::string> use = {"USE", "use"};
+        std::vector<std::string> keys = {"KEYS", "keys"};
     };
 
 
@@ -59,6 +60,7 @@
             void invokeGet(std::vector<std::string> args);
             void invokeAuth(std::vector<std::string> args);
             void invokeUse(std::vector<std::string> args);
+            void invokeKeys(std::vector<std::string> args);
             void scheduleSave();
 
             boost::asio::ip::tcp::socket socket_;