cleanup

Author: pcrespov

packages/service-library/src/servicelib/aiohttp/db_asyncpg_engine.py

@@ -40,6 +40,15 @@ def get_async_engine(app: web.Application) -> AsyncEngine:
 
 
 async def connect_to_db(app: web.Application, settings: PostgresSettings) -> None:
+    """
+    - db services up, data migrated and ready to use
+    - sets an engine in app state (use `get_async_engine(app)` to retrieve)
+    """
+    if settings.POSTGRES_CLIENT_NAME:
+        settings = settings.copy(
+            update={"POSTGRES_CLIENT_NAME": settings.POSTGRES_CLIENT_NAME + "-asyncpg"}
+        )
+
     engine = await create_async_engine_and_pg_database_ready(settings)
     _set_async_engine_to_app_state(app, engine)
 

packages/service-library/src/servicelib/db_asyncpg_utils.py

@@ -22,6 +22,12 @@
 async def create_async_engine_and_pg_database_ready(
     settings: PostgresSettings,
 ) -> AsyncEngine:
+    """
+    - creates asynio engine
+    - waits until db service is up
+    - waits until db data is migrated (i.e. ready to use)
+    - returns engine
+    """
     with log_context(
         _logger, logging.DEBUG, f"connection to db {settings.dsn_with_async_sqlalchemy}"
     ):