.

Author: sentrivana

sentry_sdk/integrations/asgi.py

@@ -216,27 +216,32 @@ async def _run_app(
                     span_ctx: "ContextManager[Union[Span, StreamedSpan, None]]"
                     if span_streaming:
                         segment: "Optional[StreamedSpan]" = None
-                        attributes: "Attributes" = {}
+                        attributes: "Attributes" = {
+                            "sentry.span.source": getattr(
+                                transaction_source, "value", transaction_source
+                            ),
+                            "sentry.origin": self.span_origin,
+                            "asgi.type": ty,
+                        }
                         sentry_scope.set_custom_sampling_context({"asgi_scope": scope})
+
                         if ty in ("http", "websocket"):
                             if (
                                 ty == "websocket"
                                 or method in self.http_methods_to_capture
                             ):
                                 sentry_sdk.traces.continue_trace(_get_headers(scope))
                                 attributes["sentry.op"] = f"{ty}.server"
+                                segment = sentry_sdk.traces.start_span(
+                                    name=transaction_name, attributes=attributes
+                                )
                         else:
                             sentry_sdk.traces.new_trace()
                             attributes["sentry.op"] = OP.HTTP_SERVER
+                            segment = sentry_sdk.traces.start_span(
+                                name=transaction_name, attributes=attributes
+                            )
 
-                        attributes["sentry.span.source"] = getattr(
-                            transaction_source, "value", transaction_source
-                        )
-                        attributes["sentry.origin"] = self.span_origin
-                        attributes["asgi.type"] = ty
-                        segment = sentry_sdk.traces.start_span(
-                            name=transaction_name, attributes=attributes
-                        )
                         span_ctx = segment or nullcontext()
 
                     else: