diff options
Diffstat (limited to 'src/io/util/write.rs')
-rw-r--r-- | src/io/util/write.rs | 17 |
1 files changed, 13 insertions, 4 deletions
diff --git a/src/io/util/write.rs b/src/io/util/write.rs index 433a421..92169eb 100644 --- a/src/io/util/write.rs +++ b/src/io/util/write.rs @@ -1,17 +1,22 @@ use crate::io::AsyncWrite; +use pin_project_lite::pin_project; use std::future::Future; use std::io; +use std::marker::PhantomPinned; use std::pin::Pin; use std::task::{Context, Poll}; -cfg_io_util! { +pin_project! { /// A future to write some of the buffer to an `AsyncWrite`. #[derive(Debug)] #[must_use = "futures do nothing unless you `.await` or poll them"] pub struct Write<'a, W: ?Sized> { writer: &'a mut W, buf: &'a [u8], + // Make this future `!Unpin` for compatibility with async trait methods. + #[pin] + _pin: PhantomPinned, } } @@ -21,7 +26,11 @@ pub(crate) fn write<'a, W>(writer: &'a mut W, buf: &'a [u8]) -> Write<'a, W> where W: AsyncWrite + Unpin + ?Sized, { - Write { writer, buf } + Write { + writer, + buf, + _pin: PhantomPinned, + } } impl<W> Future for Write<'_, W> @@ -30,8 +39,8 @@ where { type Output = io::Result<usize>; - fn poll(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<io::Result<usize>> { - let me = &mut *self; + fn poll(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<io::Result<usize>> { + let me = self.project(); Pin::new(&mut *me.writer).poll_write(cx, me.buf) } } |