bellingcat/auto-archiver-api
1
0
Fork 0
mirror of https://github.com/bellingcat/auto-archiver-api.git synced 2026-06-12 05:28:34 +03:00
Code Issues Packages Projects Releases Wiki Activity

Instrument exceptions by type

Browse Source
This commit is contained in:
Lilia Kai
2023-12-21 16:17:34 +01:00
parent c058bfd067
commit 5258da65a5
1 changed files with 15 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
src/main.py
View File

@@ -12,6 +12,7 @@ from loguru import logger
from datetime import datetime from datetime import datetime
import sqlalchemy import sqlalchemy
from prometheus_fastapi_instrumentator import Instrumentator from prometheus_fastapi_instrumentator import Instrumentator
from prometheus_client import Counter
from worker import create_archive_task, create_sheet_task, celery, insert_result_into_db from worker import create_archive_task, create_sheet_task, celery, insert_result_into_db
@@ -39,8 +40,13 @@ app.add_middleware(
allow_headers=["*"], allow_headers=["*"],
) )
EXCEPTION_COUNTER = Counter(
"exceptions",
"Number of times a certain exception has occurred.",
labelnames=("types",)
)
# prometheus exposed in /metrics with authentication # prometheus exposed in /metrics with authentication
Instrumentator(should_group_status_codes=False, excluded_handlers=["/metrics"]).instrument(app).expose(app, dependencies=[Depends(get_basic_auth)]) Instrumentator(should_group_status_codes=False, excluded_handlers=["/metrics"]).instrument(app).expose(app, dependencies=[Depends(get_server_auth)])
app.mount("/static", StaticFiles(directory="static"), name="static") app.mount("/static", StaticFiles(directory="static"), name="static")
@@ -54,13 +60,18 @@ def get_db():
try: yield session try: yield session
finally: session.close() finally: session.close()
# logging configurations # logging configurations
logger.add("logs/api_logs.log", retention="30 days", rotation="3 days") logger.add("logs/api_logs.log", retention="30 days", rotation="3 days")
@app.middleware("http") @app.middleware("http")
async def logging_middleware(request: Request, call_next): async def logging_middleware(request: Request, call_next):
response = await call_next(request) try:
logger.info(f"{request.client.host}:{request.client.port} {request.method} {request.url._url} - HTTP {response.status_code}") response = await call_next(request)
return response logger.info(f"{request.client.host}:{request.client.port} {request.method} {request.url._url} - HTTP {response.status_code}")
return response
except Exception as e:
EXCEPTION_COUNTER.labels(type(e).__name__).inc()
raise e
@app.get("/") @app.get("/")
async def home(request: Request): async def home(request: Request):