No need to wait for a connection closure (#223)

* No need to wait for a connection closure

* CHANGELOG

Author: Pliner

CHANGELOG.md

@@ -1,6 +1,7 @@
 ## unreleased
 
 # [SELECT 1 only if we haven't seen any events](https://github.com/anna-money/asyncpg-listen/pull/222)
+# [No need to wait for a connection closure](https://github.com/anna-money/asyncpg-listen/pull/223)
 
 
 ## v0.0.8 (2025-05-25)

asyncpg_listen/listener.py

@@ -46,11 +46,13 @@ class NotificationListener:
     __slots__ = (
         "__connect",
         "__reconnect_delay",
+        "__tasks",
     )
 
     def __init__(self, connect: ConnectFunc, reconnect_delay: float = 5) -> None:
         self.__reconnect_delay = reconnect_delay
         self.__connect = connect
+        self.__tasks = set()
 
     async def run(
         self,
@@ -155,7 +157,9 @@ async def __read_notifications(
                             await asyncio.sleep(per_attempt_keep_alive_budget - elapsed)
                     logger.warning("Connection was lost")
                 finally:
-                    await asyncio.shield(connection.close())
+                    close_task = asyncio.create_task(connection.close())
+                    close_task.add_done_callback(self.__tasks.discard)
+                    self.__tasks.add(close_task)
             except Exception:
                 if failed_connect_attempts < MAX_SILENCED_FAILED_CONNECT_ATTEMPTS:
                     logger.warning("Connection was lost or not established", exc_info=True)