From 17e424112d53385142aa430641910c384c4cbe5a Mon Sep 17 00:00:00 2001
From: Ruben De Smet <ruben.de.smet@rubdos.be>
Date: Wed, 18 Dec 2019 06:01:29 +0100
Subject: [PATCH] time: impl Stream for DelayQueue (#1975)

---
 tokio/src/time/delay_queue.rs | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/tokio/src/time/delay_queue.rs b/tokio/src/time/delay_queue.rs
index eb5000d2f..9e69bd3ee 100644
--- a/tokio/src/time/delay_queue.rs
+++ b/tokio/src/time/delay_queue.rs
@@ -730,6 +730,17 @@ impl<T> Default for DelayQueue<T> {
     }
 }
 
+#[cfg(feature = "stream")]
+impl<T> futures_core::Stream for DelayQueue<T> {
+    // DelayQueue seems much more specific, where a user may care that it
+    // has reached capacity, so return those errors instead of panicking.
+    type Item = Result<Expired<T>, Error>;
+
+    fn poll_next(self: Pin<&mut Self>, cx: &mut task::Context<'_>) -> Poll<Option<Self::Item>> {
+        DelayQueue::poll_expired(self.get_mut(), cx)
+    }
+}
+
 impl<T> wheel::Stack for Stack<T> {
     type Owned = usize;
     type Borrowed = usize;