Skip to content

Commit 1293cb5

Browse files
committed
added some missing stuff
1 parent 136b21c commit 1293cb5

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

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

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55
from simcore_postgres_database.utils_aiosqlalchemy import ( # type: ignore[import-not-found] # this on is unclear
66
get_pg_engine_stateinfo,
77
)
8+
from sqlalchemy.ext.asyncio import AsyncEngine
89

910
from ..db_asyncpg_utils import create_async_engine_and_pg_database_ready
1011
from ..logging_utils import log_context
@@ -31,3 +32,8 @@ async def close_db_connection(app: FastAPI) -> None:
3132
with log_context(_logger, logging.DEBUG, f"db disconnect of {app.state.engine}"):
3233
if engine := app.state.engine:
3334
await engine.dispose()
35+
36+
37+
def get_engine(app: FastAPI) -> AsyncEngine:
38+
assert isinstance(app.state.engine, AsyncEngine) # nosec
39+
return app.state.engine

0 commit comments

Comments
 (0)