codex/codex-rs/exec-server/src/trace_context.rs

use http::HeaderMap;
use http::HeaderValue;

pub(crate) fn current_rendezvous_headers() -> HeaderMap {
    let mut headers = current_trace_context_headers();
    for (header, variable) in [
        ("x-cluster-name", "OPENAI_CLUSTER"),
        ("x-openai-internal-caller", "DD_SERVICE"),
    ] {
        if let Ok(value) = std::env::var(variable)
            && !value.is_empty()
            && let Ok(value) = HeaderValue::try_from(value)
        {
            headers.insert(header, value);
        }
    }
    headers
}

pub(crate) fn current_trace_context_headers() -> HeaderMap {
    let mut headers = HeaderMap::new();
    let Some(trace) = codex_otel::current_span_w3c_trace_context() else {
        return headers;
    };
    if let Some(traceparent) = trace.traceparent
        && let Ok(value) = HeaderValue::try_from(traceparent)
    {
        headers.insert("traceparent", value);
    }
    if let Some(tracestate) = trace.tracestate
        && let Ok(value) = HeaderValue::try_from(tracestate)
    {
        headers.insert("tracestate", value);
    }
    headers
}

#[cfg(test)]
#[path = "trace_context_tests.rs"]
mod tests;