rustdis 0.1.1

A partial Redis server implementation intended purely for educational purposes
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

use futures::stream::StreamExt; // Use the correct StreamExt trait
use tokio::net::tcp::{OwnedReadHalf, OwnedWriteHalf};
use tokio::net::TcpStream;
use tokio_util::codec::FramedRead;
use uuid::Uuid;

use std::net::SocketAddr;

use crate::codec::FrameCodec;
use crate::frame::Frame;
use crate::Result;

pub struct Connection {
    pub id: Uuid,
    pub client_address: SocketAddr,
    pub writer: OwnedWriteHalf,
    reader: FramedRead<OwnedReadHalf, FrameCodec>,
}

impl Connection {
    pub fn new(stream: TcpStream, client_address: SocketAddr) -> Connection {
        let (reader, writer) = stream.into_split();
        let reader = FramedRead::new(reader, FrameCodec);
        let id = Uuid::new_v4();

        Connection {
            id,
            writer,
            reader,
            client_address,
        }
    }

    pub async fn read_frame(&mut self) -> Result<Option<Frame>> {
        match self.reader.next().await {
            Some(Ok(frame)) => Ok(Some(frame)),
            Some(Err(e)) => Err(e),
            None => Ok(None),
        }
    }
}