fix(hybrid-cloud): Override get_from_cache and get_many_from_cache wi…

…th silo limits
getsentry · Aug 16, 2023 · ad3bf32 · ad3bf32
1 parent d43f543
commit ad3bf32
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/sentry/db/models/manager/base.py b/src/sentry/db/models/manager/base.py
@@ -524,6 +524,12 @@ def create_silo_limited_copy(self: BaseManager[M], limit: SiloLimit) -> BaseMana
         "select_for_update": limit.create_override(cls.select_for_update),
         "update": limit.create_override(cls.update),
         "update_or_create": limit.create_override(cls.update_or_create),
+        "get_from_cache": limit.create_override(cls.get_from_cache)
+        if hasattr(cls, "get_from_cache")
+        else None,
+        "get_many_from_cache": limit.create_override(cls.get_many_from_cache)
+        if hasattr(cls, "get_many_from_cache")
+        else None,
     }
     manager_subclass = type(cls.__name__, (cls,), overrides)
     manager_instance = manager_subclass()