Discard messages sent after the close message has been posted to the queue (#79)

Shane32 · web-flow · commit 39f8dd828167 · 2024-10-27T09:52:32.000-04:00
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;
     }
 

Original file line number	Diff line number	Diff line change
`@@ -169,7 +169,9 @@ public Task CloseAsync(int eventId, string? description)`
`169`	`169`	`/// <inheritdoc/>`
`170`	`170`	`public Task SendMessageAsync(OperationMessage message)`
`171`	`171`	`{`
`172`		`- _pump.Post(new Message { OperationMessage = message });`
	`172`	`+ // Messages posted after requesting the connection be closed will be discarded.`
	`173`	`+ if (!_closeRequested)`
	`174`	`+ _pump.Post(new Message { OperationMessage = message });`
`173`	`175`	`return Task.CompletedTask;`
`174`	`176`	`}`
`175`	`177`