Add health check endpoints

Author: rragundez

src/app/api/v1/__init__.py

@@ -1,5 +1,6 @@
 from fastapi import APIRouter
 
+from .health import router as health_router
 from .login import router as login_router
 from .logout import router as logout_router
 from .posts import router as posts_router
@@ -9,6 +10,7 @@
 from .users import router as users_router
 
 router = APIRouter(prefix="/v1")
+router.include_router(health_router)
 router.include_router(login_router)
 router.include_router(logout_router)
 router.include_router(users_router)

src/app/api/v1/health.py

@@ -0,0 +1,58 @@
+import logging
+from datetime import UTC, datetime
+from typing import Annotated
+
+from fastapi import APIRouter, Depends, status
+from fastapi.responses import JSONResponse
+from redis.asyncio import Redis
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from ...core.config import settings
+from ...core.db.database import async_get_db
+from ...core.health import check_database_health, check_redis_health
+from ...core.schemas import HealthCheck, ReadyCheck
+from ...core.utils.cache import async_get_redis
+
+router = APIRouter(tags=["health"])
+
+STATUS_HEALTHY = "healthy"
+STATUS_UNHEALTHY = "unhealthy"
+
+LOGGER = logging.getLogger(__name__)
+
+
+@router.get("/health", response_model=HealthCheck)
+async def health():
+    http_status = status.HTTP_200_OK
+    response = {
+        "status": STATUS_HEALTHY,
+        "environment": settings.ENVIRONMENT.value,
+        "version": settings.APP_VERSION,
+        "timestamp": datetime.now(UTC).isoformat(timespec="seconds"),
+    }
+
+    return JSONResponse(status_code=http_status, content=response)
+
+
+@router.get("/ready", response_model=ReadyCheck)
+async def ready(redis: Annotated[Redis, Depends(async_get_redis)], db: Annotated[AsyncSession, Depends(async_get_db)]):
+    database_status = await check_database_health(db=db)
+    LOGGER.debug(f"Database health check status: {database_status}")
+    redis_status = await check_redis_health(redis=redis)
+    LOGGER.debug(f"Redis health check status: {redis_status}")
+
+    # Overall status
+    overall_status = STATUS_HEALTHY if database_status and redis_status else STATUS_UNHEALTHY
+    http_status = status.HTTP_200_OK if overall_status == STATUS_HEALTHY else status.HTTP_503_SERVICE_UNAVAILABLE
+
+    response = {
+        "status": overall_status,
+        "environment": settings.ENVIRONMENT.value,
+        "version": settings.APP_VERSION,
+        "app": STATUS_HEALTHY,
+        "database": STATUS_HEALTHY if database_status else STATUS_UNHEALTHY,
+        "redis": STATUS_HEALTHY if redis_status else STATUS_UNHEALTHY,
+        "timestamp": datetime.now(UTC).isoformat(timespec="seconds"),
+    }
+
+    return JSONResponse(status_code=http_status, content=response)

src/app/core/health.py

@@ -0,0 +1,25 @@
+import logging
+
+from redis.asyncio import Redis
+from sqlalchemy import text
+from sqlalchemy.ext.asyncio import AsyncSession
+
+LOGGER = logging.getLogger(__name__)
+
+
+async def check_database_health(db: AsyncSession) -> bool:
+    try:
+        await db.execute(text("SELECT 1"))
+        return True
+    except Exception as e:
+        LOGGER.exception(f"Database health check failed with error: {e}")
+        return False
+
+
+async def check_redis_health(redis: Redis) -> bool:
+    try:
+        await redis.ping()
+        return True
+    except Exception as e:
+        LOGGER.exception(f"Redis health check failed with error: {e}")
+        return False

src/app/core/schemas.py

@@ -1,15 +1,26 @@
 import uuid as uuid_pkg
-from uuid6 import uuid7
 from datetime import UTC, datetime
 from typing import Any
 
 from pydantic import BaseModel, Field, field_serializer
+from uuid6 import uuid7
 
 
 class HealthCheck(BaseModel):
-    name: str
+    status: str
+    environment: str
+    version: str
+    timestamp: str
+
+
+class ReadyCheck(BaseModel):
+    status: str
+    environment: str
     version: str
-    description: str
+    app: str
+    database: str
+    redis: str
+    timestamp: str
 
 
 # -------------- mixins --------------

src/app/core/utils/cache.py

@@ -1,7 +1,7 @@
 import functools
 import json
 import re
-from collections.abc import Callable
+from collections.abc import AsyncGenerator, Callable
 from typing import Any
 
 from fastapi import Request
@@ -173,13 +173,13 @@ async def _delete_keys_by_pattern(pattern: str) -> None:
     """
     if client is None:
         return
-    
-    cursor = 0 
+
+    cursor = 0
     while True:
         cursor, keys = await client.scan(cursor, match=pattern, count=100)
         if keys:
             await client.delete(*keys)
-        if cursor == 0: 
+        if cursor == 0:
             break
 
 
@@ -335,3 +335,12 @@ async def inner(request: Request, *args: Any, **kwargs: Any) -> Any:
         return inner
 
     return wrapper
+
+
+async def async_get_redis() -> AsyncGenerator[Redis, None]:
+    """Get a Redis client from the pool for each request."""
+    client = Redis(connection_pool=pool)
+    try:
+        yield client
+    finally:
+        await client.aclose()  # type: ignore