fix: If the client attempts to close the connection while it is openi…

…ng, do not retry opening the connection.
open-contracting · Jan 23, 2024 · 63db184 · 63db184
1 parent 5391925
commit 63db184
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/docs/changelog.rst b/docs/changelog.rst
@@ -10,6 +10,11 @@ Changed
 -  Remove log message for declaring exchange.
 -  Remove ``yapw.clients.Base.__getsafe__``.
 
+Fixed
+~~~~~
+
+-  If the client attempts to close the connection while it is opening, do not retry opening the connection.
+
 0.1.3 (2023-07-03)
 ------------------
 

diff --git a/yapw/clients.py b/yapw/clients.py
@@ -18,7 +18,7 @@
 
 import pika
 from pika.adapters.asyncio_connection import AsyncioConnection
-from pika.exceptions import ProbableAccessDeniedError, ProbableAuthenticationError
+from pika.exceptions import ConnectionOpenAborted, ProbableAccessDeniedError, ProbableAuthenticationError
 from pika.exchange_type import ExchangeType
 
 from yapw.decorators import halt
@@ -438,7 +438,7 @@ def reset(self) -> None:
 
     def connection_open_error_callback(self, connection: pika.connection.Connection, error: Exception | str) -> None:
         """Retry, once the connection couldn't be established."""
-        if isinstance(error, ProbableAccessDeniedError | ProbableAuthenticationError):
+        if isinstance(error, ConnectionOpenAborted | ProbableAccessDeniedError | ProbableAuthenticationError):
             logger.error("Stopping: %r", error)
             self.connection.ioloop.stop()
         else: