fix(profiling): Update active thread for asgi

Ensure the handling thread is set on the transaction for asgi transactions not
just main thread.

Author: Zylphrex

sentry_sdk/integrations/django/asgi.py

@@ -171,6 +171,8 @@ async def sentry_wrapped_callback(request, *args, **kwargs):
         sentry_scope = sentry_sdk.get_isolation_scope()
         if sentry_scope.profile is not None:
             sentry_scope.profile.update_active_thread_id()
+        if sentry_scope.transaction is not None:
+            sentry_scope.transaction.update_active_thread()
 
         with sentry_sdk.start_span(
             op=OP.VIEW_RENDER,

sentry_sdk/integrations/django/views.py

@@ -81,6 +81,8 @@ def sentry_wrapped_callback(request, *args, **kwargs):
         # this isn't necessary for async views since that runs on main
         if sentry_scope.profile is not None:
             sentry_scope.profile.update_active_thread_id()
+        if sentry_scope.transaction is not None:
+            sentry_scope.transaction.update_active_thread()
 
         with sentry_sdk.start_span(
             op=OP.VIEW_RENDER,

sentry_sdk/integrations/fastapi.py

@@ -91,6 +91,8 @@ def _sentry_call(*args, **kwargs):
                 sentry_scope = sentry_sdk.get_isolation_scope()
                 if sentry_scope.profile is not None:
                     sentry_scope.profile.update_active_thread_id()
+                if sentry_scope.transaction is not None:
+                    sentry_scope.transaction.update_active_thread()
                 return old_call(*args, **kwargs)
 
             dependant.call = _sentry_call

sentry_sdk/integrations/starlette.py

@@ -491,6 +491,8 @@ def _sentry_sync_func(*args, **kwargs):
 
                 if sentry_scope.profile is not None:
                     sentry_scope.profile.update_active_thread_id()
+                if sentry_scope.transaction is not None:
+                    sentry_scope.transaction.update_active_thread()
 
                 request = args[0]
 

sentry_sdk/tracing.py

@@ -329,8 +329,7 @@ def __init__(
         self._span_recorder = None  # type: Optional[_SpanRecorder]
         self._local_aggregator = None  # type: Optional[LocalAggregator]
 
-        thread_id, thread_name = get_current_thread_meta()
-        self.set_thread(thread_id, thread_name)
+        self.update_active_thread()
         self.set_profiler_id(get_profiler_id())
 
     # TODO this should really live on the Transaction class rather than the Span
@@ -732,6 +731,11 @@ def get_profile_context(self):
             "profiler_id": profiler_id,
         }
 
+    def update_active_thread(self):
+        # type: () -> None
+        thread_id, thread_name = get_current_thread_meta()
+        self.set_thread(thread_id, thread_name)
+
 
 class Transaction(Span):
     """The Transaction is the root element that holds all the spans