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

Shane32 · web-flow · commit 5f2dc823041d · 2024-10-20T14:13:57.000-04:00
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;
     }
 

Original file line number	Diff line number	Diff line change
`@@ -227,7 +227,9 @@ public Task CloseAsync(int eventId, string? description)`
`227`	`227`	`/// </remarks>`
`228`	`228`	`public virtual Task SendMessageAsync(OperationMessage message)`
`229`	`229`	`{`
`230`		`- _pump.Post(new Message { OperationMessage = message });`
	`230`	`+ // Messages posted after requesting the connection be closed will be discarded.`
	`231`	`+ if (!_closeRequested)`
	`232`	`+ _pump.Post(new Message { OperationMessage = message });`
`231`	`233`	`return Task.CompletedTask;`
`232`	`234`	`}`
`233`	`235`