From 6b004002f1b9670058525432b84f82482c3a6723 Mon Sep 17 00:00:00 2001 From: Kurbanov Bulat Date: Sat, 12 Feb 2022 13:38:36 +0300 Subject: [PATCH] Refactor healthcheck --- scripts/healthcheck.py | 5 +---- src/app/views.py | 7 ++++++- src/core/app.py | 3 ++- 3 files changed, 9 insertions(+), 6 deletions(-) diff --git a/scripts/healthcheck.py b/scripts/healthcheck.py index e21d83a..3f411e6 100644 --- a/scripts/healthcheck.py +++ b/scripts/healthcheck.py @@ -1,11 +1,8 @@ -import os - import httpx response = httpx.get( - "http://localhost:8080/healthcheck", - headers={"Authorization": os.environ["API_KEY"]}, + "http://localhost:8080/healthcheck" ) print(f"HEALTHCHECK STATUS: {response.status_code}") exit(0 if response.status_code == 200 else 1) diff --git a/src/app/views.py b/src/app/views.py index 661431f..c3e2acf 100644 --- a/src/app/views.py +++ b/src/app/views.py @@ -35,6 +35,11 @@ async def get_filename(book_id: int, file_type: str): return _get_filename(book.remote_id, book, file_type) -@router.get("/healthcheck") +healthcheck_router = APIRouter( + tags=["healthcheck"] +) + + +@healthcheck_router.get("/healthcheck") async def healthcheck(): return "Ok!" diff --git a/src/core/app.py b/src/core/app.py index b877121..92f0f61 100644 --- a/src/core/app.py +++ b/src/core/app.py @@ -2,13 +2,14 @@ from fastapi import FastAPI from prometheus_fastapi_instrumentator import Instrumentator -from app.views import router +from app.views import router, healthcheck_router def start_app() -> FastAPI: app = FastAPI() app.include_router(router) + app.include_router(healthcheck_router) Instrumentator().instrument(app).expose(app, include_in_schema=True)