syspulse-core 0.1.2

Core library for syspulse daemon manager
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

use async_trait::async_trait;
use std::time::Duration;

use super::HealthChecker;
use crate::daemon::{HealthCheckSpec, HealthStatus};
use crate::error::{Result, SyspulseError};

pub struct HttpHealthChecker {
    spec: HealthCheckSpec,
    client: reqwest::Client,
}

impl HttpHealthChecker {
    pub fn new(spec: HealthCheckSpec) -> Self {
        let client = reqwest::Client::builder()
            .timeout(Duration::from_secs(spec.timeout_secs))
            .build()
            .unwrap_or_default();
        Self { spec, client }
    }
}

#[async_trait]
impl HealthChecker for HttpHealthChecker {
    async fn check(&self) -> Result<HealthStatus> {
        match self.client.get(&self.spec.target).send().await {
            Ok(resp) => {
                if resp.status().is_success() {
                    Ok(HealthStatus::Healthy)
                } else {
                    tracing::debug!(
                        url = %self.spec.target,
                        status = %resp.status(),
                        "HTTP health check returned non-2xx"
                    );
                    Ok(HealthStatus::Unhealthy)
                }
            }
            Err(e) => {
                tracing::debug!(
                    url = %self.spec.target,
                    error = %e,
                    "HTTP health check failed"
                );
                if e.is_timeout() {
                    Err(SyspulseError::Timeout(Duration::from_secs(
                        self.spec.timeout_secs,
                    )))
                } else {
                    Ok(HealthStatus::Unhealthy)
                }
            }
        }
    }

    fn spec(&self) -> &HealthCheckSpec {
        &self.spec
    }
}