aboutsummaryrefslogtreecommitdiff
path: root/src/api.rs
blob: 56ee6f0402f6bd2ed3fc4610d189f2e2bd4340c2 (plain)
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
use crate::{Config, Status, STATUS};
use serde_json::{json, Value};
use std::{ops::Deref, sync::Arc};

pub async fn make_json_response(config: Arc<Config>) -> Value {
    let status = STATUS.read().await;
    let status = status.deref();
    let total_err = status
        .values()
        .map(|s| s.status.is_err() as usize)
        .sum::<usize>();
    let total_ok = status
        .values()
        .map(|s| s.status.is_ok() as usize)
        .sum::<usize>();

    let mut titles_failed = Vec::new();
    let services = config
        .services
        .iter()
        .enumerate()
        .map(|(i, s)| {
            let total_err = status
                .range((i, usize::MIN)..(i, usize::MAX))
                .map(|(_, s)| s.status.is_err() as usize)
                .sum::<usize>();
            let total_ok = status
                .range((i, usize::MIN)..(i, usize::MAX))
                .map(|(_, s)| s.status.is_ok() as usize)
                .sum::<usize>();
            if total_err != 0 {
                titles_failed.push(&s.title)
            }
            json!({
                "status": if total_err == 0 {
                    "ok"
                } else if total_ok == 0 {
                    "failed"
                } else {
                    "degraded"
                },
                "title": &s.title,
            })
        })
        .collect::<Vec<_>>();

    json!({
        "status": if total_err == 0 {
            "ok"
        } else if total_ok == 0 {
            "failed"
        } else {
            "degraded"
        },
        "total_ok": total_ok,
        "total_err": total_err,
        "services": services,
        "failed": titles_failed
    })
}

impl Status {
    pub fn json(&self) -> Value {
        json!({
            "ok": self.status.is_ok(),
            "message": match &self.status { Ok(m) => m, Err(m) => m },
            "time": self.time.timestamp()
        })
    }
}