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

//! REST API endpoint for health check

use crate::grpc::client::GrpcClients;
use axum::extract::Extension;
use hyper::StatusCode;
use svc_storage_client_grpc::ClientConnect;

/// Health check for load balancing
#[utoipa::path(
    get,
    path = "/health",
    tag = "svc-telemetry",
    responses(
        (status = 200, description = "Service is healthy, all dependencies running."),
        (status = 503, description = "Service is unhealthy, one or more dependencies unavailable.")
    )
)]
pub async fn health_check(
    Extension(grpc_clients): Extension<GrpcClients>,
) -> Result<(), StatusCode> {
    rest_debug!("(health_check) entry.");

    let mut ok = true;

    if grpc_clients.storage.adsb.get_client().await.is_err() {
        let error_msg = "svc-storage adsb unavailable".to_string();
        rest_error!("(health_check adsb) {}.", &error_msg);
        println!("(health_check adsb) {}.", &error_msg);
        ok = false;
    }

    if grpc_clients.gis.get_client().await.is_err() {
        let error_msg = "svc-gis unavailable".to_string();
        rest_error!("(health_check gis) {}.", &error_msg);
        println!("(health_check gis) {}.", &error_msg);
        ok = false;
    }

    match ok {
        true => {
            rest_debug!("(health_check) healthy, all dependencies running.");
            Ok(())
        }
        false => {
            rest_error!("(health_check) unhealthy, 1+ dependencies down.");
            Err(StatusCode::SERVICE_UNAVAILABLE)
        }
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[tokio::test]
    async fn test_health_check_success() {
        // Mock the GrpcClients extension
        let config = crate::config::Config::default();
        let grpc_clients = GrpcClients::default(config);
        let extension = Extension(grpc_clients);

        // Call the health_check function
        let result = health_check(extension).await;

        // Assert the expected result
        println!("{:?}", result);
        assert!(result.is_ok());
    }
}