Stage 6 done

Author: gloomweaver

src/command/command.rs

@@ -3,6 +3,8 @@ use super::super::resp::RespValue;
 pub enum Command {
     Ping(String),
     Echo(String),
+    Get(String),
+    Set(String, String),
 }
 
 impl Command {
@@ -37,6 +39,35 @@ impl Command {
 
                             Some(Command::Echo(arg))
                         }
+                        "get" => {
+                            let key = iter
+                                .next()
+                                .map(|arg| {
+                                    String::from_utf8_lossy(arg.as_bytes().clone().as_ref())
+                                        .to_string()
+                                })
+                                .expect("WHAT");
+
+                            Some(Command::Get(key))
+                        }
+                        "set" => {
+                            let key = iter
+                                .next()
+                                .map(|arg| {
+                                    String::from_utf8_lossy(arg.as_bytes().clone().as_ref())
+                                        .to_string()
+                                })
+                                .expect("WHAT");
+                            let value = iter
+                                .next()
+                                .map(|arg| {
+                                    String::from_utf8_lossy(arg.as_bytes().clone().as_ref())
+                                        .to_string()
+                                })
+                                .expect("WHAT");
+
+                            Some(Command::Set(key, value))
+                        }
                         _ => None,
                     }
                 } else {

src/main.rs

@@ -1,11 +1,16 @@
+use std::collections::HashMap;
+use std::sync::Arc;
+
 use anyhow::Result;
 use bytes::{BufMut, BytesMut};
 use redis_starter_rust::{command::Command, resp};
 use tokio::io::{AsyncReadExt, AsyncWriteExt};
 use tokio::net::{TcpListener, TcpStream};
+use tokio::sync::Mutex;
 
 #[tokio::main]
 async fn main() -> Result<()> {
+    let cache = Arc::new(Mutex::new(HashMap::new()));
     println!("Logs from your program will appear here!");
 
     let listener = TcpListener::bind("127.0.0.1:6379").await?;
@@ -15,9 +20,10 @@ async fn main() -> Result<()> {
         match incoming {
             Ok((stream, _)) => {
                 println!("new connection");
+                let memory = cache.clone();
 
                 tokio::spawn(async move {
-                    handle_connection(stream).await.unwrap();
+                    handle_connection(stream, memory).await.unwrap();
                 });
             }
             Err(e) => {
@@ -27,7 +33,10 @@ async fn main() -> Result<()> {
     }
 }
 
-async fn handle_connection(mut stream: TcpStream) -> Result<()> {
+async fn handle_connection(
+    mut stream: TcpStream,
+    memory: Arc<Mutex<HashMap<String, String>>>,
+) -> Result<()> {
     let mut buf = BytesMut::with_capacity(512);
     loop {
         let bytes_read = stream.read_buf(&mut buf).await?;
@@ -54,6 +63,22 @@ async fn handle_connection(mut stream: TcpStream) -> Result<()> {
 
                         stream.write_all(&out).await?;
                     }
+                    Command::Get(key) => {
+                        let mut out = BytesMut::with_capacity(512);
+                        let mem = memory.lock().await;
+                        let val = mem.get(&key).unwrap();
+
+                        out.put_slice(b"+");
+                        out.put_slice(&val.as_bytes());
+                        out.put_slice(b"\r\n");
+
+                        stream.write_all(&out).await?;
+                    }
+                    Command::Set(key, value) => {
+                        let mut mem = memory.lock().await;
+                        mem.insert(key, value);
+                        stream.write_all(b"+OK\r\n").await?;
+                    }
                 }
             } else {
                 stream.write_all(b"-ERR unknown command\r\n").await?;