hala-io 0.1.6

hala asynchronous io programming primitive types
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

use bytes::{BufMut, Bytes, BytesMut};

pub fn as_bytes_mut<B>(buf: &mut B) -> &mut [u8]
where
    B: BufMut,
{
    let dst = buf.chunk_mut();

    unsafe { &mut *(dst as *mut _ as *mut [u8]) }
}

pub struct ReadBuf {
    bytes: BytesMut,
}

impl ReadBuf {
    pub fn with_capacity(capacity: usize) -> Self {
        ReadBuf {
            bytes: BytesMut::with_capacity(capacity),
        }
    }

    pub fn as_mut(&mut self) -> &mut [u8] {
        let dst = self.bytes.chunk_mut();

        unsafe { &mut *(dst as *mut _ as *mut [u8]) }
    }

    pub fn filled(&mut self, advance: usize) {
        unsafe {
            self.bytes.advance_mut(advance);
        }
    }

    pub fn into_bytes_mut(mut self, advance: Option<usize>) -> BytesMut {
        if let Some(advance) = advance {
            self.filled(advance);
        }

        self.bytes
    }

    pub fn into_bytes(self, advance: Option<usize>) -> Bytes {
        self.into_bytes_mut(advance).into()
    }
}