fuel-web-utils 0.0.29

Fuel library for web utils
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

use std::time::{Duration, Instant};

use async_trait::async_trait;
use serde::{Deserialize, Serialize};

#[async_trait]
pub trait StateProvider: Send + Sync + Clone + 'static {
    async fn is_healthy(&self) -> bool;
    async fn get_health(&self) -> serde_json::Value;
    async fn get_metrics(&self) -> String;
}

#[derive(Clone, Debug, Serialize, Deserialize)]
pub struct DefaultHealthResponse {
    pub uptime: u64,
    pub is_healthy: bool,
}

#[derive(Clone)]
pub struct DefaultServerState {
    pub start_time: Instant,
}

impl Default for DefaultServerState {
    fn default() -> Self {
        Self::new()
    }
}

impl DefaultServerState {
    pub fn new() -> Self {
        Self {
            start_time: Instant::now(),
        }
    }

    pub async fn get_health(&self) -> DefaultHealthResponse {
        DefaultHealthResponse {
            uptime: self.uptime().as_secs(),
            is_healthy: true,
        }
    }

    pub fn uptime(&self) -> Duration {
        self.start_time.elapsed()
    }
}

#[async_trait]
impl StateProvider for DefaultServerState {
    async fn is_healthy(&self) -> bool {
        true
    }

    async fn get_health(&self) -> serde_json::Value {
        serde_json::to_value(DefaultHealthResponse {
            uptime: self.uptime().as_secs(),
            is_healthy: true,
        })
        .unwrap_or(serde_json::json!({}))
    }

    async fn get_metrics(&self) -> String {
        format!("uptime: {}s", self.uptime().as_secs())
    }
}