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

…queue (#1157)
graphql-dotnet · Oct 20, 2024 · 5f2dc82 · 5f2dc82
1 parent f5b7b32
commit 5f2dc82
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
@@ -227,7 +227,9 @@ public Task CloseAsync(int eventId, string? description)
     /// </remarks>
     public virtual 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;
     }