chore: add trace_id to root span

Author: prestwich

src/tasks/env.rs

@@ -10,6 +10,9 @@ use alloy::{
     primitives::{B256, U256},
     providers::{Provider, network::Network},
 };
+use init4_bin_base::deps::{
+    opentelemetry::trace::TraceContextExt, tracing_opentelemetry::OpenTelemetrySpanExt,
+};
 use signet_constants::SignetSystemConstants;
 use signet_sim::{HostEnv, RollupEnv};
 use tokio::{sync::watch, task::JoinHandle};
@@ -279,13 +282,17 @@ impl EnvTask {
             host_block.number = "initial",
             rollup_header.number = "initial",
             rollup_header.hash = "initial",
+            trace_id = tracing::field::Empty,
         );
 
         while let Some(rollup_header) = rollup_headers
             .next()
             .instrument(info_span!(parent: &span, "waiting_for_notification"))
             .await
         {
+            // Ensure that we record the OpenTelemetry trace ID in the span.
+            span.record("trace_id", span.context().span().span_context().trace_id().to_string());
+
             let host_block_number =
                 self.config.constants.rollup_block_to_host_block_num(rollup_header.number);
             let rollup_block_number = rollup_header.number;
@@ -360,6 +367,7 @@ impl EnvTask {
                 host_block.number = host_block_number + 1,
                 rollup_header.number = rollup_block_number + 1,
                 rollup_header.hash = tracing::field::Empty,
+                trace_id = tracing::field::Empty,
             );
         }
     }