GenarrativeAI/Genarrative
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

style: format api server tracing layer

Browse Source
This commit is contained in:
kdletters
2026-04-22 18:43:09 +08:00
parent 6700e99dc8
commit 7e49c3a3c6
1 changed files with 38 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

68
server-rs/crates/api-server/src/app.rs
View File

@@ -693,39 +693,47 @@ pub fn build_router(state: AppState) -> Router {
) )
}) })
.on_request(DefaultOnRequest::new().level(Level::INFO)) .on_request(DefaultOnRequest::new().level(Level::INFO))
.on_response(move |response: &axum::response::Response, latency: std::time::Duration, span: &Span| { .on_response(
let latency_ms = latency.as_millis().min(u64::MAX as u128) as u64; move |response: &axum::response::Response,
let status = response.status().as_u16(); latency: std::time::Duration,
let slow_request = latency_ms >= slow_request_threshold_ms; span: &Span| {
span.record("status", status); let latency_ms = latency.as_millis().min(u64::MAX as u128) as u64;
span.record("latency_ms", latency_ms); let status = response.status().as_u16();
if slow_request { let slow_request = latency_ms >= slow_request_threshold_ms;
warn!( span.record("status", status);
span.record("latency_ms", latency_ms);
if slow_request {
warn!(
parent: span,
status,
latency_ms,
slow_request = true,
"http request completed slowly"
);
} else {
info!(
parent: span,
status,
latency_ms,
slow_request = false,
"http request completed"
);
}
},
)
.on_failure(
|failure: ServerErrorsFailureClass,
latency: std::time::Duration,
span: &Span| {
let latency_ms = latency.as_millis().min(u64::MAX as u128) as u64;
error!(
parent: span, parent: span,
status,
latency_ms, latency_ms,
slow_request = true, failure = %failure,
"http request completed slowly" "http request failed"
); );
} else { },
info!( ),
parent: span,
status,
latency_ms,
slow_request = false,
"http request completed"
);
}
})
.on_failure(|failure: ServerErrorsFailureClass, latency: std::time::Duration, span: &Span| {
let latency_ms = latency.as_millis().min(u64::MAX as u128) as u64;
error!(
parent: span,
latency_ms,
failure = %failure,
"http request failed"
);
}),
) )
// request_id 中间件先进入请求链，确保后续 tracing、错误处理和响应头层都能复用同一份请求标识。 // request_id 中间件先进入请求链，确保后续 tracing、错误处理和响应头层都能复用同一份请求标识。
.layer(middleware::from_fn(attach_request_context)) .layer(middleware::from_fn(attach_request_context))