From fc8816c367c493585765f986f1c8c55f92e11dea Mon Sep 17 00:00:00 2001
From: Tobias Koppers <tobias.koppers@googlemail.com>
Date: Wed, 9 Oct 2024 15:10:05 +0200
Subject: [PATCH] instrument evaluation

---
 .../crates/turbopack-node/src/evaluate.rs     | 21 ++++++++++++-------
 1 file changed, 14 insertions(+), 7 deletions(-)

diff --git a/turbopack/crates/turbopack-node/src/evaluate.rs b/turbopack/crates/turbopack-node/src/evaluate.rs
index 44fe5220e3abcc..56d461599b3365 100644
--- a/turbopack/crates/turbopack-node/src/evaluate.rs
+++ b/turbopack/crates/turbopack-node/src/evaluate.rs
@@ -12,6 +12,7 @@ use indexmap::indexmap;
 use parking_lot::Mutex;
 use serde::{de::DeserializeOwned, Deserialize, Serialize};
 use serde_json::Value as JsonValue;
+use tracing::{trace_span, Instrument};
 use turbo_tasks::{
     duration_span, mark_finished, prevent_gc, util::SharedError, Completion, RawVc, TaskInput,
     TryJoinIterExt, Value, Vc,
@@ -308,6 +309,7 @@ pub fn evaluate(
     })
 }
 
+#[tracing::instrument(skip_all)]
 pub async fn compute(
     evaluate_context: impl EvaluateContext,
     sender: Vc<JavaScriptStreamSender>,
@@ -324,9 +326,9 @@ pub async fn compute(
 
         // Read this strongly consistent, since we don't want to run inconsistent
         // node.js code.
-        let pool = pool.strongly_consistent().await?;
+        let pool = pool.strongly_consistent().instrument(tracing::trace_span!("pool")).await?;
 
-        let args = evaluate_context.args().iter().try_join().await?;
+        let args = evaluate_context.args().iter().try_join().instrument(tracing::trace_span!("args")).await?;
         // Assume this is a one-off operation, so we can kill the process
         // TODO use a better way to decide that.
         let kill = !evaluate_context.keep_alive();
@@ -348,6 +350,7 @@ pub async fn compute(
             },
             PoolErrorHandler,
         )
+        .instrument(trace_span!("evaluate"))
         .await
         .map_err(|(e, _)| e)?;
 
@@ -355,7 +358,7 @@ pub async fn compute(
         // need to spawn a new thread to continually pull data out of the process,
         // and ferry that along.
         loop {
-            let output = pull_operation(&mut operation, &pool, &evaluate_context, &mut state).await?;
+            let output = pull_operation(&mut operation, &pool, &evaluate_context, &mut state).instrument(trace_span!("pull")).await?;
 
             match output {
                 LoopResult::Continue(data) => {
@@ -376,11 +379,14 @@ pub async fn compute(
             }
         }
 
-        evaluate_context.finish(state, &pool).await?;
+        async move {
+            evaluate_context.finish(state, &pool).await?;
 
-        if kill {
-            operation.wait_or_kill().await?;
-        }
+            if kill {
+                operation.wait_or_kill().await?;
+            }
+            anyhow::Ok(())
+        }.instrument(tracing::trace_span!("finish")).await?;
     };
 
     let mut sender = (sender.get)();
@@ -399,6 +405,7 @@ pub async fn compute(
 
 /// Repeatedly pulls from the NodeJsOperation until we receive a
 /// value/error/end.
+#[tracing::instrument(skip_all)]
 async fn pull_operation<T: EvaluateContext>(
     operation: &mut NodeJsOperation,
     pool: &NodeJsPool,