echosrv 0.3.0

A high-performance set of async echo server library built with Tokio for testing and development environments
Documentation 
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
108
109
110
111
112
113
114
115
116
117
118

use std::fmt;
use std::net::SocketAddr;
use std::path::PathBuf;
use std::str::FromStr;

/// Unified address type that supports both network and Unix domain sockets
#[derive(Debug, Clone, PartialEq, Eq)]
pub enum Address {
    /// Network address (TCP, UDP)
    Network(SocketAddr),
    /// Unix domain socket path
    Unix(PathBuf),
}

impl fmt::Display for Address {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        match self {
            Address::Network(addr) => write!(f, "{addr}"),
            Address::Unix(path) => write!(f, "unix:{}", path.display()),
        }
    }
}

impl From<SocketAddr> for Address {
    fn from(addr: SocketAddr) -> Self {
        Address::Network(addr)
    }
}

impl From<PathBuf> for Address {
    fn from(path: PathBuf) -> Self {
        Address::Unix(path)
    }
}

impl From<&str> for Address {
    fn from(s: &str) -> Self {
        if let Some(stripped) = s.strip_prefix("unix:") {
            Address::Unix(PathBuf::from(stripped))
        } else {
            Address::Network(s.parse().expect("Invalid socket address"))
        }
    }
}

impl FromStr for Address {
    type Err = crate::EchoError;

    fn from_str(s: &str) -> Result<Self, Self::Err> {
        if let Some(stripped) = s.strip_prefix("unix:") {
            Ok(Address::Unix(PathBuf::from(stripped)))
        } else {
            s.parse::<SocketAddr>()
                .map(Address::Network)
                .map_err(|e| crate::EchoError::Config(format!("Invalid socket address: {e}")))
        }
    }
}

impl Address {
    /// Returns true if this is a network address
    pub fn is_network(&self) -> bool {
        matches!(self, Address::Network(_))
    }

    /// Returns true if this is a Unix domain socket address
    pub fn is_unix(&self) -> bool {
        matches!(self, Address::Unix(_))
    }

    /// Get the network address if this is a network address
    pub fn as_network(&self) -> Option<&SocketAddr> {
        match self {
            Address::Network(addr) => Some(addr),
            _ => None,
        }
    }

    /// Get the Unix path if this is a Unix domain socket
    pub fn as_unix(&self) -> Option<&PathBuf> {
        match self {
            Address::Unix(path) => Some(path),
            _ => None,
        }
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_network_address() {
        let addr: Address = "127.0.0.1:8080".into();
        assert!(addr.is_network());
        assert!(!addr.is_unix());
        assert!(addr.as_network().is_some());
        assert!(addr.as_unix().is_none());
    }

    #[test]
    fn test_unix_address() {
        let addr: Address = "unix:/tmp/test.sock".into();
        assert!(!addr.is_network());
        assert!(addr.is_unix());
        assert!(addr.as_network().is_none());
        assert!(addr.as_unix().is_some());
    }

    #[test]
    fn test_display() {
        let net_addr: Address = "127.0.0.1:8080".into();
        let unix_addr: Address = "unix:/tmp/test.sock".into();

        assert_eq!(net_addr.to_string(), "127.0.0.1:8080");
        assert_eq!(unix_addr.to_string(), "unix:/tmp/test.sock");
    }
}