feat(nodestore) FilestoreNodeStorage, all credits to getsentry#76250

Author: rophy

src/sentry/nodestore/filestore/__init__.py

@@ -0,0 +1 @@
+from .backend import FilestoreNodeStorage  # NOQA

src/sentry/nodestore/filestore/backend.py

@@ -0,0 +1,63 @@
+import datetime
+import logging
+from io import BytesIO
+
+from sentry import options as options_store
+from sentry.models.file import get_storage
+from sentry.nodestore.base import NodeStorage
+
+logger = logging.getLogger("sentry.nodestore")
+
+
+class FilestoreNodeStorage(NodeStorage):
+    """
+    A backend that persist nodes to configured File Store.
+    Intended for "s3" or "gcs", which performnace may not be ideal.
+    config.yml Configuration reference:
+    >>> filestore.backend: "s3"
+    ... filestore.options:
+    ...   access_key: "xxx"
+    ...   secret_key: "xxx"
+    ...   endpoint_url: "https://s3.us-east-1.amazonaws.com"
+    ...   bucket_name: "sentry"
+    ...   location: "/sentry"
+    """
+
+    def __init__(self, prefix_path=None):
+        self.prefix_path: str = "nodestore/"
+        backend = options_store.get("filestore.backend")
+        if backend not in ["s3", "gcs"]:
+            logger.warning(
+                "FilestoreNodeStorage was intended for s3 or gcs, currently using: %s", backend
+            )
+        if prefix_path:
+            self.prefix_path = prefix_path
+
+    def _get_bytes(self, id: str):
+        storage = get_storage()
+        path = self.node_path(id)
+        return storage.open(path).read()
+
+    def _set_bytes(self, id: str, data: bytes, ttl=0):
+        storage = get_storage()
+        path = self.node_path(id)
+        storage.save(path, BytesIO(data))
+
+    def delete(self, id):
+        storage = get_storage()
+        path = self.node_path(id)
+        storage.delete(path)
+
+    def cleanup(self, cutoff: datetime.datetime):
+        """
+        This driver does not have managed TTLs.  To enable TTLs you will need to enable it on your
+        bucket.
+        """
+        raise NotImplementedError
+
+    def bootstrap(self):
+        # Nothing for this backend to do during bootstrap
+        pass
+
+    def node_path(self, id: str):
+        return f"{self.prefix_path}{id}.json"