ceres-server 0.7.0

REST API server for Ceres harvesting, embedding, and search workflows
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
71
72
73
74
75
76
77
78

//! Container liveness and database readiness endpoints.

use axum::{Json, extract::State, http::StatusCode};

use crate::dto::{HealthResponse, ProbeResponse, ServiceStatus};
use crate::state::AppState;

/// Liveness probe. This only verifies that the HTTP process can respond.
#[utoipa::path(
    get,
    path = "/api/v1/health/live",
    responses(
        (status = 200, description = "Server process is alive", body = ProbeResponse),
    ),
    tag = "system"
)]
pub async fn liveness_check() -> Json<ProbeResponse> {
    Json(ProbeResponse {
        status: "alive".to_string(),
        version: env!("CARGO_PKG_VERSION").to_string(),
    })
}

/// Readiness probe. Returns 503 until the database is reachable.
#[utoipa::path(
    get,
    path = "/api/v1/health/ready",
    responses(
        (status = 200, description = "Server and database are ready", body = HealthResponse),
        (status = 503, description = "Database is unavailable", body = HealthResponse),
    ),
    tag = "system"
)]
pub async fn readiness_check(State(state): State<AppState>) -> (StatusCode, Json<HealthResponse>) {
    database_readiness(&state).await
}

/// Backward-compatible readiness alias.
#[utoipa::path(
    get,
    path = "/api/v1/health",
    responses(
        (status = 200, description = "Server and database are ready", body = HealthResponse),
        (status = 503, description = "Database is unavailable", body = HealthResponse),
    ),
    tag = "system"
)]
pub async fn health_check(State(state): State<AppState>) -> (StatusCode, Json<HealthResponse>) {
    database_readiness(&state).await
}

async fn database_readiness(state: &AppState) -> (StatusCode, Json<HealthResponse>) {
    let db_status = match state.dataset_repo.health_check().await {
        Ok(()) => ServiceStatus {
            healthy: true,
            message: None,
        },
        Err(error) => ServiceStatus {
            healthy: false,
            message: Some(error.to_string()),
        },
    };

    let (status_code, overall_status) = if db_status.healthy {
        (StatusCode::OK, "healthy")
    } else {
        (StatusCode::SERVICE_UNAVAILABLE, "unhealthy")
    };

    (
        status_code,
        Json(HealthResponse {
            status: overall_status.to_string(),
            version: env!("CARGO_PKG_VERSION").to_string(),
            database: db_status,
        }),
    )
}