actionqueue-daemon 0.1.2

HTTP daemon with REST API and Prometheus metrics for the ActionQueue task queue engine.
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

//! Metrics route module.
//!
//! This module provides `GET /metrics`, which exports daemon-local Prometheus
//! metrics in text exposition format.

use axum::extract::State;
use axum::http::{HeaderValue, StatusCode};
use axum::response::IntoResponse;

/// Handles `GET /metrics` export requests.
#[tracing::instrument(skip_all)]
pub async fn handle(state: State<super::RouterState>) -> impl IntoResponse {
    crate::metrics::runs::update(state.0.as_ref());
    crate::metrics::attempts::update(state.0.as_ref());
    crate::metrics::wal::update(state.0.as_ref());
    crate::metrics::recovery::update(state.0.as_ref());

    match state.metrics.encode_text() {
        Ok(encoded) => {
            let mut response = encoded.body.into_response();
            if let Ok(value) = HeaderValue::from_str(&encoded.content_type) {
                response.headers_mut().insert(axum::http::header::CONTENT_TYPE, value);
            }
            response
        }
        Err(_) => {
            (StatusCode::INTERNAL_SERVER_ERROR, "metrics_encode_failed: failed to encode metrics\n")
                .into_response()
        }
    }
}

/// Registers metrics routes when enabled.
pub fn register_routes(
    router: axum::Router<super::RouterState>,
    metrics_enabled: bool,
) -> axum::Router<super::RouterState> {
    if !metrics_enabled {
        return router;
    }

    router.route("/metrics", axum::routing::get(handle))
}