tower-connect.rs

use tower_service::Service;
use futures::Future;
use tokio_io::{AsyncRead, AsyncWrite};
use std::net::ToSocketAddrs;

pub trait ConnectService<A: ToSocketAddrs>{
    type Response: AsyncRead + AsyncWrite;
    type Error;
    type Service: Service<A>;
    type Future: Future<Item =  Self::Response, Error = Self::Error>;

    fn connect(&self, target: A) -> Self::Future;
}

impl<A, S, C, Request> ConnectService<A> for C
    where A: ToSocketAddrs,
          C: Service<A>,
          C::Response: AsyncRead + AsyncWrite,
          S: Service<Request>,
{
    type Response = C::Response;
    type Error = S::Error;
    type Service = S;
    type Error = C::Error;
    type Future = C::Future;

    fn connect(&mut self, target: A) -> Self::Future {
        ConnectService::connect(self, target)
    }
}