feat: Cache get_any_feature_view results (#5175)

sebastjaeger · Sebastian Jäger · web-flow · commit 924b8a384f7f · 2025-03-21T12:00:26.000-04:00
Cache get_any_feature_view results

Signed-off-by: Sebastian Jäger &lt;sebastjaeger@gmail.com&gt;
Co-authored-by: Sebastian Jäger &lt;sebastian.jaeger@ratepay.com&gt;
diff --git a/sdk/python/feast/infra/registry/proto_registry_utils.py b/sdk/python/feast/infra/registry/proto_registry_utils.py
@@ -71,6 +71,31 @@ def wrapper(
     return wrapper
 
 
+def registry_proto_cache_name(func):
+    cache_proto_key = None
+    cache = {}
+
+    @wraps(func)
+    def wrapper(registry_proto: RegistryProto, name: str, project: str):
+        nonlocal cache_proto_key, cache
+
+        proto_key = (id(registry_proto), registry_proto.version_id)
+
+        if proto_key != cache_proto_key:
+            cache = {}
+            cache_proto_key = proto_key
+
+        key = (project, name)
+        if key in cache:
+            return cache[key]
+        else:
+            value = func(registry_proto, name, project)
+            cache[key] = value
+            return value
+
+    return wrapper
+
+
 def get_project_metadata(
     registry_proto: Optional[RegistryProto], project: str
 ) -> Optional[ProjectMetadataProto]:
@@ -94,6 +119,7 @@ def get_feature_service(
     raise FeatureServiceNotFoundException(name, project=project)
 
 
+@registry_proto_cache_name
 def get_any_feature_view(
     registry_proto: RegistryProto, name: str, project: str
 ) -> BaseFeatureView:
