attempt to prevent disposed exception on mqtt client

amadeo-alex · Oct 12, 2024 · af218bd · af218bd
1 parent 5fc4112
commit af218bd
Show file tree

Hide file tree

Showing 6 changed files with 12 additions and 9 deletions.
diff --git a/src/HASS.Agent/HASS.Agent.Satellite.Service/Functions/HelperFunctions.cs b/src/HASS.Agent/HASS.Agent.Satellite.Service/Functions/HelperFunctions.cs
@@ -61,7 +61,7 @@ internal static async Task ShutdownAsync(TimeSpan waitBeforeClosing)
 
                 // stop mqtt
                 await Variables.MqttManager.AnnounceAvailabilityAsync(true);
-                Variables.MqttManager.Disconnect();
+                await Variables.MqttManager.DisconnectAsync();
 
                 // stop rpc
                 Variables.RpcServer?.Dispose();

diff --git a/src/HASS.Agent/HASS.Agent.Satellite.Service/MQTT/MqttManager.cs b/src/HASS.Agent/HASS.Agent.Satellite.Service/MQTT/MqttManager.cs
@@ -520,14 +520,14 @@ public async Task ClearDeviceConfigAsync()
         /// <summary>
         /// Disconnect from the broker (if connected)
         /// </summary>
-        public void Disconnect()
+        public async Task DisconnectAsync()
         {
             if (_mqttClient == null)
                 return;
 
             if (IsConnected())
             {
-                _mqttClient.InternalClient.DisconnectAsync();
+                await _mqttClient.StopAsync();
                 _mqttClient.Dispose();
             }
 

diff --git a/src/HASS.Agent/HASS.Agent.Shared/Mqtt/IMqttManager.cs b/src/HASS.Agent/HASS.Agent.Shared/Mqtt/IMqttManager.cs
@@ -21,7 +21,7 @@ public interface IMqttManager
         MqttStatus GetStatus();
         Task AnnounceAvailabilityAsync(bool offline = false);
         Task ClearDeviceConfigAsync();
-        void Disconnect();
+        Task DisconnectAsync();
         Task SubscribeAsync(AbstractCommand command);
         Task UnsubscribeAsync(AbstractCommand command);
 

diff --git a/src/HASS.Agent/HASS.Agent/Forms/Configuration.cs b/src/HASS.Agent/HASS.Agent/Forms/Configuration.cs
@@ -182,7 +182,7 @@ private async void ProcessChanges()
                 await Task.Delay(250);
 
                 // disconnect mqtt so we don't get announced again
-                await Task.Run(Variables.MqttManager.Disconnect);
+                await Variables.MqttManager.DisconnectAsync();
 
                 forceRestart = true;
             }

diff --git a/src/HASS.Agent/HASS.Agent/Functions/HelperFunctions.cs b/src/HASS.Agent/HASS.Agent/Functions/HelperFunctions.cs
@@ -165,7 +165,7 @@ internal static async Task ShutdownAsync(TimeSpan waitBeforeClosing)
                 {
                     // stop mqtt
                     await Variables.MqttManager.AnnounceAvailabilityAsync(true);
-                    Variables.MqttManager.Disconnect();
+                    await Variables.MqttManager.DisconnectAsync();
 
                     // remove tray icon
                     Variables.MainForm?.HideTrayIcon();

diff --git a/src/HASS.Agent/HASS.Agent/MQTT/MqttManager.cs b/src/HASS.Agent/HASS.Agent/MQTT/MqttManager.cs
@@ -611,15 +611,18 @@ public async Task ClearDeviceConfigAsync()
         /// <summary>
         /// Disconnect from the broker (if connected)
         /// </summary>
-        public void Disconnect()
+        public async Task DisconnectAsync()
         {
             if (!Variables.AppSettings.MqttEnabled)
                 return;
 
+            if (_mqttClient == null)
+                return;
+
             if (IsConnected())
             {
-                _mqttClient?.InternalClient?.DisconnectAsync();
-                _mqttClient?.Dispose();
+                await _mqttClient.StopAsync();
+                _mqttClient.Dispose();
             }
 
             Log.Information("[MQTT] Disconnected");