Forgot to upload latest changes to the code to prevent making an unne…

…cessary websocket.
radu-matei · Apr 19, 2018 · 1f712e6 · 1f712e6
1 parent 0fb9b2a
commit 1f712e6
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/src/WebSocketManager.Client/Connection.cs b/src/WebSocketManager.Client/Connection.cs
@@ -27,16 +27,15 @@ public class Connection
 
         public Connection()
         {
-            _clientWebSocket = new ClientWebSocket();
         }
 
         public async Task StartConnectionAsync(string uri)
         {
-            // the connection was lost, probably why we got called again.
-            if (_clientWebSocket.State != WebSocketState.Open)
+            // also check if connection was lost, that's probably why we get called multiple times.
+            if (_clientWebSocket == null || _clientWebSocket.State != WebSocketState.Open)
             {
                 // create a new web-socket so the next connect call works.
-                _clientWebSocket.Dispose();
+                _clientWebSocket?.Dispose();
                 _clientWebSocket = new ClientWebSocket();
             }
             // don't do anything, we are already connected.