diff --git a/scanner/scanner/database.py b/scanner/scanner/database.py index e5184339..547ecfcd 100644 --- a/scanner/scanner/database.py +++ b/scanner/scanner/database.py @@ -5,8 +5,10 @@ from logging import getLogger from typing import Any, cast from asyncpg import Connection, Pool, create_pool +from opentelemetry import trace logger = getLogger(__name__) +tracer = trace.get_tracer("kyoo.scanner") pool: Pool @@ -55,6 +57,7 @@ async def get_db_fapi(): yield db +@tracer.start_as_current_span("migrate") async def migrate(migrations_dir="./migrations"): async with get_db() as db: _ = await db.execute( diff --git a/scanner/scanner/requests.py b/scanner/scanner/requests.py index 50ad5bff..fd6b8bb3 100644 --- a/scanner/scanner/requests.py +++ b/scanner/scanner/requests.py @@ -75,7 +75,7 @@ class RequestProcessor: self._database.add_termination_listener(terminated) await self._database.add_listener("scanner_requests", process) - logger.info("Listening for requestes") + logger.info("Listening for requests") _ = await closed.wait() logger.info("stopping...") except CancelledError: