Merge pull request #89 from seyyed-dev/main

Add `on_record` to `support tracing::field::Empty` and `span.record`

Author: oli-obk

examples/deferred.rs

@@ -1,3 +1,4 @@
+use tracing::field::Empty;
 use tracing::{debug, error, info, level_filters::LevelFilter, span, trace, warn, Level};
 use tracing_subscriber::{layer::SubscriberExt, registry::Registry, Layer};
 use tracing_tree::HierarchicalLayer;
@@ -28,6 +29,13 @@ fn main() {
 
     println!("-> This prints before the span open message");
 
+    let lazy = span!(Level::DEBUG, "lazy span", work_units = Empty);
+    lazy.record("work_units", &3);
+    lazy.in_scope(|| {
+        info!("doing some work");
+    });
+    drop(lazy);
+
     let _e2 = server_span.enter();
     info!("starting");
     std::thread::sleep(std::time::Duration::from_millis(1000));

examples/deferred.stdout

@@ -1,6 +1,11 @@
 -> This prints before the span open message
 1:main┐open: deferred::hierarchical-example version=0.1
-1:main└─┐open: deferred::server host="localhost", port=8080
+1:main└─┐open: deferred::lazy span work_units=3
+1:main  ├─ INFO deferred doing some work
+1:main ┌┘close(v): deferred::lazy span work_units=3
+1:main┌┘post_close: deferred::hierarchical-example version=0.1
+1:main└┐pre_open: deferred::hierarchical-example version=0.1
+1:main └┐open(v): deferred::server host="localhost", port=8080
 1:main  ├─ INFO deferred starting
 1:main  ├─ INFO deferred listening
 -> Deferring two levels of spans

src/lib.rs

@@ -601,6 +601,18 @@ where
             }
         }
     }
+
+    fn on_record(&self, id: &Id, values: &tracing_core::span::Record<'_>, ctx: Context<S>) {
+        let Some(_guard) = Self::is_recursive() else {
+            return;
+        };
+
+        if let Some(span) = ctx.span(id) {
+            if let Some(data) = span.extensions_mut().get_mut::<Data>() {
+                values.record(data);
+            }
+        }
+    }
 }
 
 fn scope_path<'a, R: LookupSpan<'a>>(span: &SpanRef<'a, R>) -> ScopeFromRoot<'a, R> {