administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-03 01:35:22 +03:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Use async to handle reconnects

Browse Source
This commit is contained in:
Nick Mowen 2022-11-19 10:12:25 -07:00
parent cc524eb813
commit 3b8c3c7514
1 changed files with 23 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
frigate/mqtt.py
View File

@ -1,3 +1,4 @@
import datetime
import json import json
import logging import logging
import threading import threading
@ -29,9 +30,10 @@ class FrigateMqttClient:
self.mqtt_config = config.mqtt self.mqtt_config = config.mqtt
self.camera_metrics: dict[str, CameraMetricsTypes] = camera_metrics self.camera_metrics: dict[str, CameraMetricsTypes] = camera_metrics
self.client: mqtt.Client = None self.client: mqtt.Client = None
self.__start() self.connected: bool = False
self._start()
def set_initial_topics(self) -> None: def _set_initial_topics(self) -> None:
"""Set initial state topics.""" """Set initial state topics."""
for camera_name, camera in self.config.cameras.items(): for camera_name, camera in self.config.cameras.items():
self.publish( self.publish(
@ -75,6 +77,10 @@ class FrigateMqttClient:
retain=False, retain=False,
) )
self.publish(
self.mqtt_config.topic_prefix + "/available", "online", retain=True
)
def on_recordings_command( def on_recordings_command(
self, client: mqtt.Client, userdata, message: mqtt.MQTTMessage self, client: mqtt.Client, userdata, message: mqtt.MQTTMessage
) -> None: ) -> None:
@ -278,8 +284,9 @@ class FrigateMqttClient:
"""Callback to restart frigate.""" """Callback to restart frigate."""
restart_frigate() restart_frigate()
def on_connect(self, client: mqtt.Client, userdata, flags, rc) -> None: def _on_connect(self, client: mqtt.Client, userdata, flags, rc) -> None:
"""Mqtt connection callback.""" """Mqtt connection callback."""
self.connected = True
threading.current_thread().name = "mqtt" threading.current_thread().name = "mqtt"
if rc != 0: if rc != 0:
if rc == 3: if rc == 3:
@ -300,14 +307,17 @@ class FrigateMqttClient:
logger.debug("MQTT connected") logger.debug("MQTT connected")
client.subscribe(f"{self.mqtt_config.topic_prefix}/#") client.subscribe(f"{self.mqtt_config.topic_prefix}/#")
self.publish( self._set_initial_topics()
self.mqtt_config.topic_prefix + "/available", "online", retain=True
)
def __start(self) -> None: def _on_disconnect(self, client: mqtt.Client, userdata, flags, rc) -> None:
"""Mqtt disconnection callback."""
self.connected = False
logger.error("MQTT disconnected")
def _start(self) -> None:
"""Start mqtt client.""" """Start mqtt client."""
self.client = mqtt.Client(client_id=self.mqtt_config.client_id) self.client = mqtt.Client(client_id=self.mqtt_config.client_id)
self.client.on_connect = self.on_connect self.client.on_connect = self._on_connect
self.client.will_set( self.client.will_set(
self.mqtt_config.topic_prefix + "/available", self.mqtt_config.topic_prefix + "/available",
payload="offline", payload="offline",
@ -369,27 +379,23 @@ class FrigateMqttClient:
self.mqtt_config.user, password=self.mqtt_config.password self.mqtt_config.user, password=self.mqtt_config.password
) )
try: try:
self.client.connect(self.mqtt_config.host, self.mqtt_config.port, 60) # https://stackoverflow.com/a/55390477
# with connect_async, retries are handled automatically
self.client.connect_async(self.mqtt_config.host, self.mqtt_config.port, 60)
self.client.loop_start()
except Exception as e: except Exception as e:
logger.error(f"Unable to connect to MQTT server: {e}") logger.error(f"Unable to connect to MQTT server: {e}")
self.client = None
return return
self.client.loop_start()
self.set_initial_topics()
def publish(self, topic: str, payload, retain: bool = False) -> None: def publish(self, topic: str, payload, retain: bool = False) -> None:
"""Wrapper for publishing when client is in valid state.""" """Wrapper for publishing when client is in valid state."""
if not self.client: if not self.connected:
logger.error(f"Unable to publish to {topic}: client is not connected") logger.error(f"Unable to publish to {topic}: client is not connected")
return return
self.client.publish(topic, payload, retain=retain) self.client.publish(topic, payload, retain=retain)
def add_topic_callback(self, topic: str, callback) -> None: def add_topic_callback(self, topic: str, callback) -> None:
if not self.client:
return
self.client.message_callback_add(topic, callback) self.client.message_callback_add(topic, callback)