check if _disconnect_waiter is None when closing connection

Yakifo · Nov 29, 2022 · 19960e6 · 19960e6
1 parent 09ac98d
commit 19960e6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/amqtt/mqtt/protocol/client_handler.py b/amqtt/mqtt/protocol/client_handler.py
@@ -163,7 +163,7 @@ async def handle_pingresp(self, pingresp: PingRespPacket):
 
     async def handle_connection_closed(self):
         self.logger.debug("Broker closed connection")
-        if not self._disconnect_waiter.done():
+        if self._disconnect_waiter is not None and not self._disconnect_waiter.done():
             self._disconnect_waiter.set_result(None)
 
     async def wait_disconnect(self):