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);