Add frame name to its flame span

coolreader18 · coolreader18 · commit f585f7ed1037 · 2019-07-08T21:49:18.000-05:00
diff --git a/vm/src/frame.rs b/vm/src/frame.rs
@@ -268,8 +268,15 @@ impl Frame {
         }
     }
 
-    #[cfg_attr(feature = "flame-it", flame("Frame"))]
+    // #[cfg_attr(feature = "flame-it", flame("Frame"))]
     pub fn run(&self, vm: &VirtualMachine) -> Result<ExecutionResult, PyObjectRef> {
+        flame_guard!(format!("Frame::run({})", self.code.obj_name));
+        // flame doesn't include notes in the html graph :(
+        flame_note!(
+            flame::StrCow::from("CodeObj name"),
+            self.code.obj_name.clone().into()
+        );
+
         let filename = &self.code.source_path.to_string();
 
         // This is the name of the object being run:
diff --git a/vm/src/macros.rs b/vm/src/macros.rs
@@ -273,3 +273,14 @@ macro_rules! flame_guard {
         let _guard = ::flame::start_guard($name);
     };
 }
+
+macro_rules! flame_note {
+    ($name:expr, $desc:expr$(,)?) => {
+        #[cfg(feature = "flame-it")]
+        ::flame::note($name, ::std::option::Option::Some($desc));
+    };
+    ($name:expr$(,)?) => {
+        #[cfg(feature = "flame-it")]
+        ::flame::note($name, ::std::option::Option::None);
+    };
+}
