請求鏈路Id

Request Id 中介軟體具有高度靈活性,生成器(IdGenerator) 用於產生 ID,您可以定義自己的 ID 生成器,只需實現 IdGenerator 這個 trait。預設提供的生成器是 UlidGenerator

此外,您可以控制是否覆蓋現有已存在的 requestid。還能設置 header_name 等參數,詳情請查閱文件

範例程式碼

main.rs
Cargo.toml
request-id/src/main.rs
use salvo::prelude::*;

#[handler]
async fn hello(req: &mut Request) -> String {
    format!("Request id: {:?}", req.header::<String>("x-request-id"))
}

#[tokio::main]
async fn main() {
    tracing_subscriber::fmt().init();

    let acceptor = TcpListener::new("0.0.0.0:5800").bind().await;
    let router = Router::new().hoop(RequestId::new()).get(hello);
    Server::new(acceptor).serve(router).await;
}