diff options
Diffstat (limited to 'third_party/rust/http-body/src/next.rs')
-rw-r--r-- | third_party/rust/http-body/src/next.rs | 31 |
1 files changed, 31 insertions, 0 deletions
diff --git a/third_party/rust/http-body/src/next.rs b/third_party/rust/http-body/src/next.rs new file mode 100644 index 0000000000..fc87ffcf01 --- /dev/null +++ b/third_party/rust/http-body/src/next.rs @@ -0,0 +1,31 @@ +use crate::Body; + +use core::future::Future; +use core::pin::Pin; +use core::task; + +#[must_use = "futures don't do anything unless polled"] +#[derive(Debug)] +/// Future that resolves to the next data chunk from `Body` +pub struct Data<'a, T: ?Sized>(pub(crate) &'a mut T); + +impl<'a, T: Body + Unpin + ?Sized> Future for Data<'a, T> { + type Output = Option<Result<T::Data, T::Error>>; + + fn poll(mut self: Pin<&mut Self>, ctx: &mut task::Context<'_>) -> task::Poll<Self::Output> { + Pin::new(&mut self.0).poll_data(ctx) + } +} + +#[must_use = "futures don't do anything unless polled"] +#[derive(Debug)] +/// Future that resolves to the optional trailers from `Body` +pub struct Trailers<'a, T: ?Sized>(pub(crate) &'a mut T); + +impl<'a, T: Body + Unpin + ?Sized> Future for Trailers<'a, T> { + type Output = Result<Option<http::HeaderMap>, T::Error>; + + fn poll(mut self: Pin<&mut Self>, ctx: &mut task::Context<'_>) -> task::Poll<Self::Output> { + Pin::new(&mut self.0).poll_trailers(ctx) + } +} |