From 8c6e061bda3d493d9f0990b9e35688f06d213d12 Mon Sep 17 00:00:00 2001
From: Daniel Marbach <danielmarbach@users.noreply.github.com>
Date: Tue, 30 Jul 2024 15:17:55 +0200
Subject: [PATCH] Support cancellation on the flow control block

---
 projects/RabbitMQ.Client/client/impl/ChannelBase.cs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/projects/RabbitMQ.Client/client/impl/ChannelBase.cs b/projects/RabbitMQ.Client/client/impl/ChannelBase.cs
index f1b75c191..979b0285e 100644
--- 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);