Skip to content

Commit

Permalink
add logging to CachingRegistry
Browse files Browse the repository at this point in the history
  • Loading branch information
zabarn committed Feb 22, 2025
1 parent df265fc commit e659fdc
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions sdk/python/feast/infra/registry/caching_registry.py
Original file line number Diff line number Diff line change
Expand Up @@ -37,6 +37,7 @@ def __init__(self, project: str, cache_ttl_seconds: int, cache_mode: str):
)
self.cached_registry_proto = self.proto()
self.cached_registry_proto_created = _utc_now()
logger.info(f"Registry initialized with cache mode: {cache_mode}")
if cache_mode == "thread":
self._start_thread_async_refresh(cache_ttl_seconds)
atexit.register(self._exit_handler)
Expand Down Expand Up @@ -456,6 +457,7 @@ def _refresh_cached_registry_if_necessary(self):
self.refresh()

def _start_thread_async_refresh(self, cache_ttl_seconds):
logger.info(f"Starting registry cache refresh thread with TTL {cache_ttl_seconds}")
self.refresh()
if cache_ttl_seconds <= 0:
return
Expand All @@ -464,6 +466,8 @@ def _start_thread_async_refresh(self, cache_ttl_seconds):
)
self.registry_refresh_thread.daemon = True
self.registry_refresh_thread.start()
logger.info("Registry cache refresh thread started")

def _exit_handler(self):
logger.info("Exiting, cancelling registry cache refresh thread")
self.registry_refresh_thread.cancel()

0 comments on commit e659fdc

Please sign in to comment.