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
use {Async, Sink, Poll}; use futures_core::task; use futures_channel::mpsc::{Sender, SendError, UnboundedSender}; impl<T> Sink for Sender<T> { type SinkItem = T; type SinkError = SendError; fn poll_ready(&mut self, cx: &mut task::Context) -> Poll<(), Self::SinkError> { self.poll_ready(cx) } fn start_send(&mut self, msg: T) -> Result<(), Self::SinkError> { self.start_send(msg) } fn poll_flush(&mut self, _: &mut task::Context) -> Poll<(), Self::SinkError> { Ok(Async::Ready(())) } fn poll_close(&mut self, _: &mut task::Context) -> Poll<(), Self::SinkError> { self.close_channel(); Ok(Async::Ready(())) } } impl<T> Sink for UnboundedSender<T> { type SinkItem = T; type SinkError = SendError; fn poll_ready(&mut self, cx: &mut task::Context) -> Poll<(), Self::SinkError> { UnboundedSender::poll_ready(&*self, cx) } fn start_send(&mut self, msg: T) -> Result<(), Self::SinkError> { self.start_send(msg) } fn poll_flush(&mut self, _: &mut task::Context) -> Poll<(), Self::SinkError> { Ok(Async::Ready(())) } fn poll_close(&mut self, _: &mut task::Context) -> Poll<(), Self::SinkError> { self.close_channel(); Ok(Async::Ready(())) } } impl<'a, T> Sink for &'a UnboundedSender<T> { type SinkItem = T; type SinkError = SendError; fn poll_ready(&mut self, cx: &mut task::Context) -> Poll<(), Self::SinkError> { UnboundedSender::poll_ready(*self, cx) } fn start_send(&mut self, msg: T) -> Result<(), Self::SinkError> { self.unbounded_send(msg) .map_err(|err| err.into_send_error()) } fn poll_flush(&mut self, _: &mut task::Context) -> Poll<(), Self::SinkError> { Ok(Async::Ready(())) } fn poll_close(&mut self, _: &mut task::Context) -> Poll<(), Self::SinkError> { self.close_channel(); Ok(Async::Ready(())) } }