ref: Add _is_segment to streaming spans

Author: sentrivana

sentry_sdk/_span_batcher.py

@@ -91,6 +91,7 @@ def _to_transport_format(item: "StreamedSpan") -> "Any":
             "span_id": item.span_id,
             "name": item._name,
             "status": item._status,
+            "is_segment": item._is_segment(),
             "start_timestamp": item._start_timestamp.timestamp(),
         }
 

sentry_sdk/traces.py

@@ -439,6 +439,9 @@ def start_timestamp(self) -> "Optional[datetime]":
     def timestamp(self) -> "Optional[datetime]":
         return self._timestamp
 
+    def _is_segment(self) -> bool:
+        return self._segment == self
+
 
 class NoOpStreamedSpan(StreamedSpan):
     __slots__ = (
@@ -518,6 +521,9 @@ def set_attributes(self, attributes: "Attributes") -> None:
     def remove_attribute(self, key: str) -> None:
         pass
 
+    def _is_segment(self) -> bool:
+        return True if self._scope is not None else False
+
     @property
     def status(self) -> "str":
         return SpanStatus.OK.value