perf: add tracing to nodestore (#67002)

Author: anonrig

src/sentry/nodestore/base.py

@@ -272,6 +272,7 @@ def _get_cache_items(self, id_list):
             return self.cache.get_many(id_list)
         return {}
 
+    @sentry_sdk.tracing.trace
     def _set_cache_item(self, id, data):
         if self.cache and data:
             self.cache.set(id, data)

src/sentry/nodestore/bigtable/backend.py

@@ -66,6 +66,7 @@ def _get_bytes_multi(self, id_list: list[str]) -> dict[str, bytes | None]:
         rv.update(self.store.get_many(id_list))
         return rv
 
+    @sentry_sdk.tracing.trace
     def _set_bytes(self, id, data, ttl=None):
         self.store.set(id, data, ttl)
 

src/sentry/nodestore/django/backend.py

@@ -4,6 +4,7 @@
 import math
 import pickle
 
+import sentry_sdk
 from django.utils import timezone
 
 from sentry.db.models import create_or_update
@@ -50,6 +51,7 @@ def delete_multi(self, id_list):
         Node.objects.filter(id__in=id_list).delete()
         self._delete_cache_items(id_list)
 
+    @sentry_sdk.tracing.trace
     def _set_bytes(self, id, data, ttl=None):
         create_or_update(Node, id=id, values={"data": compress(data), "timestamp": timezone.now()})
 

src/sentry/nodestore/filesystem/backend.py

@@ -2,6 +2,7 @@
 import os
 from datetime import timezone
 
+import sentry_sdk
 from django.conf import settings
 
 from sentry.nodestore.base import NodeStorage
@@ -27,6 +28,7 @@ def _get_bytes(self, id: str):
         with open(self.node_path(id), "rb") as file:
             return file.read()
 
+    @sentry_sdk.tracing.trace
     def _set_bytes(self, id: str, data: bytes, ttl=0):
         with open(self.node_path(id), "wb") as file:
             file.write(data)