axum-core/body: Implement Stream::size_hint for BodyDataStream (#…

…3195)
tokio-rs · Jan 24, 2025 · b0b8174 · b0b8174
1 parent 0cb6865
commit b0b8174
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/axum-core/src/body.rs b/axum-core/src/body.rs
@@ -155,6 +155,14 @@ impl Stream for BodyDataStream {
             }
         }
     }
+
+    #[inline]
+    fn size_hint(&self) -> (usize, Option<usize>) {
+        let size_hint = self.inner.size_hint();
+        let lower = usize::try_from(size_hint.lower()).unwrap_or_default();
+        let upper = size_hint.upper().and_then(|v| usize::try_from(v).ok());
+        (lower, upper)
+    }
 }
 
 impl http_body::Body for BodyDataStream {