zyst 1.0.3

Redis-compatible server
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49

use crate::errors::format_redis_error;
use crate::process::process_command;
use crate::resp::parse_resp_command;
use crate::types::Db;
use std::error::Error;
use tokio::io::{AsyncReadExt, AsyncWriteExt};
use tokio::net::TcpStream;

pub async fn handle_client(mut socket: TcpStream, db: Db) -> Result<(), Box<dyn Error>> {
    let mut buffer = [0; 1024];

    loop {
        let bytes_read = socket.read(&mut buffer).await?;

        if bytes_read == 0 {
            // Client disconnected
            return Ok(());
        }

        let raw_command = String::from_utf8_lossy(&buffer[..bytes_read])
            .trim()
            .to_string();

        let parsed_commands = match parse_resp_command(&raw_command) {
            Ok(parsed) => parsed,
            Err(e) => {
                let error_response = format_redis_error(e);
                socket.write_all(error_response.as_bytes()).await?;
                socket.flush().await?;
                continue;
            }
        };

        for parsed in parsed_commands {
            match process_command(parsed, &db, false).await {
                Ok(resp) => {
                    let response = resp.to_string();
                    socket.write_all(response.as_bytes()).await?;
                    socket.flush().await?;
                }
                Err(e) => {
                    let error_response = format_redis_error(e);
                    socket.write_all(error_response.as_bytes()).await?;
                    socket.flush().await?;
                }
            }
        }
    }
}