我正在实现一个tokio/axum HTTP服务器。在运行服务器的功能中,我配置路由、添加共享应用程序服务和添加跟踪层。
我的跟踪配置如下:
let tracing_layer = TraceLayer::new_for_http()
.make_span_with(|_request: &Request<Body>| {
let request_id = Uuid::new_v4().to_string();
tracing::info_span!("http-request", %request_id)
})
.on_request(|request: &Request<Body>, _span: &Span| {
tracing::info!("request: {} {}", request.method(), request.uri().path())
})
.on_response(
|response: &Response<BoxBody>, latency: Duration, _span: &Span| {
tracing::info!("response: {} {:?}", response.status(), latency)
},
)
.on_failure(
|error: ServerErrorsFailureClass, _latency: Duration, _span: &Span| {
tracing::error!("error: {}", error)
},
);
let app = Router::new()
// routes
.layer(tracing_layer)
// other layers
...
为了稍微组织一下代码,我将跟踪层配置移到了一个单独的函数中。诀窍是为这个函数提供一个编译返回类型。
第一种方法是按原样移动代码,并让IDE生成返回类型:
TraceLayer<SharedClassifier<ServerErrorsAsFailures>, fn(&Request<Body>) -> Span, fn(&Request<Body>, &Span), fn(&Response<BoxBody>, Duration, &Span), DefaultOnBodyChunk, DefaultOnEos, fn(ServerErrorsFailureClass, Duration, &Span)>
这是完全不可读的,但最糟糕的是它没有编译:;预期fn指针,找到闭包";
在第二种方法中,我将fn
更改为impl Fn
,这意味着闭包类型。再次,我得到一个错误,我的闭包不是Clone
。
第三,我尝试将闭包提取到单独的函数中。但后来我得到了";预期fn指针,找到fn项";。
我能做些什么1(使它可编译,2(使它更可读?
根据经验,由于所有的泛型,像这样分解代码非常困难。相反,我建议使用接受并返回axum::Routers
的函数。这样你就可以绕过所有的泛型:
fn add_middleware(router: Router) -> Router {
router.layer(
TraceLayer::new_for_http().make_span_with(...)
)
}