feat(engine): Add health check endpoints

TracecatHQ · Mar 19, 2024 · d4863f9 · d4863f9
1 parent 5ed4ae2
commit d4863f9
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 3 deletions.
diff --git a/aws/stack.py b/aws/stack.py
@@ -136,7 +136,7 @@ def __init__(self, scope: Construct, id: str, **kwargs) -> None:
                 build_args={"API_MODULE": "tracecat.api.app:app"},
             ),
             health_check=ecs.HealthCheck(
-                command=["CMD-SHELL", "curl -f http://localhost:8000"],
+                command=["CMD-SHELL", "curl -f http://localhost:8000/health"],
                 interval=Duration.seconds(120),
                 retries=5,
                 start_period=Duration.seconds(60),
@@ -167,7 +167,7 @@ def __init__(self, scope: Construct, id: str, **kwargs) -> None:
                 build_args={"API_MODULE": "tracecat.runner.app:app", "PORT": "8001"},
             ),
             health_check=ecs.HealthCheck(
-                command=["CMD-SHELL", "curl -f http://localhost:8001"],
+                command=["CMD-SHELL", "curl -f http://localhost:8001/health"],
                 interval=Duration.seconds(120),
                 retries=5,
                 start_period=Duration.seconds(60),

diff --git a/tracecat/api/app.py b/tracecat/api/app.py
@@ -94,6 +94,11 @@ def root() -> dict[str, str]:
     return {"message": "Hello world. I am the API."}
 
 
+@app.get("/health")
+def check_health() -> dict[str, str]:
+    return {"message": "Hello world. I am the API. This is the health endpoint."}
+
+
 ### Workflows
 
 

diff --git a/tracecat/runner/app.py b/tracecat/runner/app.py
@@ -121,7 +121,12 @@ async def valid_workflow(workflow_id: str) -> str:
 # Endpoints
 @app.get("/")
 def root() -> dict[str, str]:
-    return {"message": "Hello. I am a runner."}
+    return {"message": "Hello. I am the runner."}
+
+
+@app.get("/health")
+def check_health() -> dict[str, str]:
+    return {"message": "Hello world. I am the runner. This is the health endpoint."}
 
 
 class StartWorkflowResponse(BaseModel):