From 6a328de5d528184c2f7dc3ab43f12c66aee60637 Mon Sep 17 00:00:00 2001
From: Nick Thomas <me@ur.gs>
Date: Sun, 16 Oct 2022 21:27:48 +0100
Subject: [PATCH] Even more resiliency

---
 src/main.rs | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index 036315f..f5c4647 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -186,7 +186,9 @@ fn mqtt_connect(uri: String, client_id: String) -> Result<mqtt::Client, mqtt::Er
         .finalize();
 
     // Create a client.
-    let mqtt_client = mqtt::Client::new(mqtt_opts)?;
+    let mut mqtt_client = mqtt::Client::new(mqtt_opts)?;
+
+    mqtt_client.set_timeout(Duration::from_secs(5));
 
     // Define the set of options for the connection.
     let mqtt_conn_opts = mqtt::ConnectOptionsBuilder::new()
@@ -248,7 +250,7 @@ fn main() {
 
     // TODO: Publish Home Assistant autodiscovery message (retained)
 
-    loop {
+    while mqtt_client.is_connected() {
         match gather(&solax_url, &flags.solax_password) {
             Ok(msg) => {
                 if let Some(e) = publish(&mqtt_client, msg, mqtt_qos) {