Rust中tracing+OpenTelemetry动态添加上下文属性的实现疑问

方案一：使用tracing::Span动态记录字段 #

tracing的Span支持通过record方法动态添加字段，无需在创建span时预先定义，步骤如下：

• 创建基础span（可不带动态字段）
• 调用span.record("key", &value)添加动态键值对
• 进入span上下文，后续trace事件会自动关联这些字段

示例代码：

// 创建一个span
let mut span = tracing::info_span!("my_operation");
// 动态添加字段
span.record("user_id", &12345);
span.record("request_id", &"abc-123");
// 进入span上下文，后续trace会继承这些字段
let _enter = span.enter();

// 后续日志/事件会带上动态字段
tracing::info!("Processing request");

如果需要在已进入的span上下文里追加字段，可直接获取当前span操作：

// 在任意已绑定span的上下文中
tracing::Span::current().record("status", &"success");

方案二：自定义tracing扩展实现MDC容器 #

如果需要更接近Java MDC的上下文绑定式键值存储，可通过tracing的Extension trait实现自定义数据容器，将其附加到tracing的Context中，订阅者可直接从Context读取数据。

1. 定义MDC扩展结构 #

use tracing::span::Extensions;
use std::collections::HashMap;

#[derive(Debug, Clone, Default)]
pub struct Mdc(HashMap<String, String>);

impl Mdc {
    pub fn insert(&mut self, key: String, value: String) {
        self.0.insert(key, value);
    }

    pub fn get(&self, key: &str) -> Option<&String> {
        self.0.get(key)
    }

    pub fn remove(&mut self, key: &str) {
        self.0.remove(key);
    }
}

2. 向当前tracing Context注入MDC数据 #

use tracing::Context;

// 获取当前上下文并插入MDC
let mut cx = Context::current();
cx.extensions_mut().insert(Mdc::default());
if let Some(mdc) = cx.extensions_mut().get_mut::<Mdc>() {
    mdc.insert("user_id".into(), "12345".into());
    mdc.insert("request_id".into(), "abc-123".into());
}

3. 在自定义订阅者层中读取MDC #

在你的create_custom_layer()实现里，处理事件或span时可从Context提取MDC：

use tracing_subscriber::Layer;
use tracing::Event;

struct CustomLayer;

impl<S> Layer<S> for CustomLayer
where
    S: tracing::Subscriber,
{
    fn on_event(&self, event: &Event<'_>, cx: Context<'_, S>) {
        // 从上下文获取MDC并读取字段
        if let Some(mdc) = cx.extensions().get::<Mdc>() {
            if let Some(user_id) = mdc.get("user_id") {
                println!("Event associated user_id: {}", user_id);
            }
        }
        // 其他自定义处理逻辑...
    }
}

fn create_custom_layer() -> CustomLayer {
    CustomLayer
}

解决OpenTelemetry Context属性返回None的问题 #

你之前遇到的属性读取为空问题，通常是因为上下文传递或属性设置时机错误，可通过以下方式修复：

• 设置属性时，确保当前上下文已绑定OpenTelemetry的SpanContext
• 在订阅者中通过tracing_opentelemetry提供的扩展提取OpenTelemetry Context：

use opentelemetry::Context as OtelContext;
use tracing_opentelemetry::OpenTelemetrySpanExt;

// 在订阅者层提取Otel Context
let otel_cx = cx.extensions().get::<OtelContext>().cloned().unwrap_or_default();
if let Some(value) = otel_cx.get_value("custom_key") {
    // 使用属性值
}

也可直接通过tracing::Span操作OpenTelemetry属性：

use tracing_opentelemetry::OpenTelemetrySpanExt;

// 在已进入的span中添加Otel属性
tracing::Span::current().set_attribute("custom_key", "custom_value");

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