use futures::{try_ready, Async, Future, Poll};
use hyper::client::connect::Connect;
use tower_service::Service;
pub use hyper::client::connect::{Destination, HttpConnector};
#[derive(Debug)]
pub struct Connector<C> {
inner: C,
}
#[derive(Debug)]
pub struct ConnectorFuture<C>
where
C: Connect,
{
inner: C::Future,
}
impl<C> Connector<C>
where
C: Connect,
{
pub fn new(inner: C) -> Self {
Connector { inner }
}
}
impl<C> Service<Destination> for Connector<C>
where
C: Connect,
{
type Response = C::Transport;
type Error = C::Error;
type Future = ConnectorFuture<C>;
fn poll_ready(&mut self) -> Poll<(), Self::Error> {
Ok(().into())
}
fn call(&mut self, target: Destination) -> Self::Future {
let fut = self.inner.connect(target);
ConnectorFuture { inner: fut }
}
}
impl<C> Future for ConnectorFuture<C>
where
C: Connect,
{
type Item = C::Transport;
type Error = C::Error;
fn poll(&mut self) -> Poll<Self::Item, Self::Error> {
let (transport, _) = try_ready!(self.inner.poll());
Ok(Async::Ready(transport))
}
}