環境

• Time 2022-01-13
• Rust 1.57.0
• Tokio 1.15.0

概念

參考：https://tokio.rs/tokio/tutorial/spawning

示例

main.rs

use mini_redis::Command::{Get, Set};
use mini_redis::{Command, Connection, Frame};
use std::collections::HashMap;
use tokio::net::{TcpListener, TcpStream};

#[tokio::main]
async fn main() -> mini_redis::Result<()> {
    let listener = TcpListener::bind("127.0.0.1:6379").await?;

    loop {
        let (socket, address) = listener.accept().await?;
        println!("客戶端: {}", address);
        // 提交任務
        tokio::spawn(process(socket)).await??;
    }
}

async fn process(socket: TcpStream) -> mini_redis::Result<()> {
    let mut database = HashMap::new();
    // Connection 是 mini redis 定義的內容
    let mut client = Connection::new(socket);

    // 迴圈接收資料
    while let Some(frame) = client.read_frame().await? {
        let response = match Command::from_frame(frame)? {
            Set(cmd) => {
                database.insert(cmd.key().to_string(), cmd.value().to_vec());
                Frame::Simple("OK".to_string())
            }
            Get(cmd) => {
                if let Some(value) = database.get(cmd.key()) {
                    // 使用 into 轉成 Bytes 型別
                    Frame::Bulk(value.clone().into())
                } else {
                    Frame::Null
                }
            }
            cmd => panic!("unimplemented {:?}", cmd),
        };

        client.write_frame(&response).await.unwrap();
    }

    Ok(())
}
 

總結

可以設定客戶端傳送的值，並且通過命令獲取到。不過資料是客戶端隔離的，每個客戶端自己只能看到自己的資料。

附錄