feat(mqtt): improve connection handling

adutchak · Apr 9, 2024 · 1e4ff5b · 1e4ff5b
1 parent aebb027
commit 1e4ff5b
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/main.go b/main.go
@@ -75,8 +75,9 @@ func main() {
 				continue
 			}
 
-			if token := mqttClient.Connect(); token.Wait() && token.Error() != nil {
-				l.Error("Error connecting to MQTT", token.Error())
+			err = mqttclient.ConnectToMqtt(mqttClient)
+			if err != nil {
+				l.Error("Failed to connect to MQTT", err)
 			}
 			defer mqttClient.Disconnect(250)
 

diff --git a/pkg/mqttclient/mqttclient.go b/pkg/mqttclient/mqttclient.go
@@ -37,3 +37,10 @@ func GetMqttClient(configuration *config.Config) (mqtt.Client, error) {
 	}
 	return client, nil
 }
+
+func ConnectToMqtt(client mqtt.Client) error {
+	if token := client.Connect(); token.Wait() && token.Error() != nil {
+		return token.Error()
+	}
+	return nil
+}