resuma 0.4.1

Resuma — SSR + Resumability + Islands + Server Actions + JS Bridge for Rust
Documentation 
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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104

//! Production ops helpers shared by `ResumaApp` and `FlowApp`:
//! health/readiness probes, request-id + latency tracing, and graceful shutdown.

use std::time::Instant;

use axum::body::Body;
use axum::http::{HeaderName, HeaderValue, Request, StatusCode};
use axum::middleware::Next;
use axum::response::{IntoResponse, Response};

/// Header carrying a per-request correlation id (read from the client if present,
/// otherwise generated). Mirrored back on the response.
pub const REQUEST_ID_HEADER: &str = "x-request-id";

/// Liveness probe path. Returns `200 OK` as soon as the process can serve HTTP.
pub const HEALTH_PATH: &str = "/health";

/// Readiness probe path. Returns `200 OK` when the app is ready to receive traffic.
pub const READY_PATH: &str = "/ready";

/// Liveness handler — the process is up and the axum stack is serving.
pub async fn health() -> Response {
    (StatusCode::OK, "ok").into_response()
}

/// Readiness handler — safe default returns ready; apps with external
/// dependencies (DB, cache) can register their own `/ready` page to override.
pub async fn ready() -> Response {
    (StatusCode::OK, "ready").into_response()
}

/// Generate or propagate a request id, emit a tracing span with method/path/
/// status/latency, and echo `x-request-id` on the response.
pub async fn request_id_middleware(mut req: Request<Body>, next: Next) -> Response {
    let incoming = req
        .headers()
        .get(REQUEST_ID_HEADER)
        .and_then(|v| v.to_str().ok())
        .filter(|s| !s.is_empty() && s.len() <= 128)
        .map(|s| s.to_string());
    let request_id = incoming.unwrap_or_else(super::security::random_token);

    let method = req.method().clone();
    let path = req.uri().path().to_string();

    // Make the id available to handlers via request extensions.
    req.extensions_mut().insert(RequestId(request_id.clone()));
    if let Ok(value) = HeaderValue::from_str(&request_id) {
        req.headers_mut()
            .insert(HeaderName::from_static("x-request-id"), value);
    }

    let started = Instant::now();
    let mut res = next.run(req).await;
    let latency_ms = started.elapsed().as_millis();
    let status = res.status().as_u16();

    tracing::info!(
        request_id = %request_id,
        method = %method,
        path = %path,
        status,
        latency_ms,
        "request"
    );

    if let Ok(value) = HeaderValue::from_str(&request_id) {
        res.headers_mut()
            .insert(HeaderName::from_static("x-request-id"), value);
    }
    res
}

/// Per-request correlation id stored in request extensions.
#[derive(Debug, Clone)]
pub struct RequestId(pub String);

/// Resolve when the process receives a shutdown signal (`Ctrl+C` on all
/// platforms, plus `SIGTERM` on Unix for Fly.io / Kubernetes rolling deploys).
pub async fn shutdown_signal() {
    let ctrl_c = async {
        let _ = tokio::signal::ctrl_c().await;
    };

    #[cfg(unix)]
    let terminate = async {
        match tokio::signal::unix::signal(tokio::signal::unix::SignalKind::terminate()) {
            Ok(mut sig) => {
                sig.recv().await;
            }
            Err(_) => std::future::pending::<()>().await,
        }
    };

    #[cfg(not(unix))]
    let terminate = std::future::pending::<()>();

    tokio::select! {
        _ = ctrl_c => {}
        _ = terminate => {}
    }

    tracing::info!("shutdown signal received; draining connections");
}