perf(snuba): add orjson experiment (#68938)

Author: anonrig

src/sentry/options/defaults.py

@@ -623,6 +623,8 @@
     flags=FLAG_PRIORITIZE_DISK | FLAG_AUTOMATOR_MODIFIABLE,
 )
 
+register("snuba.snql.enable-orjson", default=0.0, flags=FLAG_AUTOMATOR_MODIFIABLE)
+
 # Kafka Publisher
 register("kafka-publisher.raw-event-sample-rate", default=0.0, flags=FLAG_AUTOMATOR_MODIFIABLE)
 register("kafka-publisher.max-event-size", default=100000, flags=FLAG_AUTOMATOR_MODIFIABLE)

src/sentry/snuba/tasks.py

@@ -4,10 +4,12 @@
 from datetime import timedelta
 from typing import TYPE_CHECKING
 
+import orjson
 import sentry_sdk
 from django.utils import timezone
 
 from sentry import features
+from sentry.features.rollout import in_random_rollout
 from sentry.models.environment import Environment
 from sentry.search.events.types import ParamsType
 from sentry.snuba.dataset import Dataset, EntityKey
@@ -250,14 +252,23 @@ def _create_snql_in_snuba(subscription, snuba_query, snql_query, entity_subscrip
         )
 
     entity_key = get_entity_key_from_request(snql_query)
+
+    if in_random_rollout("snuba.snql.enable-orjson"):
+        post_body: str | bytes = orjson.dumps(body)
+    else:
+        post_body = json.dumps(body)
     response = _snuba_pool.urlopen(
         "POST",
         f"/{snuba_query.dataset}/{entity_key.value}/subscriptions",
-        body=json.dumps(body),
+        body=post_body,
     )
     if response.status != 202:
         metrics.incr("snuba.snql.subscription.http.error", tags={"dataset": snuba_query.dataset})
         raise SnubaError("HTTP %s response from Snuba!" % response.status)
+
+    if in_random_rollout("snuba.snql.enable-orjson"):
+        with sentry_sdk.start_span(op="sentry.utils.json.loads"):
+            return orjson.loads(response.data)["subscription_id"]
     return json.loads(response.data)["subscription_id"]