io-tether 0.6.1

//! Tether implementations for TCP sockets
use super::*;

use tokio::net::{TcpStream, ToSocketAddrs};

/// Wrapper for building [`TcpStream`]s
#[derive(Debug, Clone, PartialEq, Eq, PartialOrd, Ord)]
pub struct TcpConnector<A>(A);

impl<A> TcpConnector<A> {
    pub fn new(address: A) -> Self {
        Self(address)
    }

    /// Get a ref to the address
    #[inline]
    pub fn get_addr(&self) -> &A {
        &self.0
    }

    /// Get a mutable ref to the address
    #[inline]
    pub fn get_addr_mut(&mut self) -> &mut A {
        &mut self.0
    }
}

impl<A, R> Tether<TcpConnector<A>, R>
where
    R: Resolver<TcpConnector<A>>,
    A: 'static + ToSocketAddrs + Clone + Send + Sync,
{
    /// Helper function for building a TCP connection
    pub async fn connect_tcp(address: A, resolver: R) -> Result<Self, std::io::Error> {
        let connector = TcpConnector::new(address);
        Tether::connect(connector, resolver).await
    }
}

impl<A> Connector for TcpConnector<A>
where
    A: 'static + ToSocketAddrs + Clone + Send + Sync,
{
    type Output = TcpStream;

    fn connect(&mut self) -> PinFut<Result<Self::Output, std::io::Error>> {
        let address = self.0.clone();
        Box::pin(TcpStream::connect(address))
    }
}