liyunjia/chuanyue-service
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

暂时关闭request id 校验

Browse Source
This commit is contained in:
李运家 2024-10-31 20:14:56 +08:00
parent d511241fcf
commit a18316c680
1 changed files with 12 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
server/src/lib.rs
View File

@ -31,17 +31,17 @@ pub async fn serve() {
fn init() -> Router { fn init() -> Router {
let auth: Router = library::typed_router::get_router(); let auth: Router = library::typed_router::get_router();
let trace_layer = TraceLayer::new_for_http().make_span_with(|request: &Request<Body>| { // let trace_layer = TraceLayer::new_for_http().make_span_with(|request: &Request<Body>| {
let req_id = match request // let req_id = match request
.headers() // .headers()
.get("x-request-id") // .get("x-request-id")
.and_then(|value| value.to_str().ok()) // .and_then(|value| value.to_str().ok())
{ // {
Some(v) if v.chars().all(|c| c.is_alphanumeric()) => v.to_string(),// 确保请求ID只包含字母和数字 // Some(v) if v.chars().all(|c| c.is_alphanumeric()) => v.to_string(),// 确保请求ID只包含字母和数字
_ => String::from("unknown"), // _ => panic!("无效的请求ID"),
}; // };
tracing::error_span!("request_id", id = req_id) // tracing::error_span!("request_id", id = req_id)
}); // });
// 配置路由 // 配置路由
// layer之间存在顺序依赖，勿改。layer执行顺序和配置顺序一致 // layer之间存在顺序依赖，勿改。layer执行顺序和配置顺序一致
@ -51,10 +51,10 @@ fn init() -> Router {
.nest(&config!().server.prefix_url, auth) .nest(&config!().server.prefix_url, auth)
.layer( .layer(
ServiceBuilder::new() ServiceBuilder::new()
// .layer(trace_layer)
.layer(axum::middleware::from_fn( .layer(axum::middleware::from_fn(
library::middleware::req_id::handle, library::middleware::req_id::handle,
)) ))
.layer(trace_layer)
.layer(axum::middleware::from_fn( .layer(axum::middleware::from_fn(
library::middleware::cors::handle) library::middleware::cors::handle)
) )