Discard messages sent after the close message has been posted to the …

…queue
Shane32 · Oct 27, 2024 · 02955ae · 02955ae
1 parent cbd7a33
commit 02955ae
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/GraphQL.AspNetCore3/WebSockets/WebSocketConnection.cs b/src/GraphQL.AspNetCore3/WebSockets/WebSocketConnection.cs
@@ -169,7 +169,9 @@ public Task CloseAsync(int eventId, string? description)
     /// <inheritdoc/>
     public Task SendMessageAsync(OperationMessage message)
     {
-        _pump.Post(new Message { OperationMessage = message });
+        // Messages posted after requesting the connection be closed will be discarded.
+        if (!_closeRequested)
+            _pump.Post(new Message { OperationMessage = message });
         return Task.CompletedTask;
     }