public async Task DisableMethodsAsync(TimeSpan timeout)
{
Logging.Enter(this, timeout, nameof(DisableMethodsAsync));
Debug.Assert(_methodSendingLink != null);
Debug.Assert(_methodReceivingLink != null);
bool enteredSemaphore = await _methodLinkSemaphore.WaitAsync(timeout).ConfigureAwait(false);
if (!enteredSemaphore)
{
throw new TimeoutException("Failed to enter the semaphore required for ensuring that AMQP method sender and receiver links are closed.");
}
try
{
ICollection <Task> tasks = new List <Task>();
if (_methodReceivingLink != null)
{
tasks.Add(_methodReceivingLink.CloseAsync(timeout));
}
if (_methodSendingLink != null)
{
tasks.Add(_methodSendingLink.CloseAsync(timeout));
}
if (tasks.Count > 0)
{
await Task.WhenAll(tasks).ConfigureAwait(false);
_methodReceivingLink = null;
_methodSendingLink = null;
}
}
finally
{
Logging.Exit(this, timeout, nameof(DisableMethodsAsync));
_methodLinkSemaphore.Release();
}
}
public async Task DisableMethodsAsync(CancellationToken cancellationToken)
{
if (Logging.IsEnabled)
{
Logging.Enter(this, nameof(DisableMethodsAsync));
}
Debug.Assert(_methodSendingLink != null);
Debug.Assert(_methodReceivingLink != null);
try
{
await _methodLinkSemaphore.WaitAsync(cancellationToken).ConfigureAwait(false);
}
catch (OperationCanceledException)
{
throw new TimeoutException("Failed to enter the semaphore required for ensuring that AMQP method sender and receiver links are closed.");
}
// These event handlers are in place for network drop cases and will try to close the session that this
// link belongs to, but that isn't necessary when the client is deliberately closing just the link.
if (_methodReceiverLinkDisconnected != null)
{
_methodReceivingLink.Closed -= _methodReceiverLinkDisconnected;
}
if (_methodSenderLinkDisconnected != null)
{
_methodSendingLink.Closed -= _methodSenderLinkDisconnected;
}
try
{
ICollection <Task> tasks = new List <Task>(2);
if (_methodReceivingLink != null)
{
tasks.Add(_methodReceivingLink.CloseAsync(cancellationToken));
}
if (_methodSendingLink != null)
{
tasks.Add(_methodSendingLink.CloseAsync(cancellationToken));
}
if (tasks.Count > 0)
{
await Task.WhenAll(tasks).ConfigureAwait(false);
_methodReceivingLink = null;
_methodSendingLink = null;
}
}
finally
{
if (Logging.IsEnabled)
{
Logging.Exit(this, nameof(DisableMethodsAsync));
}
_methodLinkSemaphore.Release();
}
}