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

build: add response time header

Browse Source
This commit is contained in:
kdletters
2026-04-21 01:33:13 +08:00
parent c5db90f37a
commit 6ada003c78
3 changed files with 14 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
server-rs/apps/api-server/src/response_headers.rs
View File

@@ -7,14 +7,16 @@ use axum::{
use crate::{
api_response::API_VERSION,
request_context::{resolve_request_id, X_REQUEST_ID_HEADER},
request_context::{resolve_request_id, RequestContext, X_REQUEST_ID_HEADER},
};
pub const API_VERSION_HEADER: &str = "x-api-version";
pub const RESPONSE_TIME_HEADER: &str = "x-response-time-ms";
pub const ROUTE_VERSION_HEADER: &str = "x-route-version";
pub async fn propagate_request_id_header(request: Request, next: Next) -> Response {
let request_id = resolve_request_id(&request);
let request_context = request.extensions().get::<RequestContext>().cloned();
let mut response = next.run(request).await;
if let Some(request_id) = request_id {
@@ -37,5 +39,13 @@ pub async fn propagate_request_id_header(request: Request, next: Next) -> Respon
.insert(HeaderName::from_static(ROUTE_VERSION_HEADER), header_value);
}
if let Some(request_context) = request_context {
if let Ok(header_value) = HeaderValue::from_str(&request_context.elapsed().to_string()) {
response
.headers_mut()
.insert(HeaderName::from_static(RESPONSE_TIME_HEADER), header_value);
}
}
response
}