athena_rs 3.26.3

Hyper performant polyglot Database driver
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

//! Cluster health payload assembly helpers.
//!
//! This module centralizes mapping from probe tuples to mirror payload rows and
//! assembly of the top-level cluster health response.

use crate::AppState;
use crate::api::health_capabilities::local_capabilities;
use crate::api::health_contracts::{ClusterHealthPayload, ClusterMirrorHealth};
use crate::api::health_status::{
    athena_deadpool_status, gateway_auth_store_status, gateway_benchmark_client_status,
};

/// Applies cluster probe metrics side effects and returns mirror payload rows.
pub(super) fn apply_cluster_probe_metrics(
    app_state: &AppState,
    probes: Vec<(String, ClusterMirrorHealth)>,
) -> Vec<ClusterMirrorHealth> {
    probes
        .into_iter()
        .map(|(url, probe)| {
            app_state.metrics_state.set_cluster_probe(
                &url,
                crate::api::metrics::ClusterProbeMetric {
                    up: probe.status == "online",
                    latency_ms: probe.latency_ms,
                    download_bytes_per_sec: probe.download_bytes_per_sec,
                },
            );
            probe
        })
        .collect()
}

/// Builds the cluster health payload from app status and mirrored probe rows.
pub(super) fn build_cluster_health_payload(
    app_state: &AppState,
    mirrors: Vec<ClusterMirrorHealth>,
) -> ClusterHealthPayload {
    ClusterHealthPayload {
        message: "Athena is online".to_string(),
        version: env!("CARGO_PKG_VERSION").to_string(),
        athena_api: "online".to_string(),
        athena_deadpool: athena_deadpool_status(app_state).to_string(),
        athena_scylladb: "online".to_string(),
        gateway_auth_store: gateway_auth_store_status(app_state).to_string(),
        gateway_benchmark_client: gateway_benchmark_client_status(app_state).to_string(),
        cargo_toml_version: env!("CARGO_PKG_VERSION").to_string(),
        capabilities: local_capabilities(),
        mirrors,
    }
}