3diy
/
moonraker
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

mqtt: fix connect/reconnect issues 

Handle all potential exceptions.  Run the connect/disconnect
in another thread, as its possible for some calls to block
the event loop.

Signed-off-by:  Eric Callahan <arksine.code@gmail.com>
This commit is contained in:
Eric Callahan 2021-12-06 08:38:27 -05:00
parent ac73036857
commit 1ad83cec97
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
moonraker/components/mqtt.py
View File

@ -232,8 +232,9 @@ class MQTTClient(APITransport, Subscribable):
retries = 5
for _ in range(retries):
try:
self.client.connect(self.address, self.port)
except (ConnectionRefusedError, socket.gaierror) as e:
await self.event_loop.run_in_thread(
self.client.connect, self.address, self.port)
except Exception as e:
logging.info(f"MQTT connection error, {e}, "
f"retries remaining: {retries}")
await asyncio.sleep(2.)
@ -351,14 +352,15 @@ class MQTTClient(APITransport, Subscribable):
async def _do_reconnect(self) -> None:
logging.info("Attempting MQTT Reconnect")
self.event_loop
while True:
try:
await asyncio.sleep(2.)
except asyncio.CancelledError:
break
try:
self.client.reconnect()
except (ConnectionRefusedError, socket.gaierror):
await self.event_loop.run_in_thread(self.client.reconnect)
except Exception:
continue
self.client.socket().setsockopt(
socket.SOL_SOCKET, socket.SO_SNDBUF, 2048)