ymir 0.3.0

Core library for building scalable application using 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

use std::sync::LazyLock;

use axum::{extract::Request, middleware::Next, response::Response};
use http::HeaderValue;
use regex::Regex;

#[derive(Debug, Clone)]
pub struct RequestId(String);

impl RequestId {
    /// Get the request id
    #[must_use]
    pub fn get(&self) -> &str {
        self.0.as_str()
    }
}

const X_REQUEST_ID: &str = "x-request-id";
const MAX_LEN: usize = 255;

static ID_CLEANUP: LazyLock<Regex> = LazyLock::new(|| Regex::new(r"[^\w\-@]").unwrap());

pub async fn request_id_middleware(mut request: Request, next: Next) -> Response {
    let header_request_id = request.headers().get(X_REQUEST_ID).cloned();
    let request_id = make_request_id(header_request_id);
    request
        .extensions_mut()
        .insert(RequestId(request_id.clone()));
    let mut res = next.run(request).await;

    if let Ok(v) = HeaderValue::from_str(request_id.as_str()) {
        res.headers_mut().insert(X_REQUEST_ID, v);
    } else {
        tracing::warn!("could not set request ID into response headers: `{request_id}`",);
    }
    res
}

fn make_request_id(maybe_request_id: Option<HeaderValue>) -> String {
    maybe_request_id
        .and_then(|hdr| {
            let id: Option<String> = hdr.to_str().ok().map(|s| {
                ID_CLEANUP
                    .replace_all(s, "")
                    .chars()
                    .take(MAX_LEN)
                    .collect()
            });
            id.filter(|s| !s.is_empty())
        })
        .unwrap_or_else(|| ulid::Ulid::new().to_string())
}