Add cross organisation methods for the string indexer

Author: RaduW

src/sentry/sentry_metrics/indexer/base.py

@@ -265,3 +265,19 @@ def reverse_resolve(self, use_case_id: UseCaseKey, org_id: int, id: int) -> Opti
         Returns None if the entry cannot be found.
         """
         raise NotImplementedError()
+
+    def resolve_shared_org(self, string: str) -> Optional[int]:
+        """
+        Look up the index for a shared (cross organisation) string.
+
+        Typically, this function will only lookup strings that are statically defined but
+        regardless of the mechanism these are strings that are not organisation or use-case specific.
+        """
+        raise NotImplementedError()
+
+    def reverse_shared_org_resolve(self, id: int) -> Optional[str]:
+        """Lookup the stored string given integer for a shared (cross organisation) ID.
+
+        Returns None if the entry cannot be found.
+        """
+        raise NotImplementedError()

src/sentry/sentry_metrics/indexer/cache.py

@@ -167,3 +167,13 @@ def resolve(self, use_case_id: UseCaseKey, org_id: int, string: str) -> Optional
 
     def reverse_resolve(self, use_case_id: UseCaseKey, org_id: int, id: int) -> Optional[str]:
         return self.indexer.reverse_resolve(use_case_id, org_id, id)
+
+    def resolve_shared_org(self, string: str) -> Optional[int]:
+        raise NotImplementedError(
+            "This class should not be used directly, use a wrapping class that derives from StaticStringIndexer"
+        )
+
+    def reverse_shared_org_resolve(self, id: int) -> Optional[str]:
+        raise NotImplementedError(
+            "This class should not be used directly, use a wrapping class that derives from StaticStringIndexer"
+        )

src/sentry/sentry_metrics/indexer/cloudspanner/cloudspanner.py

@@ -431,6 +431,16 @@ def reverse_resolve(self, use_case_id: UseCaseKey, org_id: int, id: int) -> Opti
         else:
             return str(results_list[0][0])
 
+    def resolve_shared_org(self, string: str) -> Optional[int]:
+        raise NotImplementedError(
+            "This class should not be used directly, use the wrapping class CloudSpannerIndexer"
+        )
+
+    def reverse_shared_org_resolve(self, id: int) -> Optional[str]:
+        raise NotImplementedError(
+            "This class should not be used directly, use the wrapping class CloudSpannerIndexer"
+        )
+
 
 class CloudSpannerIndexer(StaticStringIndexer):
     def __init__(self, **kwargs: Any) -> None:

src/sentry/sentry_metrics/indexer/mock.py

@@ -67,6 +67,16 @@ def _record(self, org_id: int, string: str) -> Optional[int]:
             self._reverse[index] = string
         return index
 
+    def resolve_shared_org(self, string: str) -> Optional[int]:
+        raise NotImplementedError(
+            "This class should not be used directly, use the wrapping class SimpleIndexer"
+        )
+
+    def reverse_shared_org_resolve(self, id: int) -> Optional[str]:
+        raise NotImplementedError(
+            "This class should not be used directly, use the wrapping class SimpleIndexer"
+        )
+
 
 class SimpleIndexer(StaticStringIndexer):
     def __init__(self) -> None:

src/sentry/sentry_metrics/indexer/postgres/postgres_v2.py

@@ -198,6 +198,16 @@ def reverse_resolve(self, use_case_id: UseCaseKey, org_id: int, id: int) -> Opti
     def _table(self, use_case_id: UseCaseKey) -> IndexerTable:
         return TABLE_MAPPING[use_case_id]
 
+    def resolve_shared_org(self, string: str) -> Optional[int]:
+        raise NotImplementedError(
+            "This class should not be used directly, use the wrapping class PostgresIndexer"
+        )
+
+    def reverse_shared_org_resolve(self, id: int) -> Optional[str]:
+        raise NotImplementedError(
+            "This class should not be used directly, use the wrapping class PostgresIndexer"
+        )
+
 
 class PostgresIndexer(StaticStringIndexer):
     def __init__(self) -> None:

src/sentry/sentry_metrics/indexer/strings.py

@@ -175,3 +175,13 @@ def reverse_resolve(self, use_case_id: UseCaseKey, org_id: int, id: int) -> Opti
         if id in REVERSE_SHARED_STRINGS:
             return REVERSE_SHARED_STRINGS[id]
         return self.indexer.reverse_resolve(use_case_id=use_case_id, org_id=org_id, id=id)
+
+    def resolve_shared_org(self, string: str) -> Optional[int]:
+        if string in SHARED_STRINGS:
+            return SHARED_STRINGS[string]
+        return None
+
+    def reverse_shared_org_resolve(self, id: int) -> Optional[str]:
+        if id in REVERSE_SHARED_STRINGS:
+            return REVERSE_SHARED_STRINGS[id]
+        return None