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

mqtt: use component_init for post initalization 

Signed-off-by:  Eric Callahan <arksine.code@gmail.com>
This commit is contained in:
Eric Callahan 2021-11-02 12:47:45 -04:00 committed by Eric Callahan
parent fd59ace4f8
commit 799e763050
1 changed files with 14 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
moonraker/components/mqtt.py
View File

@ -229,31 +229,28 @@ class MQTTClient(APITransport, Subscribable):
f"Moonraker API topics - Request: {self.api_request_topic}, "
f"Response: {self.api_resp_topic}")
self.event_loop.register_callback(self._initialize)
async def _initialize(self) -> None:
async def component_init(self) -> None:
# We must wait for the IOLoop (asyncio event loop) to start
# prior to retreiving it
self.helper = AIOHelper(self.client)
if self.user_name is not None:
self.client.username_pw_set(self.user_name, self.password)
retries = 15
while retries:
retries = 5
for _ in range(retries):
try:
self.client.connect(self.address, self.port)
except ConnectionRefusedError:
retries -= 1
if retries:
logging.info("Unable to connect to MQTT broker, "
f"retries remaining: {retries}")
await asyncio.sleep(2.)
continue
self.server.set_failed_component("mqtt")
self.server.add_warning(
f"MQTT Broker Connection at ({self.address}, {self.port}) "
"refused. Check your client and broker configuration.")
return
break
logging.info("Unable to connect to MQTT broker, "
f"retries remaining: {retries}")
await asyncio.sleep(2.)
else:
break
else:
self.server.set_failed_component("mqtt")
self.server.add_warning(
f"MQTT Broker Connection at ({self.address}, {self.port}) "
"refused. Check your client and broker configuration.")
return
self.client.socket().setsockopt(
socket.SOL_SOCKET, socket.SO_SNDBUF, 2048)