...

src/cmd.rs

@@ -33,7 +33,10 @@ pub enum Cmd {
     Ttl(String),
     Exists(String),
     Quit,
-    Unknow,
+    Client(Vec<String>),
+    ClientSetName(String),
+    ClientGetName,
+    Unknow(String),
 }
 
 impl Cmd {
@@ -274,7 +277,30 @@ impl Cmd {
                             }
                             Cmd::Quit
                         }
-                        _ => Cmd::Unknow,
+                        "client" => {
+                            if cmd.len() > 1 {
+                                match cmd[1].to_lowercase().as_str() {
+                                    "setname" => {
+                                        if cmd.len() == 3 {
+                                            Cmd::ClientSetName(cmd[2].clone())
+                                        } else {
+                                            return Err(DBError("wrong number of arguments for 'client setname' command".to_string()));
+                                        }
+                                    }
+                                    "getname" => {
+                                        if cmd.len() == 2 {
+                                            Cmd::ClientGetName
+                                        } else {
+                                            return Err(DBError("wrong number of arguments for 'client getname' command".to_string()));
+                                        }
+                                    }
+                                    _ => Cmd::Client(cmd[1..].to_vec()),
+                                }
+                            } else {
+                                Cmd::Client(vec![])
+                            }
+                        }
+                        _ => Cmd::Unknow(cmd[0].clone()),
                     },
                     protocol.0,
                 ))
@@ -288,7 +314,7 @@ impl Cmd {
 
     pub async fn run(
         &self,
-        server: &Server,
+        server: &mut Server,
         protocol: Protocol,
         queued_cmd: &mut Option<Vec<(Cmd, Protocol)>>,
     ) -> Result<Protocol, DBError> {
@@ -347,14 +373,20 @@ impl Cmd {
             Cmd::Ttl(key) => ttl_cmd(server, key).await,
             Cmd::Exists(key) => exists_cmd(server, key).await,
             Cmd::Quit => Ok(Protocol::SimpleString("OK".to_string())),
-            Cmd::Unknow => Ok(Protocol::err("unknown cmd")),
+            Cmd::Client(_) => Ok(Protocol::SimpleString("OK".to_string())),
+            Cmd::ClientSetName(name) => client_setname_cmd(server, name).await,
+            Cmd::ClientGetName => client_getname_cmd(server).await,
+            Cmd::Unknow(s) => {
+                println!("\x1b[31;1munknown command: {}\x1b[0m", s);
+                Ok(Protocol::err(&format!("ERR unknown command '{}'", s)))
+            }
         }
     }
 }
 
 async fn exec_cmd(
     queued_cmd: &mut Option<Vec<(Cmd, Protocol)>>,
-    server: &Server,
+    server: &mut Server,
 ) -> Result<Protocol, DBError> {
     if queued_cmd.is_some() {
         let mut vec = Vec::new();
@@ -593,3 +625,15 @@ async fn exists_cmd(server: &Server, key: &str) -> Result<Protocol, DBError> {
         Err(e) => Ok(Protocol::err(&e.0)),
     }
 }
+
+async fn client_setname_cmd(server: &mut Server, name: &str) -> Result<Protocol, DBError> {
+    server.client_name = Some(name.to_string());
+    Ok(Protocol::SimpleString("OK".to_string()))
+}
+
+async fn client_getname_cmd(server: &Server) -> Result<Protocol, DBError> {
+    match &server.client_name {
+        Some(name) => Ok(Protocol::BulkString(name.clone())),
+        None => Ok(Protocol::Null),
+    }
+}