Implement FusedStream for Interval

tokio-rs · Jan 5, 2020 · 6d58acb · 6d58acb
1 parent e38b16c
commit 6d58acb
Showing 1 changed file with 13 additions and 5 deletions.
diff --git a/tokio/src/time/interval.rs b/tokio/src/time/interval.rs
@@ -129,11 +129,19 @@ impl Interval {
     }
 }
 
-#[cfg(feature = "stream")]
-impl crate::stream::Stream for Interval {
-    type Item = Instant;
+cfg_stream! {
+    impl futures_core::FusedStream for Interval {
+        fn is_terminated(&self) -> bool {
+            // NB: intervals never terminate.
+            false
+        }
+    }
+
+    impl crate::stream::Stream for Interval {
+        type Item = Instant;
 
-    fn poll_next(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Option<Instant>> {
-        Poll::Ready(Some(ready!(self.poll_tick(cx))))
+        fn poll_next(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Option<Instant>> {
+            Poll::Ready(Some(ready!(self.poll_tick(cx))))
+        }
     }
 }