Add instrumentation (Part 1) (#761)

1 files changed, 13 insertions, 0 deletions

diff --git a/library/src/layout/mod.rs b/library/src/layout/mod.rs
index e84e35fc..25014498 100644
--- a/library/src/layout/mod.rs
+++ b/library/src/layout/mod.rs
@@ -68,6 +68,7 @@ pub trait LayoutRoot {
 }
 
 impl LayoutRoot for Content {
+    #[tracing::instrument(name = "Content::layout_root", skip_all)]
     fn layout_root(&self, vt: &mut Vt, styles: StyleChain) -> SourceResult<Document> {
         #[comemo::memoize]
         fn cached(
@@ -87,6 +88,8 @@ impl LayoutRoot for Content {
                 .layout_root(&mut vt, styles)
         }
 
+        tracing::info!("Starting layout");
+
         cached(
             self,
             vt.world,
@@ -126,6 +129,7 @@ pub trait Layout {
 }
 
 impl Layout for Content {
+    #[tracing::instrument(name = "Content::layout", skip_all)]
     fn layout(
         &self,
         vt: &mut Vt,
@@ -151,6 +155,8 @@ impl Layout for Content {
                 .layout(&mut vt, styles, regions)
         }
 
+        tracing::info!("Layouting `Content`");
+
         cached(
             self,
             vt.world,
@@ -164,6 +170,7 @@ impl Layout for Content {
 }
 
 /// Realize into an element that is capable of root-level layout.
+#[tracing::instrument(skip_all)]
 fn realize_root<'a>(
     vt: &mut Vt,
     scratch: &'a Scratch<'a>,
@@ -245,6 +252,7 @@ impl<'a, 'v, 't> Builder<'a, 'v, 't> {
         }
     }
 
+    #[tracing::instrument(skip_all)]
     fn accept(
         &mut self,
         mut content: &'a Content,
@@ -310,6 +318,7 @@ impl<'a, 'v, 't> Builder<'a, 'v, 't> {
         }
     }
 
+    #[tracing::instrument(skip_all)]
     fn styled(
         &mut self,
         elem: &'a Content,
@@ -324,6 +333,7 @@ impl<'a, 'v, 't> Builder<'a, 'v, 't> {
         Ok(())
     }
 
+    #[tracing::instrument(skip(self, local, outer))]
     fn interrupt_style(
         &mut self,
         local: &Styles,
@@ -358,6 +368,7 @@ impl<'a, 'v, 't> Builder<'a, 'v, 't> {
         Ok(())
     }
 
+    #[tracing::instrument(skip(self))]
     fn interrupt_list(&mut self) -> SourceResult<()> {
         if !self.list.items.is_empty() {
             let staged = mem::take(&mut self.list.staged);
@@ -371,6 +382,7 @@ impl<'a, 'v, 't> Builder<'a, 'v, 't> {
         Ok(())
     }
 
+    #[tracing::instrument(skip(self))]
     fn interrupt_par(&mut self) -> SourceResult<()> {
         self.interrupt_list()?;
         if !self.par.0.is_empty() {
@@ -382,6 +394,7 @@ impl<'a, 'v, 't> Builder<'a, 'v, 't> {
         Ok(())
     }
 
+    #[tracing::instrument(skip_all)]
     fn interrupt_page(&mut self, styles: Option<StyleChain<'a>>) -> SourceResult<()> {
         self.interrupt_par()?;
         let Some(doc) = &mut self.doc else { return Ok(()) };