如何在Rust中用tracing库动态调整特定调用链的日志过滤级别

我正在为Rust应用做埋点，希望为特定的Span树动态设置日志过滤级别——不是全局修改日志级别，而是针对某段代码路径的特定执行实例生效。目前已通过为父Span添加元字段always_log，并将该字段通过Extension传播给子Span，再根据字段或Extension的存在进行条件过滤实现了需求，代码如下：

use tracing::span::Attributes;
use tracing::{span, Level, Subscriber};
use tracing_core::{Interest, Metadata};
use tracing_subscriber::fmt;
use tracing_subscriber::layer::{Context, Layer, SubscriberExt};
use tracing_subscriber::registry::LookupSpan;

struct PropagateExtensionLayer;

struct AlwaysLog;

impl<S> Layer<S> for PropagateExtensionLayer
where
    S: Subscriber + for<'lookup> LookupSpan<'lookup>,
{
    fn on_new_span(
        &self,
        _attrs: &Attributes<'_>,
        id: &tracing_core::span::Id,
        ctx: Context<'_, S>,
    ) {
        let span = ctx.span(id).unwrap();
        let should_propagate_field = span
            .fields()
            .iter()
            .any(|field| field.name() == "always_log")
            || span.parent().map_or(false, |parent| {
                parent.extensions().get::<AlwaysLog>().is_some()
            });

        if should_propagate_field {
            span.extensions_mut().insert(AlwaysLog);
        }
    }

    fn register_callsite(&self, metadata: &'static Metadata<'static>) -> Interest {
        if metadata
            .fields()
            .iter()
            .any(|field| field.name() == "always_log")
        {
            Interest::always()
        } else {
            Interest::sometimes()
        }
    }
}

pub struct FilterLayer;

impl<S> Layer<S> for FilterLayer
where
    S: tracing::Subscriber + for<'lookup> LookupSpan<'lookup>,
{
    fn enabled(&self, metadata: &tracing::Metadata<'_>, ctx: Context<'_, S>) -> bool {
        let always_log = metadata
            .fields()
            .iter()
            .any(|field| field.name() == "always_log")
            || ctx
                .lookup_current()
                .map_or(false, |span| span.extensions().get::<AlwaysLog>().is_some());
        always_log || metadata.level() <= &Level::ERROR
    }
}

fn main() {
    let subscriber = tracing_subscriber::registry()
        .with(FilterLayer)
        .with(PropagateExtensionLayer)
        .with(fmt::Layer::default());

    tracing::subscriber::set_global_default(subscriber).expect("Failed to set subscriber");

    let parent_span = span!(Level::INFO, "parent", always_log = true, key1 = "value1");
    parent_span.in_scope(|| {
        let child_span = span!(Level::INFO, "child", key2 = "value2");
        child_span.in_scope(|| {
            tracing::info!("This should be logged and parent spans activated");
        });
    });

    let parent_span = span!(Level::INFO, "parent", key1 = "value1");
    parent_span.in_scope(|| {
        let child_span = span!(Level::INFO, "child", key2 = "value2");
        child_span.in_scope(|| {
            tracing::info!("This should not be logged");
            tracing::error!("This should also be logged but parent spans deactivated");
        });
    });
}

但我不确定这是否是最优方案，希望解决以下三个问题：

问题1：是否可以直接为span添加扩展，并条件性地强制启用该span？ #

不行。tracing的调用点注册（register_callsite）发生在Span实例创建之前，此时只能基于静态元数据（如Span名称、字段定义）决定Interest，而扩展是Span实例创建后才能添加的动态数据。因此无法在调用点注册阶段就基于扩展来决定是否启用Span。你当前通过always_log字段触发Interest::always()是唯一可行的静态触发方式，后续再通过扩展传播状态是合理的。

问题2：是否有更简单、更优或更规范的方式实现这种条件过滤行为？ #

有几个优化方向和替代思路：

优化现有方案 #

• 简化标记字段：把always_log字段改成标记类型（如always_log = ()），避免传递不必要的值，同时减少字段匹配的开销。
• 合并Layer：将PropagateExtensionLayer和FilterLayer合并成单个Layer，减少订阅者的Layer数量，降低上下文切换的开销。

替代思路 #

• 利用动态过滤函数：使用tracing_subscriber::filter::DynFilterFn，在过滤函数中动态查询当前Span的扩展，但前提是Span已被创建（即Interest足够），因此还是需要通过静态字段触发初始的Interest::always()，逻辑和当前方案一致。
• 跨线程支持：tracing的Span上下文通过线程本地存储（TLS）传播，使用tokio::spawn等支持TLS继承的异步运行时，Span的扩展状态会自动跟随上下文跨线程传播，无需额外处理，你的当前方案已经天然支持这一点。

问题3：为何第一个示例中的子span会被启用？ #

这是因为tracing的Interest传播规则：当父Span被订阅者标记为Interest::always()时，子Span的Interest会被自动提升。

具体来说，第一个示例中父Span带有always_log字段，register_callsite返回Interest::always()，订阅者会始终对该父Span实例感兴趣。当子Span创建时，即使子Span的调用点注册返回Interest::sometimes()，由于父Span处于激活状态且被订阅者关注，子Span的实例也会被自动标记为感兴趣，从而进入on_new_span方法，继承父Span的AlwaysLog扩展，最终被FilterLayer允许日志输出。

而第二个示例中，父Span没有always_log字段，调用点注册返回Interest::sometimes()，且父Span本身不满足过滤条件（级别不是ERROR），因此子Span不会被启用，日志被过滤。

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