tower_hyper/client/
future.rs

1use crate::Body;
2use futures::{Async, Future, Poll};
3use hyper::Response;
4
5/// Lift a hyper ResponseFuture to one which returns a `tower_http::Body`.
6#[derive(Debug)]
7pub struct ResponseFuture<F> {
8    pub(super) inner: F,
9}
10
11impl<F> Future for ResponseFuture<F>
12where
13    F: Future<Item = Response<hyper::Body>, Error = hyper::Error>,
14{
15    type Item = Response<Body>;
16    type Error = hyper::Error;
17
18    fn poll(&mut self) -> Poll<Self::Item, Self::Error> {
19        match self.inner.poll() {
20            Ok(futures::Async::Ready(body)) => {
21                let body = body.map(Body::from);
22                Ok(Async::Ready(body))
23            }
24            Ok(Async::NotReady) => Ok(Async::NotReady),
25            Err(e) => Err(e),
26        }
27    }
28}