Support cancellation on the flow control block (#1643)

danielmarbach · web-flow · commit 36a16ba58ce1 · 2024-07-30T07:35:43.000-07:00
Co-authored-by: Daniel Marbach &lt;danielmarbach@users.noreply.github.com&gt;
diff --git a/projects/RabbitMQ.Client/client/impl/ChannelBase.cs b/projects/RabbitMQ.Client/client/impl/ChannelBase.cs
@@ -453,7 +453,7 @@ protected ValueTask ModelSendAsync<TMethod, THeader>(in TMethod method, in THead
         {
             if (!_flowControlBlock.IsSet)
             {
-                _flowControlBlock.Wait();
+                _flowControlBlock.Wait(cancellationToken);
             }
 
             return Session.TransmitAsync(in method, in header, body, cancellationToken);

Original file line number	Diff line number	Diff line change
`@@ -453,7 +453,7 @@ protected ValueTask ModelSendAsync<TMethod, THeader>(in TMethod method, in THead`
`453`	`453`	`{`
`454`	`454`	`if (!_flowControlBlock.IsSet)`
`455`	`455`	`{`
`456`		`- _flowControlBlock.Wait();`
	`456`	`+ _flowControlBlock.Wait(cancellationToken);`
`457`	`457`	`}`
`458`	`458`
`459`	`459`	`return Session.TransmitAsync(in method, in header, body, cancellationToken);`