Handle EXISTS command

mleleszi · Jun 11, 2024 · 9d0c95b · 9d0c95b
1 parent 9af929e
commit 9d0c95b
Show file tree

Hide file tree

Showing 4 changed files with 21 additions and 0 deletions.
diff --git a/src/controller.cpp b/src/controller.cpp
@@ -1,5 +1,6 @@
 #include <algorithm>
 #include <iostream>
+#include <numeric>
 
 #include "controller.h"
 #include "protocol.h"
@@ -31,6 +32,8 @@ RedisType::RedisValue Controller::handleCommand(const std::vector<RedisType::Bul
         return handleSet(command);
     } else if (commandType == "GET") {
         return handleGet(command);
+    } else if (commandType == "EXISTS") {
+        return handleExists(command);
     }
 
     return RedisType::SimpleError("ERR unsupported command");
@@ -57,6 +60,9 @@ RedisType::RedisValue Controller::handleSet(const std::vector<RedisType::BulkStr
 
     return RedisType::SimpleString("OK");
 }
+
+// TODO: WORKS INCORRECTLY FOR INTEGERS
+// TODO: set and get works incorrectly when ends with number??
 RedisType::RedisValue Controller::handleGet(const std::vector<RedisType::BulkString> &command) {
     if (command.size() != 2) { return RedisType::SimpleError("ERR wrong number of arguments for 'get' command"); }
 
@@ -68,3 +74,14 @@ RedisType::RedisValue Controller::handleGet(const std::vector<RedisType::BulkStr
 
     return RedisType::BulkString();
 }
+RedisType::RedisValue Controller::handleExists(const std::vector<RedisType::BulkString> &command) {
+    if (command.size() < 2) { return RedisType::SimpleError("ERR wrong number of arguments for 'exists' command"); }
+
+    int count =
+            std::accumulate(command.begin() + 1, command.end(), 0, [this](int acc, const RedisType::BulkString &cmd) {
+                auto key = extractStringFromBytes(*cmd.data, 0, cmd.data->size());
+                return acc + (dataStore.exists(key) ? 1 : 0);
+            });
+
+    return RedisType::Integer(count);
+}
diff --git a/src/controller.h b/src/controller.h
@@ -12,6 +12,7 @@ class Controller {
     RedisType::RedisValue handlePing(const std::vector<RedisType::BulkString> &command);
     RedisType::RedisValue handleSet(const std::vector<RedisType::BulkString> &command);
     RedisType::RedisValue handleGet(const std::vector<RedisType::BulkString> &command);
+    RedisType::RedisValue handleExists(const std::vector<RedisType::BulkString> &command);
 
 
     DataStore dataStore;

diff --git a/src/datastore.h b/src/datastore.h
@@ -15,6 +15,8 @@ class DataStore {
 
     void set(const std::string &key, const std::string &val) { store[key] = val; }
 
+    bool exists(const std::string &key) { return store.contains(key); }
+
 
 private:
     std::unordered_map<std::string, std::string> store;

diff --git a/src/redis_type.h b/src/redis_type.h
@@ -24,6 +24,7 @@ namespace RedisType {
 
     struct Integer {
         std::int64_t data;
+        Integer(const int64_t &data) : data(data) {}
     };
 
     struct SimpleError {