buffet 0.3.3

Thread-local buffer pool for the `loona` crate.
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

use crate::{BufResult, IoBufMut, Piece, ReadOwned, WriteOwned};

use tokio::io::{AsyncRead, AsyncWrite, AsyncWriteExt};

impl<T> ReadOwned for T
where
    T: AsyncRead + Unpin,
{
    async fn read_owned<B: IoBufMut>(&mut self, mut buf: B) -> BufResult<usize, B> {
        let buf_slice = unsafe { buf.slice_mut() };
        let res = tokio::io::AsyncReadExt::read(self, buf_slice).await;
        (res, buf)
    }
}

impl<T> WriteOwned for T
where
    T: AsyncWrite + Unpin,
{
    async fn write_owned(&mut self, buf: impl Into<Piece>) -> BufResult<usize, Piece> {
        let buf = buf.into();
        let res = tokio::io::AsyncWriteExt::write(self, &buf[..]).await;
        (res, buf)
    }

    // TODO: implement writev, for performance. this involves wrapping
    // everything in `IoSlice`, advancing correctly, etc. It's not fun, but it
    // should yield a boost for non-uring codepaths.

    async fn shutdown(&mut self) -> std::io::Result<()> {
        AsyncWriteExt::shutdown(self).await
    }
}