use std::pin::Pin;
use std::task::Context;
use std::task::Poll;
use futures_io::AsyncRead;
use crate::AsyncInput;
#[must_use]
#[repr(transparent)]
pub struct FuturesAsyncRead<I> {
inner: I,
}
impl<I> FuturesAsyncRead<I> {
#[inline(always)]
pub const fn new(inner: I) -> Self {
Self { inner }
}
#[inline(always)]
#[must_use]
pub const fn get_ref(&self) -> &I {
&self.inner
}
#[inline(always)]
#[must_use]
pub const fn get_mut(&mut self) -> &mut I {
&mut self.inner
}
#[inline(always)]
#[must_use]
pub fn get_pin_mut(self: Pin<&mut Self>) -> Pin<&mut I> {
unsafe { self.map_unchecked_mut(|this| &mut this.inner) }
}
#[inline(always)]
#[must_use]
pub fn into_inner(self) -> I {
self.inner
}
}
impl<I> AsyncRead for FuturesAsyncRead<I>
where
I: AsyncInput<Item = u8>,
{
#[inline]
fn poll_read(self: Pin<&mut Self>, cx: &mut Context<'_>, output: &mut [u8]) -> Poll<std::io::Result<usize>> {
if output.is_empty() {
return Poll::Ready(Ok(0));
}
self.get_pin_mut().poll_read(cx, output)
}
}