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

/// Helper methods to act on hyper::Body
use futures::stream::{Stream, StreamExt};
use hyper::body::Bytes;

/// Additional function for hyper::Body
pub trait BodyExt {
    /// Raw body type
    type Raw;

    /// Error if we can't gather up the raw body
    type Error;

    /// Collect the body into a raw form
    fn into_raw(self) -> futures::future::BoxFuture<'static, Result<Self::Raw, Self::Error>>;
}

impl<T, E> BodyExt for T
where
    T: Stream<Item = Result<Bytes, E>> + Unpin + Send + 'static,
{
    type Raw = Vec<u8>;
    type Error = E;

    fn into_raw(mut self) -> futures::future::BoxFuture<'static, Result<Self::Raw, Self::Error>> {
        Box::pin(async {
            let mut raw = Vec::new();
            while let (Some(chunk), rest) = self.into_future().await {
                raw.extend_from_slice(&chunk?);
                self = rest;
            }
            Ok(raw)
        })
    }
}