Axum中间件中实现单请求tracing::Span的问题排查

你的问题出在异步场景下误用了同步的Span守卫对象：span.enter()返回的_guard是线程绑定的同步上下文，当代码执行到await时，异步任务可能被调度到其他线程，此时_guard的线程绑定失效。后续的debug!("finished processing")会尝试使用已经脱离线程的span上下文，导致日志无关联；而线程复用又会让下一个请求意外继承上一个请求的span，出现双span的混乱情况。

修正方案 #

移除同步的enter()守卫，改用instrument包裹整个请求处理的异步逻辑，确保span上下文正确跟随异步任务流转：

use axum::{http::{Request, Response}, middleware::Next};
use std::time::Instant;
use uuid::Uuid;
use tracing::{debug, debug_span, field, HeaderValue};

async fn middleware_fn(req: Request, next: Next) -> Response {
    let start = Instant::now();

    let method = req.method();
    let uri = req.uri();
    let req_id = Uuid::now_v7();

    let span = debug_span!("request", %method, %uri, req_id = ?req_id, user = field::Empty);

    // 用instrument包裹整个请求处理流程，替代同步的enter守卫
    async move {
        debug!("started processing request");

        // 无需再给next.run单独instrument，外层span已覆盖整个上下文
        let mut resp = next.run(req).await;

        resp.headers_mut().append(
            "x-request-id",
            HeaderValue::from_str(&req_id.to_string()).unwrap(),
        );

        debug!(latency = ?start.elapsed().as_millis(), status = ?resp.status(), "finished processing");

        resp
    }.instrument(span).await
}

关键修正点 #

1. 移除同步守卫：span.enter()仅适用于同步代码，异步场景下会因线程调度导致上下文丢失。
2. 统一异步上下文：用instrument包裹整个请求处理的async块，确保从请求开始到响应返回的所有异步操作都绑定到同一个span，无论任务被调度到哪个线程，上下文都会正确传递。
3. 避免重复instrument：不再给next.run(req)单独添加instrument(span.clone())，外层的instrument已经覆盖了后续所有异步逻辑，重复操作反而会导致上下文混乱。

内容的提问来源于stack exchange，提问作者Ahmet Yazıcı