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
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107

use crate::MakeWriter;

use std::net::ToSocketAddrs;

impl MakeWriter for std::vec::IntoIter<std::net::SocketAddr> {
    type Writer = std::net::TcpStream;

    #[inline(always)]
    fn make(&self) -> std::io::Result<Self::Writer> {
        for addr in self.as_slice().iter() {
            match std::net::TcpStream::connect_timeout(addr, core::time::Duration::from_secs(1)) {
                Ok(socket) => return Ok(socket),
                Err(_) => continue,
            }
        }

        Err(std::io::Error::new(std::io::ErrorKind::NotFound, "cannot connect to fluentd"))
    }
}

///Creates writer by resolving address from provided string.
impl MakeWriter for &'static str {
    type Writer = std::net::TcpStream;

    #[inline(always)]
    fn make(&self) -> std::io::Result<Self::Writer> {
        let addrs = self.to_socket_addrs()?;

        for addr in addrs.as_slice().iter() {
            match std::net::TcpStream::connect_timeout(addr, core::time::Duration::from_secs(1)) {
                Ok(socket) => return Ok(socket),
                Err(_) => continue,
            }
        }

        Err(std::io::Error::new(std::io::ErrorKind::NotFound, "cannot connect to fluentd"))
    }
}

///Creates writer by resolving address from provided string and port.
impl MakeWriter for (&'static str, u16) {
    type Writer = std::net::TcpStream;

    #[inline(always)]
    fn make(&self) -> std::io::Result<Self::Writer> {
        let addrs = self.to_socket_addrs()?;

        for addr in addrs.as_slice().iter() {
            match std::net::TcpStream::connect_timeout(addr, core::time::Duration::from_secs(1)) {
                Ok(socket) => return Ok(socket),
                Err(_) => continue,
            }
        }

        Err(std::io::Error::new(std::io::ErrorKind::NotFound, "cannot connect to fluentd"))
    }
}

impl MakeWriter for std::net::SocketAddr {
    type Writer = std::net::TcpStream;

    #[inline(always)]
    fn make(&self) -> std::io::Result<Self::Writer> {
        match std::net::TcpStream::connect_timeout(self, core::time::Duration::from_secs(1)) {
            Ok(socket) => Ok(socket),
            Err(_) => Err(std::io::Error::new(std::io::ErrorKind::NotFound, "cannot connect to fluentd")),
        }
    }
}

impl MakeWriter for [std::net::SocketAddr; 1] {
    type Writer = std::net::TcpStream;

    #[inline(always)]
    fn make(&self) -> std::io::Result<Self::Writer> {
        match std::net::TcpStream::connect_timeout(&self[0], core::time::Duration::from_secs(1)) {
            Ok(socket) => Ok(socket),
            Err(_) => Err(std::io::Error::new(std::io::ErrorKind::NotFound, "cannot connect to fluentd")),
        }
    }
}

//While we can use generics, it doesn't really make sense to store addresses in such big arrays.
macro_rules! impl_for_socket_addr_array {
    ($($idx:literal),+) => {

        $(
            impl MakeWriter for [std::net::SocketAddr; $idx] {
                type Writer = std::net::TcpStream;

                #[inline(always)]
                fn make(&self) -> std::io::Result<Self::Writer> {
                    for addr in self {
                        match std::net::TcpStream::connect_timeout(addr, core::time::Duration::from_secs(1)) {
                            Ok(socket) => return Ok(socket),
                            Err(_) => continue,
                        }
                    }

                    Err(std::io::Error::new(std::io::ErrorKind::NotFound, "cannot connect to fluentd"))
                }
            }
        )+
    }
}

impl_for_socket_addr_array!(2,3,4,5,6,7,8,9,10,11,12);