Added flag to track replaying awareness of tracing span

src/vm/mod.rs

@@ -76,6 +76,11 @@ pub struct CoreVM {
     // State machine
     context: Context,
     last_transition: Result<State, Error>,
+
+    // Flag to indicate whether span replay attribute should be set
+    // When replaying this is set on the sys call
+    // When not replaying this is reset on the sys call that transitioned the state
+    tracing_replaying_flag: bool,
 }
 
 impl CoreVM {
@@ -166,6 +171,7 @@ impl super::VM for CoreVM {
                 options,
             },
             last_transition: Ok(State::WaitingStart),
+            tracing_replaying_flag: true,
         })
     }
 

src/vm/transitions/mod.rs

@@ -64,6 +64,29 @@ impl CoreVM {
                 let was_closed = matches!(s, State::Ended | State::Suspended);
                 match TransitionAndReturn::transition_and_return(s, &mut self.context, event) {
                     Ok((new_state, output)) => {
+                        match new_state {
+                            State::Replaying {
+                                current_await_point,
+                                ..
+                            } => {
+                                // Replay record is not yet set in the span
+                                if self.tracing_replaying_flag && current_await_point.is_some() {
+                                    tracing::Span::current().record("replaying", true);
+                                    self.tracing_replaying_flag = false;
+                                }
+                            }
+                            State::Processing {
+                                current_await_point,
+                                ..
+                            } => {
+                                // Replay record is not yet reset in the span
+                                if !self.tracing_replaying_flag && current_await_point.is_some() {
+                                    tracing::Span::current().record("replaying", false);
+                                    self.tracing_replaying_flag = true;
+                                }
+                            }
+                            _ => {}
+                        }
                         self.last_transition = Ok(new_state);
                         Ok(output)
                     }