X-Git-Url: https://piware.de/gitweb/?p=learn-rust.git;a=blobdiff_plain;f=tokio-tutorial-mini-redis%2Fsrc%2Fmain.rs;h=d255ab36c270eb741ebc9f22c18d0a02aff6448a;hp=a49865584dc70a67904e2e95634c4e5cbfe01a62;hb=098305fcbcb11b65b55f5969b346ba8fc77cd13c;hpb=bd9188aabf12e3071c68d738c614da69473a64d7 diff --git a/tokio-tutorial-mini-redis/src/main.rs b/tokio-tutorial-mini-redis/src/main.rs index a498655..d255ab3 100644 --- a/tokio-tutorial-mini-redis/src/main.rs +++ b/tokio-tutorial-mini-redis/src/main.rs @@ -1,17 +1,45 @@ -use mini_redis::{client, Result}; +use std::collections::HashMap; + +use mini_redis::{Connection, Frame}; +use mini_redis::Command::{self, Get, Set}; +use tokio::net::{TcpListener, TcpStream}; #[tokio::main] -async fn main() -> Result<()> { - // Open a connection to the mini-redis address. - let mut client = client::connect("127.0.0.1:6379").await?; +async fn main() { + let listener = TcpListener::bind("127.0.0.1:6379").await.unwrap(); - // Set the key "hello" with value "world" - client.set("hello", "world".into()).await?; + loop { + // The second item contains the IP and port of the new connection + let (socket, _) = listener.accept().await.unwrap(); + tokio::spawn(async move { process(socket).await }); + } +} - // Get key "hello" - let result = client.get("hello").await?; +async fn process(socket: TcpStream) { + let mut db = HashMap::new(); + let mut connection = Connection::new(socket); - println!("got value from the server; result={:?}", result); + 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.insert(cmd.key().to_string(), cmd.value().to_vec()); + Frame::Simple("OK".to_string()) + } + Get(cmd) => { + if let Some(value) = db.get(cmd.key()) { + // `Frame::Bulk` expects data to be of type `Bytes`. This + // type will be covered later in the tutorial. For now, + // `&Vec` is converted to `Bytes` using `into()`. + Frame::Bulk(value.clone().into()) + } else { + Frame::Null + } + } + cmd => panic!("unimplemented {:?}", cmd), + }; - Ok(()) + // Write the response to the client + connection.write_frame(&response).await.unwrap(); + } }