diff options
| author | Federico Caselli <cfederico87@gmail.com> | 2022-09-15 20:18:58 +0000 |
|---|---|---|
| committer | Gerrit Code Review <gerrit@ci3.zzzcomputing.com> | 2022-09-15 20:18:58 +0000 |
| commit | 04ea009a83d9c5f1f9e7ae66420e9124bae7ce66 (patch) | |
| tree | b4afbfa17752980c91715d86c254777f56fa38cd /lib/sqlalchemy/ext/asyncio/session.py | |
| parent | d9ecad5af633992ffdf85f2f8925a55d6221ff76 (diff) | |
| parent | 939f77c94b4e47ea611672a85047b61ab7e3590b (diff) | |
| download | sqlalchemy-04ea009a83d9c5f1f9e7ae66420e9124bae7ce66.tar.gz | |
Merge "Improved usage of ``asyncio.shield()``" into main
Diffstat (limited to 'lib/sqlalchemy/ext/asyncio/session.py')
| -rw-r--r-- | lib/sqlalchemy/ext/asyncio/session.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/lib/sqlalchemy/ext/asyncio/session.py b/lib/sqlalchemy/ext/asyncio/session.py index d4b6b6d50..60b77f3ea 100644 --- a/lib/sqlalchemy/ext/asyncio/session.py +++ b/lib/sqlalchemy/ext/asyncio/session.py @@ -851,7 +851,8 @@ class AsyncSession(ReversibleProxy[Session]): return self async def __aexit__(self, type_: Any, value: Any, traceback: Any) -> None: - await asyncio.shield(self.close()) + task = asyncio.create_task(self.close()) + await asyncio.shield(task) def _maker_context_manager(self: _AS) -> _AsyncSessionContextManager[_AS]: return _AsyncSessionContextManager(self) @@ -1516,7 +1517,8 @@ class _AsyncSessionContextManager(Generic[_AS]): await self.trans.__aexit__(type_, value, traceback) await self.async_session.__aexit__(type_, value, traceback) - await asyncio.shield(go()) + task = asyncio.create_task(go()) + await asyncio.shield(task) class AsyncSessionTransaction( |