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 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71
use crate::{Sink, Poll}; use futures_core::task::LocalWaker; use futures_channel::mpsc::{Sender, SendError, UnboundedSender}; use std::pin::Pin; impl<T> Sink for Sender<T> { type SinkItem = T; type SinkError = SendError; fn poll_ready(mut self: Pin<&mut Self>, lw: &LocalWaker) -> Poll<Result<(), Self::SinkError>> { (*self).poll_ready(lw) } fn start_send(mut self: Pin<&mut Self>, msg: T) -> Result<(), Self::SinkError> { (*self).start_send(msg) } fn poll_flush(self: Pin<&mut Self>, _: &LocalWaker) -> Poll<Result<(), Self::SinkError>> { Poll::Ready(Ok(())) } fn poll_close(mut self: Pin<&mut Self>, _: &LocalWaker) -> Poll<Result<(), Self::SinkError>> { self.close_channel(); Poll::Ready(Ok(())) } } impl<T> Sink for UnboundedSender<T> { type SinkItem = T; type SinkError = SendError; fn poll_ready(self: Pin<&mut Self>, lw: &LocalWaker) -> Poll<Result<(), Self::SinkError>> { UnboundedSender::poll_ready(&*self, lw) } fn start_send(mut self: Pin<&mut Self>, msg: T) -> Result<(), Self::SinkError> { UnboundedSender::start_send(&mut *self, msg) } fn poll_flush(self: Pin<&mut Self>, _: &LocalWaker) -> Poll<Result<(), Self::SinkError>> { Poll::Ready(Ok(())) } fn poll_close(self: Pin<&mut Self>, _: &LocalWaker) -> Poll<Result<(), Self::SinkError>> { self.close_channel(); Poll::Ready(Ok(())) } } impl<'a, T> Sink for &'a UnboundedSender<T> { type SinkItem = T; type SinkError = SendError; fn poll_ready(self: Pin<&mut Self>, lw: &LocalWaker) -> Poll<Result<(), Self::SinkError>> { UnboundedSender::poll_ready(*self, lw) } fn start_send(self: Pin<&mut Self>, msg: T) -> Result<(), Self::SinkError> { self.unbounded_send(msg) .map_err(|err| err.into_send_error()) } fn poll_flush(self: Pin<&mut Self>, _: &LocalWaker) -> Poll<Result<(), Self::SinkError>> { Poll::Ready(Ok(())) } fn poll_close(self: Pin<&mut Self>, _: &LocalWaker) -> Poll<Result<(), Self::SinkError>> { self.close_channel(); Poll::Ready(Ok(())) } }