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

…queue
graphql-dotnet · Oct 7, 2024 · 6f7d462 · 6f7d462
1 parent a84c553
commit 6f7d462
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/Transports.AspNetCore/WebSockets/WebSocketConnection.cs b/src/Transports.AspNetCore/WebSockets/WebSocketConnection.cs
@@ -220,7 +220,9 @@ public Task CloseAsync(int eventId, string? description)
     /// </remarks>
     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;
     }