main.py

from fastapi import FastAPI, status

from examples.probes import (
    LIVENESS_CHECKS,
    READINESS_CHECKS,
    READINESS_CHECKS_FAIL,
    READINESS_CHECKS_SUCCESS,
    STARTUP_CHECKS,
    custom_handler,
)
from fast_healthchecks.integrations.fastapi import HealthcheckRouter, Probe

app_integration = FastAPI()
app_integration.include_router(
    HealthcheckRouter(
        Probe(
            name="liveness",
            checks=LIVENESS_CHECKS,
        ),
        Probe(
            name="readiness",
            checks=READINESS_CHECKS,
        ),
        Probe(
            name="startup",
            checks=STARTUP_CHECKS,
        ),
        debug=True,
        prefix="/health",
    ),
)

app_success = FastAPI()
app_success.include_router(
    HealthcheckRouter(
        Probe(
            name="liveness",
            checks=[],
        ),
        Probe(
            name="readiness",
            checks=READINESS_CHECKS_SUCCESS,
        ),
        Probe(
            name="startup",
            checks=[],
        ),
        debug=True,
        prefix="/health",
    ),
)

app_fail = FastAPI()
app_fail.include_router(
    HealthcheckRouter(
        Probe(
            name="liveness",
            checks=[],
        ),
        Probe(
            name="readiness",
            checks=READINESS_CHECKS_FAIL,
        ),
        Probe(
            name="startup",
            checks=[],
        ),
        debug=True,
        prefix="/health",
    ),
)

app_custom = FastAPI()
app_custom.include_router(
    HealthcheckRouter(
        Probe(
            name="liveness",
            checks=[],
            summary="Check if the application is alive",
        ),
        Probe(
            name="readiness",
            checks=READINESS_CHECKS_SUCCESS,
            summary="Check if the application is ready",
        ),
        Probe(
            name="startup",
            checks=[],
            summary="Check if the application is starting up",
        ),
        success_handler=custom_handler,
        failure_handler=custom_handler,
        success_status=status.HTTP_200_OK,
        failure_status=status.HTTP_503_SERVICE_UNAVAILABLE,
        debug=True,
        prefix="/custom_health",
    ),
)