Merge branch 'potel-base' into antonpirker/potel/aws

Author: antonpirker

sentry_sdk/integrations/arq.py

@@ -1,7 +1,5 @@
 import sys
 
-from opentelemetry.trace.status import StatusCode
-
 import sentry_sdk
 from sentry_sdk.consts import OP, SPANSTATUS
 from sentry_sdk.integrations import DidNotEnable, Integration

sentry_sdk/integrations/opentelemetry/span_processor.py

@@ -43,6 +43,9 @@
     from sentry_sdk._types import Event
 
 
+DEFAULT_MAX_SPANS = 1000
+
+
 class SentrySpanProcessor(SpanProcessor):
     """
     Converts OTel spans into Sentry spans so they can be sent to the Sentry backend.
@@ -79,7 +82,7 @@ def on_end(self, span):
             # if have a root span ending, we build a transaction and send it
             self._flush_root_span(span)
         else:
-            self._children_spans[span.parent.span_id].append(span)
+            self._append_child_span(span)
 
     # TODO-neel-potel not sure we need a clear like JS
     def shutdown(self):
@@ -150,6 +153,20 @@ def _flush_root_span(self, span):
 
         sentry_sdk.capture_event(transaction_event)
 
+    def _append_child_span(self, span):
+        # type: (ReadableSpan) -> None
+        if not span.parent:
+            return
+
+        max_spans = (
+            sentry_sdk.get_client().options["_experiments"].get("max_spans")
+            or DEFAULT_MAX_SPANS
+        )
+
+        children_spans = self._children_spans[span.parent.span_id]
+        if len(children_spans) < max_spans:
+            children_spans.append(span)
+
     def _collect_children(self, span):
         # type: (ReadableSpan) -> List[ReadableSpan]
         if not span.context:

tests/tracing/test_misc.py

@@ -8,6 +8,25 @@
 from sentry_sdk.utils import Dsn
 
 
+def test_span_trimming(sentry_init, capture_events):
+    sentry_init(traces_sample_rate=1.0, _experiments={"max_spans": 3})
+    events = capture_events()
+
+    with start_span(name="hi"):
+        for i in range(10):
+            with start_span(op="foo{}".format(i)):
+                pass
+
+    (event,) = events
+
+    assert len(event["spans"]) == 3
+
+    span1, span2, span3 = event["spans"]
+    assert span1["op"] == "foo0"
+    assert span2["op"] == "foo1"
+    assert span3["op"] == "foo2"
+
+
 def test_transaction_naming(sentry_init, capture_events):
     sentry_init(traces_sample_rate=1.0)
     events = capture_events()