fix: dont allow for new refresh threadpool tasks to start during shutdown

robhowley · robhowley · commit c04820cb35ae · 2024-10-14T14:40:06.000-04:00
Signed-off-by: Rob Howley &lt;howley.robert@gmail.com&gt;
diff --git a/sdk/python/feast/feature_server.py b/sdk/python/feast/feature_server.py
@@ -85,11 +85,12 @@ def stop_refresh():
             active_timer.cancel()
 
     def async_refresh():
+        if shutting_down:
+            return
+
         store.refresh_registry()
         nonlocal registry_proto
         registry_proto = store.registry.proto()
-        if shutting_down:
-            return
 
         if registry_ttl_sec:
             nonlocal active_timer
@@ -222,8 +223,12 @@ def write_to_online_store(body=Depends(get_body)):
         )
 
     @app.get("/health")
-    def health():
-        return Response(status_code=status.HTTP_200_OK)
+    async def health():
+        return (
+            Response(status_code=status.HTTP_200_OK)
+            if registry_proto
+            else Response(status_code=status.HTTP_503_SERVICE_UNAVAILABLE)
+        )
 
     @app.post("/materialize", dependencies=[Depends(inject_user_details)])
     def materialize(body=Depends(get_body)):
