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
use futures::{Async, Poll, Stream}; use lapin::Consumer as ConsumerInner; use log::trace; use crate::{confirmation::Watcher, message::Delivery, Error}; #[derive(Clone, Debug)] pub struct Consumer(pub(crate) ConsumerInner); impl Stream for Consumer { type Item = Delivery; type Error = Error; fn poll(&mut self) -> Poll<Option<Delivery>, Error> { trace!("consumer poll; polling transport"); let mut inner = self.0.inner(); trace!( "consumer poll; acquired inner lock, consumer_tag={}", inner.tag() ); if !inner.has_task() { inner.set_task(Box::new(Watcher::default())); } if let Some(delivery) = inner.next_delivery() { match delivery { Ok(Some(delivery)) => { trace!( "delivery; consumer_tag={}, delivery_tag={:?}", inner.tag(), delivery.delivery_tag ); Ok(Async::Ready(Some(delivery))) } Ok(None) => { trace!("consumer canceled; consumer_tag={}", inner.tag()); Ok(Async::Ready(None)) } Err(error) => Err(error), } } else { trace!("delivery; status=NotReady, consumer_tag={}", inner.tag()); Ok(Async::NotReady) } } }