feat: Add health endpoint to py server (#3202)

robhowley · Rob Howley · web-flow · commit 43222f21046c · 2022-09-12T11:49:23.000-07:00
feat: add health endpoint to py server

Signed-off-by: Rob Howley &lt;rhowley@seatgeek.com&gt;

Signed-off-by: Rob Howley &lt;rhowley@seatgeek.com&gt;
Co-authored-by: Rob Howley &lt;rhowley@seatgeek.com&gt;
diff --git a/sdk/python/feast/feature_server.py b/sdk/python/feast/feature_server.py
@@ -4,7 +4,7 @@
 
 import pandas as pd
 import uvicorn
-from fastapi import FastAPI, HTTPException, Request
+from fastapi import FastAPI, HTTPException, Request, Response, status
 from fastapi.logger import logger
 from fastapi.params import Depends
 from google.protobuf.json_format import MessageToDict, Parse
@@ -124,6 +124,10 @@ def write_to_online_store(body=Depends(get_body)):
             # Raise HTTPException to return the error message to the client
             raise HTTPException(status_code=500, detail=str(e))
 
+    @app.get("/health")
+    def health():
+        return Response(status_code=status.HTTP_200_OK)
+
     return app
 
 
