From bedb9b301c9b2aaccb8b15e3c7889b13e0fa3964 Mon Sep 17 00:00:00 2001
From: Lyra <teromene@teromene.fr>
Date: Thu, 24 Mar 2022 07:12:49 +0100
Subject: [PATCH] Modify values when a record event is found (#8)

---
 src/lib.rs | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/src/lib.rs b/src/lib.rs
index 20e4151..76c6c57 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -486,6 +486,23 @@ where
         self.enter_span(ctx.span(id).expect("Span not found."), ts);
     }
 
+    fn on_record(&self, id: &span::Id, values: &span::Record<'_>, ctx: Context<'_, S>) {
+        if self.include_args {
+            let span = ctx.span(id).unwrap();
+            let mut exts = span
+                .extensions_mut();
+
+            let args = exts
+                .get_mut::<ArgsWrapper>();
+
+            if let Some(args) = args {
+                let args = Arc::make_mut(&mut args.args);
+                values.record(&mut JsonVisitor { object: args });
+            }
+
+        }
+    }
+
     fn on_event(&self, event: &Event<'_>, _ctx: Context<'_, S>) {
         let ts = self.get_ts();
         let callsite = self.get_callsite(EventOrSpan::Event(event));