rust/neon
1
0
Fork 0
mirror of https://github.com/neondatabase/neon.git synced 2026-07-08 06:30:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

cr feedback

Browse Source
This commit is contained in:
Shany Pozin
2023-02-27 13:00:13 +02:00
parent 6ce1638df2
commit 84a8089ae7
1 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
libs/utils/src/http/endpoint.rs
View File

@@ -11,7 +11,7 @@ use routerify::ext::RequestExt;
use routerify::RequestInfo;
use routerify::{Middleware, Router, RouterBuilder, RouterService};
use tokio::task::JoinError;
use tracing::{self};
use tracing;
use std::future::Future;
use std::net::TcpListener;
@@ -67,11 +67,15 @@ pub fn add_request_id_middleware<B: hyper::body::HttpBody + Send + Sync + 'stati
) -> Middleware<B, ApiError> {
Middleware::pre(move |mut req| async move {
let headers = req.headers_mut();
let name = HeaderName::from_str("UUID").unwrap();
let name = HeaderName::from_str("UUID").expect("created header name");
let request_id = uuid::Uuid::new_v4().to_string();
let value = HeaderValue::from_str(&request_id).unwrap();
headers.insert(name, value);
tracing::info!("{} {} {}", req.method(), req.uri().path(), request_id);
if req.method() == Method::GET {
tracing::debug!("{} {} {}", req.method(), req.uri().path(), request_id);
} else {
tracing::info!("{} {} {}", req.method(), req.uri().path(), request_id);
}
Ok(req)
})
}