fix: Fixed registry cache init

ntkathole · ntkathole · commit 38ae4ddebcdb · 2025-09-15T12:28:59.000+05:30
Signed-off-by: ntkathole &lt;nikhilkathole2683@gmail.com&gt;
diff --git a/sdk/python/feast/infra/registry/caching_registry.py b/sdk/python/feast/infra/registry/caching_registry.py
@@ -435,16 +435,15 @@ def _refresh_cached_registry_if_necessary(self):
         if self.cache_mode == "sync":
 
             def is_cache_expired():
-                if self.cached_registry_proto == RegistryProto():
-                    if self.cached_registry_proto_ttl.total_seconds() == 0:
-                        return False
-                    else:
-                        return True
-
-                # Cache is expired if it's None or creation time is None
                 if (
                     self.cached_registry_proto is None
-                    or not hasattr(self, "cached_registry_proto_created")
+                    or self.cached_registry_proto == RegistryProto()
+                ):
+                    return True
+
+                # Cache is expired if creation time is None
+                if (
+                    not hasattr(self, "cached_registry_proto_created")
                     or self.cached_registry_proto_created is None
                 ):
                     return True
diff --git a/sdk/python/feast/infra/registry/snowflake.py b/sdk/python/feast/infra/registry/snowflake.py
@@ -404,8 +404,6 @@ def _apply_object(
             if not self.purge_feast_metadata:
                 self._set_last_updated_metadata(update_datetime, project)
 
-        self.refresh()
-
     def apply_permission(
         self, permission: Permission, project: str, commit: bool = True
     ):
@@ -494,7 +492,6 @@ def _delete_object(
                 raise not_found_exception(name, project)
             self._set_last_updated_metadata(_utc_now(), project)
 
-            self.refresh()
             return cursor.rowcount
 
     def delete_permission(self, name: str, project: str, commit: bool = True):
@@ -1128,6 +1125,18 @@ def process_project(project: Project):
             project_name = project.name
             last_updated_timestamp = project.last_updated_timestamp
 
+            try:
+                cached_project = self.get_project(project_name, True)
+            except ProjectObjectNotFoundException:
+                cached_project = None
+
+            allow_cache = False
+
+            if cached_project is not None:
+                allow_cache = (
+                    last_updated_timestamp <= cached_project.last_updated_timestamp
+                )
+
             r.projects.extend([project.to_proto()])
             last_updated_timestamps.append(last_updated_timestamp)
 
@@ -1142,7 +1151,7 @@ def process_project(project: Project):
                 (self.list_validation_references, r.validation_references),
                 (self.list_permissions, r.permissions),
             ]:
-                objs: List[Any] = lister(project_name, allow_cache=False)  # type: ignore
+                objs: List[Any] = lister(project_name, allow_cache)  # type: ignore
                 if objs:
                     obj_protos = [obj.to_proto() for obj in objs]
                     for obj_proto in obj_protos:
