feat(profiling): Support for multithreaded profiles

sentry_sdk/_types.py

@@ -45,6 +45,7 @@
         "attachment",
         "session",
         "internal",
+        "profile",
     ]
     SessionStatus = Literal["ok", "exited", "crashed", "abnormal"]
     EndpointType = Literal["store", "envelope"]

sentry_sdk/client.py

@@ -403,6 +403,7 @@ def capture_event(
             if is_transaction:
                 if "profile" in event_opt:
                     event_opt["profile"]["transaction_id"] = event_opt["event_id"]
+                    event_opt["profile"]["environment"] = event_opt.get("environment")
                     event_opt["profile"]["version_name"] = event_opt.get("release", "")
                     envelope.add_profile(event_opt.pop("profile"))
                 envelope.add_transaction(event_opt)

sentry_sdk/envelope.py

@@ -252,6 +252,8 @@ def data_category(self):
             return "error"
         elif ty == "client_report":
             return "internal"
+        elif ty == "profile":
+            return "profile"
         else:
             return "default"
 

sentry_sdk/integrations/profiling.py

@@ -0,0 +1,14 @@
+from sentry_sdk.integrations import DidNotEnable, Integration
+from sentry_sdk.profiler import _setup_profiler
+
+
+class ProfilingIntegration(Integration):
+    identifier = "profiling"
+
+    @staticmethod
+    def setup_once():
+        # type: () -> None
+        try:
+            _setup_profiler()
+        except ValueError:
+            raise DidNotEnable("Profiling can only be enabled from the main thread.")

sentry_sdk/integrations/wsgi.py

@@ -11,7 +11,7 @@
 from sentry_sdk.tracing import Transaction, TRANSACTION_SOURCE_ROUTE
 from sentry_sdk.sessions import auto_session_tracking
 from sentry_sdk.integrations._wsgi_common import _filter_headers
-from sentry_sdk.profiler import profiling
+from sentry_sdk.profiler import start_profiling
 
 from sentry_sdk._types import MYPY
 
@@ -131,7 +131,7 @@ def __call__(self, environ, start_response):
 
                     with hub.start_transaction(
                         transaction, custom_sampling_context={"wsgi_environ": environ}
-                    ), profiling(transaction, hub):
+                    ), start_profiling(transaction, hub):
                         try:
                             rv = self.app(
                                 environ,