Tracing trace: toggle the profiling of memory at runtime

tracing-trace/src/layer.rs

@@ -20,24 +20,26 @@ pub struct TraceLayer {
     sender: tokio::sync::mpsc::UnboundedSender<Entry>,
     callsites: RwLock<HashMap<OpaqueIdentifier, ResourceId>>,
     start_time: std::time::Instant,
+    profile_memory: bool,
 }
 
 impl Trace {
-    pub fn new() -> (Self, TraceLayer) {
+    pub fn new(profile_memory: bool) -> (Self, TraceLayer) {
         let (sender, receiver) = tokio::sync::mpsc::unbounded_channel();
         let trace = Trace { receiver };
         let layer = TraceLayer {
             sender,
             callsites: Default::default(),
             start_time: std::time::Instant::now(),
+            profile_memory,
         };
         (trace, layer)
     }
 }
 
 impl<W: Write> TraceWriter<W> {
-    pub fn new(writer: W) -> (Self, TraceLayer) {
+    pub fn new(writer: W, profile_memory: bool) -> (Self, TraceLayer) {
-        let (trace, layer) = Trace::new();
+        let (trace, layer) = Trace::new(profile_memory);
         (trace.into_writer(writer), layer)
     }
 
@@ -97,7 +99,11 @@ impl TraceLayer {
     }
 
     fn memory_stats(&self) -> Option<MemoryStats> {
+        if self.profile_memory {
             MemoryStats::fetch()
+        } else {
+            None
+        }
     }
 
     fn send(&self, entry: Entry) {

tracing-trace/src/main.rs

@@ -74,7 +74,7 @@ fn on_panic(info: &std::panic::PanicInfo) {
 
 fn main() {
     let (mut trace, profiling_layer) =
-        tracing_trace::TraceWriter::new(std::fs::File::create("trace.json").unwrap());
+        tracing_trace::TraceWriter::new(std::fs::File::create("trace.json").unwrap(), true);
 
     let subscriber = tracing_subscriber::registry()
         // any number of other subscriber layers may be added before or