ref(rust): Don't implement separate scope management

sentry_sdk/integrations/rust_tracing.py

@@ -40,8 +40,6 @@
 from sentry_sdk.tracing import Span as SentrySpan
 from sentry_sdk.utils import SENSITIVE_DATA_SUBSTITUTE
 
-TraceState = Optional[Tuple[Optional[SentrySpan], SentrySpan]]
-
 
 class RustTracingLevel(Enum):
     Trace = "TRACE"
@@ -170,7 +168,7 @@
             else self.include_tracing_fields
         )
 
-    def on_event(self, event: str, _span_state: "TraceState") -> None:
+    def on_event(self, event: str, sentry_span: "SentrySpan") -> None:
         deserialized_event = json.loads(event)
         metadata = deserialized_event.get("metadata", {})
 
@@ -184,7 +182,7 @@
         elif event_type == EventTypeMapping.Event:
             process_event(deserialized_event)
 
-    def on_new_span(self, attrs: str, span_id: str) -> "TraceState":
+    def on_new_span(self, attrs: str, span_id: str) -> "SentrySpan":
         attrs = json.loads(attrs)
         metadata = attrs.get("metadata", {})
 
@@ -210,35 +208,26 @@
             "origin": self.origin,
         }
 
-        scope = sentry_sdk.get_current_scope()
-        parent_sentry_span = scope.span
-        if parent_sentry_span:
-            sentry_span = parent_sentry_span.start_child(**kwargs)
-        else:
-            sentry_span = scope.start_span(**kwargs)
-
+        sentry_span = sentry_sdk.start_span(**kwargs)
         fields = metadata.get("fields", [])
         for field in fields:
             if self._include_tracing_fields():
                 sentry_span.set_data(field, attrs.get(field))
             else:
                 sentry_span.set_data(field, SENSITIVE_DATA_SUBSTITUTE)
 
-        scope.span = sentry_span
-        return (parent_sentry_span, sentry_span)
+        sentry_span.__enter__()
+        return sentry_span
 
-    def on_close(self, span_id: str, span_state: "TraceState") -> None:
-        if span_state is None:
+    def on_close(self, span_id: str, sentry_span: "SentrySpan") -> None:
+        if sentry_span is None:
             return
 
-        parent_sentry_span, sentry_span = span_state
-        sentry_span.finish()
-        sentry_sdk.get_current_scope().span = parent_sentry_span
+        sentry_span.__exit__()
 
-    def on_record(self, span_id: str, values: str, span_state: "TraceState") -> None:
-        if span_state is None:
+    def on_record(self, span_id: str, values: str, sentry_span: "SentrySpan") -> None:
+        if sentry_span is None:
             return
-        _parent_sentry_span, sentry_span = span_state
 
         deserialized_values = json.loads(values)
         for key, value in deserialized_values.items():