Extended AfterDisconnect Support (#93)

Author: pglombardo

Examples/Reconnect/Program.cs

@@ -135,5 +135,6 @@
     }
 }
 
-Console.WriteLine("Disconnecting gracefully...");
+Console.WriteLine("Disconnecting gracefully and waiting 5 seconds...");
 await client.DisconnectAsync().ConfigureAwait(false);
+await Task.Delay(5000).ConfigureAwait(false);

Examples/Reconnect/Reconnect.csproj

@@ -14,7 +14,7 @@
 
   <!-- Update the version to match -->
   <ItemGroup>
-    <PackageReference Include="HiveMQtt" Version="0.4.0" />
+    <PackageReference Include="HiveMQtt" Version="0.4.1" />
   </ItemGroup>
 
 </Project>

Source/HiveMQtt/Client/HiveMQClient.cs

@@ -158,11 +158,11 @@ public async Task<bool> DisconnectAsync(DisconnectOptions? options = null)
             this.OnDisconnectSent -= eventHandler;
         }
 
-        this.connectState = ConnectState.Disconnected;
-
         // Close the socket
         this.CloseSocket();
 
+        this.connectState = ConnectState.Disconnected;
+
         // Clear the send queue
         this.sendQueue.Clear();
         return true;

Source/HiveMQtt/Client/HiveMQClientSocket.cs

@@ -137,8 +137,6 @@ internal async Task<bool> ConnectSocketAsync()
 
     internal bool CloseSocket(bool? shutdownPipeline = true)
     {
-        this.cancellationSource.Cancel();
-
         if (shutdownPipeline == true)
         {
             // Shutdown the pipeline
@@ -150,6 +148,8 @@ internal bool CloseSocket(bool? shutdownPipeline = true)
         this.socket?.Shutdown(SocketShutdown.Both);
         this.socket?.Close();
 
+        this.cancellationSource.Cancel();
+
         return true;
     }
 }

Source/HiveMQtt/Client/HiveMQClientTrafficProcessor.cs

@@ -353,6 +353,7 @@ private Task<bool> TrafficInflowProcessorAsync(CancellationToken cancellationTok
             switch (this.connectState)
             {
                 case ConnectState.Disconnecting:
+                case ConnectState.Disconnected:
                     // This disconnect was either user or broker initiated.
                     // Launch the AfterDisconnect event with a clean disconnect set to true.
                     this.AfterDisconnectEventLauncher(true);
@@ -364,8 +365,6 @@ private Task<bool> TrafficInflowProcessorAsync(CancellationToken cancellationTok
                     break;
                 case ConnectState.Connecting:
                     break;
-                case ConnectState.Disconnected:
-                    break;
                 default:
                     break;
             }