pong-rs 0.3.3

A network monitoring tool that performs ICMP, TCP, and HTTP pings
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

use crate::executor::Executor;
use crate::ping_error::PingError;
use crate::task::tcp::tcp_ping::TcpPing;
use async_trait::async_trait;
use log::trace;
use std::net::SocketAddr;
use std::time::Duration;
use wheel_rs::dns_utils::parse_host_port;

#[derive(Clone)]
pub struct TcpExecutor {
    ip_addr: String,
    port: u16,
    tcp_ping: TcpPing,
    timeout: Duration,
}

impl TcpExecutor {
    /// # 构造函数
    /// ## 参数
    /// * `host_port` - 要ping的主机名及端口号
    /// * `timeout` - 一个 `Duration`，表示超时时间
    pub fn new(host_port: String, timeout: Duration) -> Self {
        // 解析主机的字符串成IP地址和端口号
        let (ip_addr, port) = parse_host_port(&host_port).unwrap();
        // 创建SocketAddr对象
        let socket_addr = SocketAddr::new(ip_addr, port);

        let tcp_ping = TcpPing::new(socket_addr);

        Self {
            ip_addr: ip_addr.to_string(),
            port,
            tcp_ping,
            timeout,
        }
    }
}

#[async_trait]
impl Executor for TcpExecutor {
    fn get_name(&self) -> String {
        String::from("TCP")
    }

    async fn exec(&self) -> Result<(), PingError> {
        trace!(
            "开始执行 TCP 任务: ping {}",
            format!("{}:{}", self.ip_addr, self.port)
        );
        self.tcp_ping.ping(self.timeout)
    }
}