via 1.0.1

An async multi-threaded web framework for people who appreciate simplicity.
Documentation 
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
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56

use bytes::Bytes;
use futures_core::Stream;
use http_body::{Body, Frame, SizeHint};
use std::pin::Pin;
use std::task::{Context, Poll};

use crate::body::size_hint;
use crate::error::Error;

/// Convert a `Stream + Send` into an `impl Body`.
#[must_use = "streams do nothing unless polled"]
pub struct StreamAdapter<T> {
    /// The `Stream` that we are adapting to an `impl Body`.
    stream: T,
}

impl<T, E> StreamAdapter<T>
where
    T: Stream<Item = Result<Frame<Bytes>, E>> + Send + Unpin,
{
    pub fn new(stream: T) -> Self {
        Self { stream }
    }
}

impl<T: Unpin> StreamAdapter<T> {
    fn project(self: Pin<&mut Self>) -> Pin<&mut T> {
        let this = self.get_mut();
        let ptr = &mut this.stream;

        Pin::new(ptr)
    }
}

impl<T, E> Body for StreamAdapter<T>
where
    T: Stream<Item = Result<Frame<Bytes>, E>> + Send + Unpin,
    Error: From<E>,
{
    type Data = Bytes;
    type Error = Error;

    fn poll_frame(
        self: Pin<&mut Self>,
        context: &mut Context<'_>,
    ) -> Poll<Option<Result<Frame<Self::Data>, Self::Error>>> {
        self.project()
            .poll_next(context)
            .map_err(|error| error.into())
    }

    fn size_hint(&self) -> SizeHint {
        let hint = self.stream.size_hint();
        size_hint::from_stream_for_body(hint)
    }
}