X-Git-Url: https://piware.de/gitweb/?p=learn-rust.git;a=blobdiff_plain;f=tokio-tutorial-mini-redis%2Fsrc%2Fmain.rs;fp=tokio-tutorial-mini-redis%2Fsrc%2Fmain.rs;h=0000000000000000000000000000000000000000;hp=2488f033ead6e2808c08fe6f4499e3b0818f9d15;hb=8130bd0b1f91135585cf0507e1b5b651cf05400a;hpb=c5233c520ab09608cabb8492b5da2292589f23b8 diff --git a/tokio-tutorial-mini-redis/src/main.rs b/tokio-tutorial-mini-redis/src/main.rs deleted file mode 100644 index 2488f03..0000000 --- a/tokio-tutorial-mini-redis/src/main.rs +++ /dev/null @@ -1,47 +0,0 @@ -use std::collections::HashMap; -use std::sync::{Arc, Mutex}; - -use bytes::Bytes; -use mini_redis::{Connection, Frame}; -use mini_redis::Command::{self, Get, Set}; -use tokio::net::{TcpListener, TcpStream}; - -type Db = Arc>>; - -#[tokio::main] -async fn main() { - let listener = TcpListener::bind("127.0.0.1:6379").await.unwrap(); - let db: Db = Arc::new(Mutex::new(HashMap::new())); - - loop { - // The second item contains the IP and port of the new connection - let (socket, _) = listener.accept().await.unwrap(); - let db_i = db.clone(); - tokio::spawn(async move { process(socket, db_i).await }); - } -} - -async fn process(socket: TcpStream, db: Db) { - let mut connection = Connection::new(socket); - - while let Some(frame) = connection.read_frame().await.unwrap() { - let response = match Command::from_frame(frame).unwrap() { - Set(cmd) => { - // The value is stored as `Vec` - db.lock().unwrap().insert(cmd.key().to_string(), cmd.value().clone()); - Frame::Simple("OK".to_string()) - } - Get(cmd) => { - if let Some(value) = db.lock().unwrap().get(cmd.key()) { - Frame::Bulk(value.clone()) - } else { - Frame::Null - } - } - cmd => panic!("unimplemented {:?}", cmd), - }; - - // Write the response to the client - connection.write_frame(&response).await.unwrap(); - } -}