在tower-http/hyper/axum中如何将变量绑定到请求并透传到外部任务队列

解决方案 #

下面是3种可落地的实现方案，你可以根据自己的项目场景选择：

方案1：生成ID时存入请求扩展（最推荐，实现成本最低） #

如果你的tower-http版本在v0.4.0及以上，MakeSpan的make_span方法支持传入可变的请求对象，你可以在生成请求ID的同时把值存入请求扩展，后续整个请求处理链路都可以直接从请求对象中取出ID，不需要读取span字段。
修改后的实现代码：

#[derive(Clone)]
pub struct RequestSpan;

impl<B> tower_http::trace::MakeSpan<B> for RequestSpan {
    fn make_span(&mut self, request: &mut http::Request<B>) -> tracing::Span {
        let request_id = ulid::Ulid::new().to_string();
        // 把请求ID存入请求扩展，后续处理逻辑直接取
        request.extensions_mut().insert(request_id.clone());
        tracing::error_span!(
            "rq",
            id = %request_id,
            method = %request.method(),
            uri = %request.uri(),
            version = ?request.version(),
        )
    }
}

在需要提交任务的位置直接从请求扩展中读取ID：

let request_id = req.extensions().get::<String>().unwrap();
// 把request_id作为元数据一起提交到任务队列即可

方案2：从当前活跃Span提取ID #

如果你用的是旧版本tower-http，无法修改请求扩展，可以直接从当前活跃的tracing span中读取ID字段：

use tracing::Span;
use tracing_subscriber::registry::LookupSpan;

fn get_current_request_id() -> Option<String> {
    let span = Span::current();
    let span_id = span.id()?;
    tracing::dispatcher::get_default(|dispatcher| {
        dispatcher
            .downcast_ref::<tracing_subscriber::Registry>()
            .and_then(|registry| registry.span(&span_id))
            .and_then(|span_data| span_data.get("id").map(|val| val.to_string()))
    })
}

方案3：跨进程消费端链路对齐 #

如果你的任务队列消费者是独立进程/服务，需要保证全链路ID统一，做以下处理：

• 提交任务时，把请求ID序列化存入任务的元数据或者payload字段
• 消费端处理任务时，主动创建携带该ID的span：

// 消费端任务处理逻辑
fn process_task(task: Task) {
    // 用任务携带的请求ID创建span，后续日志都会带上同一个ID
    let _span_guard = tracing::error_span!("task_process", id = %task.request_id).entered();
    // 任务处理逻辑
}

内容的提问来源于stack exchange，提问作者imbolc