Add orchestration with enhanced health check

docker-compose.yml

@@ -1,12 +1,17 @@
-
 services:
   fastapi:
-    build: .
+    build:
+      context: .
+      dockerfile: Dockerfile
     image: fastapi-app
     networks:
       - app-network
     deploy:
       replicas: 3
+      restart_policy:
+        condition: on-failure
+    environment:
+      - APP_ENV=production
   prometheus:
     image: prom/prometheus:latest
     ports:

main.py

@@ -1,5 +1,6 @@
 import logging
-from fastapi import FastAPI
+import os
+from fastapi import FastAPI, HTTPException
 from prometheus_fastapi_instrumentator import Instrumentator
 from prometheus_client import Counter
 
@@ -33,4 +34,7 @@ async def info():
 async def health():
     logger.info("Accessing health endpoint")
     custom_requests.labels(endpoint="health").inc()
-    return {"status": "healthy"}
+    env = os.getenv("APP_ENV", "production")
+    if env != "production":
+        raise HTTPException(status_code=503, detail="Not in production mode")
+    return {"status": "healthy", "env": env}

project4_fastapi.txt

@@ -0,0 +1 @@
+{"message":"Welcome to my DevOps API!"}

project4_health.txt

@@ -0,0 +1 @@
+{"status":"healthy"}

project4_healthcheck.txt

@@ -0,0 +1 @@
+{"status":"healthy"}

project4_metrics.txt

@@ -0,0 +1 @@
+{"status":"success","data":{"resultType":"vector","result":[{"metric":{"__name__":"custom_requests_total","endpoint":"root","instance":"fastapi:80","job":"fastapi"},"value":[1756641419.820,"1"]},{"metric":{"__name__":"custom_requests_total","endpoint":"health","instance":"fastapi:80","job":"fastapi"},"value":[1756641419.820,"1"]}]}}

project4_metrics_final.txt

@@ -0,0 +1 @@
+{"status":"success","data":{"resultType":"vector","result":[]}}

project4_ps.txt

@@ -0,0 +1,6 @@
+NAME                        IMAGE                    COMMAND                  SERVICE      CREATED              STATUS              PORTS
+server-setup-fastapi-1      fastapi-app              "uvicorn main:app --…"   fastapi      About a minute ago   Up About a minute   
+server-setup-fastapi-2      fastapi-app              "uvicorn main:app --…"   fastapi      About a minute ago   Up About a minute   
+server-setup-fastapi-3      fastapi-app              "uvicorn main:app --…"   fastapi      About a minute ago   Up About a minute   
+server-setup-nginx-1        nginx:latest             "/docker-entrypoint.…"   nginx        About a minute ago   Up About a minute   0.0.0.0:80->80/tcp, [::]:80->80/tcp
+server-setup-prometheus-1   prom/prometheus:latest   "/bin/prometheus --c…"   prometheus   About a minute ago   Up About a minute   0.0.0.0:9090->9090/tcp, [::]:9090->9090/tcp

project4_test.txt

@@ -0,0 +1,25 @@
+============================= test session starts ==============================
+platform linux -- Python 3.12.3, pytest-8.4.1, pluggy-1.6.0 -- /home/ubuntu/server-setup/venv/bin/python3
+cachedir: .pytest_cache
+rootdir: /home/ubuntu/server-setup
+plugins: anyio-4.10.0, httpx-0.35.0
+collecting ... collected 4 items
+
+test_main.py::test_root PASSED                                           [ 25%]
+test_main.py::test_health FAILED                                         [ 50%]
+test_main.py::test_info PASSED                                           [ 75%]
+test_main.py::test_metrics PASSED                                        [100%]
+
+=================================== FAILURES ===================================
+_________________________________ test_health __________________________________
+
+    def test_health():
+        response = client.get("/health")
+>       assert response.status_code == 200
+E       assert 503 == 200
+E        +  where 503 = <Response [503 Service Unavailable]>.status_code
+
+test_main.py:15: AssertionError
+=========================== short test summary info ============================
+FAILED test_main.py::test_health - assert 503 == 200
+========================= 1 failed, 3 passed in 0.63s ==========================

test_main.py

@@ -13,7 +13,7 @@ def test_root():
 def test_health():
     response = client.get("/health")
     assert response.status_code == 200
-    assert response.json() == {"status": "healthy"}
+    assert response.json().get("status") == "healthy"
 
 
 def test_info():