feat: improve client

josephnhtam · Sep 15, 2024 · 8167bf9 · 8167bf9
1 parent 40fac01
commit 8167bf9
Showing 1 changed file with 10 additions and 8 deletions.
diff --git a/src/LiveStreamingServerNet.Networking/Internal/Client.cs b/src/LiveStreamingServerNet.Networking/Internal/Client.cs
@@ -77,16 +77,18 @@ public async Task RunAsync(CancellationToken stoppingToken)
             {
                 _logger.ClientLoopError(ClientId, ex);
             }
+            finally
+            {
+                linkedCts.Cancel();
 
-            linkedCts.Cancel();
-
-            await DisposeAsync(handler);
-            await DisposeAsync(_bufferSender);
-            await DisposeAsync(networkStream);
-            CloseTcpClient();
+                await DisposeAsync(handler);
+                await DisposeAsync(_bufferSender);
+                await DisposeAsync(networkStream);
+                CloseTcpClient();
 
-            _stoppedTcs.TrySetResult();
-            _logger.ClientDisconnected(ClientId);
+                _stoppedTcs.TrySetResult();
+                _logger.ClientDisconnected(ClientId);
+            }
         }
 
         private IClientHandler CreateClientHandler()