Make sure corresponding Connection and Channel token properly float into handlers

Author: danielmarbach

projects/RabbitMQ.Client/Impl/Channel.cs

@@ -201,7 +201,7 @@ protected void TakeOver(Channel other)
         public Task CloseAsync(ushort replyCode, string replyText, bool abort,
             CancellationToken cancellationToken)
         {
-            var args = new ShutdownEventArgs(ShutdownInitiator.Application, replyCode, replyText);
+            var args = new ShutdownEventArgs(ShutdownInitiator.Application, replyCode, replyText, cancellationToken: cancellationToken);
             return CloseAsync(args, abort, cancellationToken);
         }
 
@@ -850,7 +850,7 @@ await Session.Connection.HandleConnectionBlockedAsync(reason, cancellationToken)
         protected async Task<bool> HandleConnectionCloseAsync(IncomingCommand cmd, CancellationToken cancellationToken)
         {
             var method = new ConnectionClose(cmd.MethodSpan);
-            var reason = new ShutdownEventArgs(ShutdownInitiator.Peer, method._replyCode, method._replyText, method._classId, method._methodId);
+            var reason = new ShutdownEventArgs(ShutdownInitiator.Peer, method._replyCode, method._replyText, method._classId, method._methodId, cancellationToken: cancellationToken);
             try
             {
                 /*
@@ -895,7 +895,7 @@ protected async Task<bool> HandleConnectionStartAsync(IncomingCommand cmd, Cance
         {
             if (m_connectionStartCell is null)
             {
-                var reason = new ShutdownEventArgs(ShutdownInitiator.Library, Constants.CommandInvalid, "Unexpected Connection.Start");
+                var reason = new ShutdownEventArgs(ShutdownInitiator.Library, Constants.CommandInvalid, "Unexpected Connection.Start", cancellationToken: cancellationToken);
                 await Session.Connection.CloseAsync(reason, false,
                     InternalConstants.DefaultConnectionCloseTimeout,
                     cancellationToken)

projects/RabbitMQ.Client/Impl/Connection.Receive.cs

@@ -62,7 +62,8 @@ await ReceiveLoopAsync(mainLoopToken)
                 var ea = new ShutdownEventArgs(ShutdownInitiator.Library,
                     Constants.InternalError,
                     "Thread aborted (AppDomain unloaded?)",
-                    exception: taex);
+                    exception: taex,
+                    cancellationToken: mainLoopToken);
                 await HandleMainLoopExceptionAsync(ea)
                     .ConfigureAwait(false);
             }
@@ -73,7 +74,8 @@ await HandleMainLoopExceptionAsync(ea)
                 var ea = new ShutdownEventArgs(ShutdownInitiator.Library,
                     0,
                     "End of stream",
-                    exception: eose);
+                    exception: eose,
+                    cancellationToken: mainLoopToken);
                 await HandleMainLoopExceptionAsync(ea)
                     .ConfigureAwait(false);
             }
@@ -91,7 +93,8 @@ await HardProtocolExceptionHandlerAsync(hpe, mainLoopToken)
                 var ea = new ShutdownEventArgs(ShutdownInitiator.Library,
                     Constants.InternalError,
                     fileLoadException.Message,
-                    exception: fileLoadException);
+                    exception: fileLoadException,
+                    cancellationToken: mainLoopToken);
                 await HandleMainLoopExceptionAsync(ea)
                     .ConfigureAwait(false);
             }
@@ -106,7 +109,8 @@ await HandleMainLoopExceptionAsync(ea)
                     var ea = new ShutdownEventArgs(ShutdownInitiator.Library,
                         Constants.InternalError,
                         ocex.Message,
-                        exception: ocex);
+                        exception: ocex,
+                        cancellationToken: mainLoopToken);
                     await HandleMainLoopExceptionAsync(ea)
                         .ConfigureAwait(false);
                 }
@@ -116,7 +120,8 @@ await HandleMainLoopExceptionAsync(ea)
                 var ea = new ShutdownEventArgs(ShutdownInitiator.Library,
                     Constants.InternalError,
                     ex.Message,
-                    exception: ex);
+                    exception: ex,
+                    cancellationToken: mainLoopToken);
                 await HandleMainLoopExceptionAsync(ea)
                     .ConfigureAwait(false);
             }

projects/RabbitMQ.Client/Impl/Connection.cs

@@ -254,7 +254,7 @@ await _channel0.ConnectionOpenAsync(_config.VirtualHost, cancellationToken)
             {
                 try
                 {
-                    var ea = new ShutdownEventArgs(ShutdownInitiator.Library, Constants.InternalError, "FailedOpen");
+                    var ea = new ShutdownEventArgs(ShutdownInitiator.Library, Constants.InternalError, "FailedOpen", cancellationToken: cancellationToken);
                     await CloseAsync(ea, true,
                         InternalConstants.DefaultConnectionAbortTimeout,
                         cancellationToken).ConfigureAwait(false);
@@ -299,7 +299,7 @@ internal void EnsureIsOpen()
         public Task CloseAsync(ushort reasonCode, string reasonText, TimeSpan timeout, bool abort,
             CancellationToken cancellationToken = default)
         {
-            var reason = new ShutdownEventArgs(ShutdownInitiator.Application, reasonCode, reasonText);
+            var reason = new ShutdownEventArgs(ShutdownInitiator.Application, reasonCode, reasonText, cancellationToken: cancellationToken);
             return CloseAsync(reason, abort, timeout, cancellationToken);
         }