voidio 0.1.14

VOID I/O - High-performance Cross-platform I/O for Rust.
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
50
51
52
53
54
55

use std::net::SocketAddr;

use super::connection::QuicConnection;

pub struct QuicStream<'a> {
    pub(crate) id: u32,
    pub(crate) src: &'a QuicConnection,
    pub(crate) ondata_handler: Option<Box<dyn FnMut(&[u8]) + Send>>,
    pub(crate) onclose_handler: Option<Box<dyn FnMut() + Send>>,
}

impl<'a> QuicStream<'a> {
    pub fn new(id: u32, src: &'a QuicConnection) -> Self {
        Self {
            id,
            src,
            ondata_handler: None,
            onclose_handler: None,
        }
    }

    pub fn id(&self) -> u32 {
        self.id
    }

    pub fn src_connection(&self) -> &QuicConnection {
        self.src
    }

    pub fn write(&self, data: &[u8]) -> Result<(), String> {
        Err("QuicStream::write is not implemented yet".to_string())
    }

    pub fn on_data<F>(&mut self, h: F) where F: FnMut(&[u8]) + Send + 'static {
        self.ondata_handler = Some(Box::new(h));
    }

    pub fn read(&self, buf: &mut [u8]) -> Result<usize, String> {
        Err("QuicStream::read is not implemented yet".to_string())
    }

    pub fn on_close<F>(&mut self, h: F) where F: FnMut() + Send + 'static {
        self.onclose_handler = Some(Box::new(h));
    }

    pub fn src(&self) -> SocketAddr {
        self.src.address
    }
}

impl<'a> std::fmt::Display for QuicStream<'a> {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        write!(f, "QuicStream(id: {}, src: {})", self.id, self.src.address)
    }
}