add on_record

Author: seyyed-dev

examples/deferred.stdout

@@ -1,8 +1,8 @@
 -> This prints before the span open message
 1:main┐open: deferred::hierarchical-example version=0.1
-1:main└─┐open: deferred::lazy span 
+1:main└─┐open: deferred::lazy span work_units=3
 1:main  ├─ INFO deferred doing some work
-1:main ┌┘close(v): deferred::lazy span 
+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

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> {