WebSocket: Catch the cancellation of the loop in iterator.

Gremlinq · Mar 20, 2024 · 3b709bd · 3b709bd
1 parent 05482c2
commit 3b709bd
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/Providers.Core/Factory/WebSocketGremlinqClientFactory.cs b/src/Providers.Core/Factory/WebSocketGremlinqClientFactory.cs
@@ -335,6 +335,10 @@ private async Task Loop(CancellationToken ct)
                             {
                                 bytes = await _client.ReceiveAsync(ct);
                             }
+                            catch (OperationCanceledException)
+                            {
+                                return;
+                            }
                             catch (InvalidOperationException)
                             {
                                 return;