From b973921dba5d6fb39457e834e60a39e35b3e7cad Mon Sep 17 00:00:00 2001 From: Vovetta Date: Tue, 25 Jul 2023 21:14:45 +0300 Subject: [PATCH] Fix: connection leak on TimeoutError --- asyncpg/compat.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/asyncpg/compat.py b/asyncpg/compat.py index b9b13fa5..7ed33bf5 100644 --- a/asyncpg/compat.py +++ b/asyncpg/compat.py @@ -58,8 +58,8 @@ async def wait_for(fut, timeout): try: return await asyncio.wait_for(fut, timeout) - except asyncio.CancelledError: - if fut.done(): + except (asyncio.CancelledError, asyncio.TimeoutError): + if fut.done() and not fut.cancelled(): return fut.result() else: raise