fix: Actually wait for websocket tasks to complete

Otherwise we leak 500mb of tasks per second
buttplugio · Jun 25, 2023 · a909020 · a909020
1 parent 99a23fd
commit a909020
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/Buttplug.Client.Connectors.WebsocketConnector/ButtplugWebsocketConnector.cs b/Buttplug.Client.Connectors.WebsocketConnector/ButtplugWebsocketConnector.cs
@@ -136,8 +136,7 @@ private async Task RunClientLoop(CancellationToken token)
                         writeTask,
                     };
 
-                    var completedTaskIndex = Task.WhenAny(msgTasks);
-
+                    var _ = await Task.WhenAny(msgTasks);
                     if (readTask.IsCompleted)
                     {
                         var incomingMsg = await readTask.ConfigureAwait(false);
@@ -186,6 +185,10 @@ private async Task RunClientLoop(CancellationToken token)
                             throw new ButtplugClientConnectorException("Websocket Client Read Error", e);
                         }
                     }
+                    else if (writeTask.IsCanceled)
+                    {
+                        Console.WriteLine("Write cancelled");
+                    }
                 }
             }
             catch (Exception e)