chore: speed up mqtt session shutdown by avoiding a sleep

Author: allenporter

roborock/mqtt/roborock_session.py

@@ -49,6 +49,7 @@ def __init__(self, params: MqttParams):
         self._params = params
         self._background_task: asyncio.Task[None] | None = None
         self._healthy = False
+        self._stop = False
         self._backoff = MIN_BACKOFF_INTERVAL
         self._client: aiomqtt.Client | None = None
         self._client_lock = asyncio.Lock()
@@ -81,6 +82,7 @@ async def start(self) -> None:
 
     async def close(self) -> None:
         """Cancels the MQTT loop and shutdown the client library."""
+        self._stop = True
         if self._background_task:
             self._background_task.cancel()
             try:
@@ -135,10 +137,14 @@ async def _run_task(self, start_future: asyncio.Future[None] | None) -> None:
                 _LOGGER.exception("Uncaught error during MQTT session: %s", err)
 
             self._healthy = False
+            if self._stop:
+                _LOGGER.debug("MQTT session closed, stopping retry loop")
+                return
             _LOGGER.info("MQTT session disconnected, retrying in %s seconds", self._backoff.total_seconds())
             await asyncio.sleep(self._backoff.total_seconds())
             self._backoff = min(self._backoff * BACKOFF_MULTIPLIER, MAX_BACKOFF_INTERVAL)
 
+
     @asynccontextmanager
     async def _mqtt_client(self, params: MqttParams) -> aiomqtt.Client:
         """Connect to the MQTT broker and listen for messages."""