code refactoring

amenezes · Mar 23, 2024 · 2423e74 · 2423e74
1 parent d52def3
commit 2423e74
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 5 deletions.
diff --git a/rabbit/client.py b/rabbit/client.py
@@ -74,7 +74,7 @@ async def persistent_connect(self, **kwargs) -> None:
                 self.transport, self.protocol = await aioamqp.connect(**kwargs)
                 await self.protocol.wait_closed()
                 self.transport.close()
-            except (OSError, aioamqp.exceptions.AmqpClosedConnection) as err:
+            except (OSError, AmqpClosedConnection) as err:
                 logger.error(
                     f"ConnectionError: [error='{err}', host='{kwargs.get('host')}', port={kwargs.get('port')}, login='{kwargs.get('login')}']"
                 )

diff --git a/rabbit/dlx.py b/rabbit/dlx.py
@@ -43,8 +43,7 @@ def channel(self, channel: Channel) -> None:
     async def configure(self) -> None:
         """Configure DLX channel, queues and exchange."""
         try:
-            await self._configure_queue()
-            await self._configure_exchange()
+            await asyncio.gather(self._configure_queue(), self._configure_exchange())
             await self._configure_queue_bind()
         except AttributeNotInitialized:
             logger.debug("Waiting client initialization...DLX")

diff --git a/rabbit/subscribe.py b/rabbit/subscribe.py
@@ -90,8 +90,7 @@ async def configure(self, channel: Union[None, Channel] = None) -> None:
         """Configure subscriber channel, queues and exchange."""
         await self.qos(prefetch_count=self.concurrent)
         with suppress(SynchronizationError):
-            await self._configure_queue()
-            await self._dlx.configure()
+            await asyncio.gather(self._configure_queue(), self._dlx.configure())
             await self._configure_exchange()
             await self._configure_queue_bind()