private async Task OpenMethodsSenderLinkAsync(AmqpIotSession amqpIotSession, string correlationIdSuffix, CancellationToken cancellationToken) { if (_methodSendingLink == null || _methodSendingLink.IsClosing()) { _methodSendingLink?.SafeClose(); _methodSendingLink = await amqpIotSession .OpenMethodsSenderLinkAsync(_deviceIdentity, correlationIdSuffix, cancellationToken) .ConfigureAwait(false); if (_methodSenderLinkDisconnected == null) { _methodSenderLinkDisconnected = (obj, arg) => { amqpIotSession.SafeClose(); }; } _methodSendingLink.Closed += _methodSenderLinkDisconnected; if (Logging.IsEnabled) { Logging.Associate(this, _methodSendingLink, nameof(_methodSendingLink)); } } }
private async Task OpenMethodsSenderLinkAsync(AmqpIotSession amqpIotSession, string correlationIdSuffix, TimeSpan timeout) { if (_methodSendingLink == null || _methodSendingLink.IsClosing()) { _methodSendingLink?.SafeClose(); _methodSendingLink = await amqpIotSession.OpenMethodsSenderLinkAsync(_deviceIdentity, correlationIdSuffix, timeout).ConfigureAwait(false); _methodSendingLink.Closed += (obj, arg) => { amqpIotSession.SafeClose(); }; Logging.Associate(this, _methodSendingLink, nameof(_methodSendingLink)); } }
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(); } }
internal async Task <AmqpIotSession> EnsureSessionIsOpenAsync(TimeSpan timeout) { if (_closed) { throw new IotHubException("Device is now offline.", false); } Logging.Enter(this, timeout, nameof(EnsureSessionIsOpenAsync)); bool enteredSemaphore = await _sessionSemaphore.WaitAsync(timeout).ConfigureAwait(false); if (!enteredSemaphore) { throw new TimeoutException("Failed to enter the semaphore required for opening an AMQP session."); } try { if (_amqpIotSession == null || _amqpIotSession.IsClosing()) { _amqpIotSession?.SafeClose(); _amqpIotSession = await _amqpConnectionHolder.OpenSessionAsync(_deviceIdentity, timeout).ConfigureAwait(false); Logging.Associate(this, _amqpIotSession, nameof(_amqpIotSession)); if (_deviceIdentity.AuthenticationModel == AuthenticationModel.SasIndividual) { _amqpAuthenticationRefresher = await _amqpConnectionHolder.CreateRefresherAsync(_deviceIdentity, timeout).ConfigureAwait(false); Logging.Associate(this, _amqpAuthenticationRefresher, nameof(_amqpAuthenticationRefresher)); } _amqpIotSession.Closed += OnSessionDisconnected; _messageSendingLink = await _amqpIotSession.OpenTelemetrySenderLinkAsync(_deviceIdentity, timeout).ConfigureAwait(false); _messageSendingLink.Closed += (obj, arg) => { _amqpIotSession.SafeClose(); }; Logging.Associate(this, _messageSendingLink, nameof(_messageSendingLink)); } if (_disposed) { throw new IotHubException("Device is now offline.", false); } } catch (Exception) { Cleanup(); throw; } finally { _sessionSemaphore.Release(); } Logging.Exit(this, timeout, nameof(EnsureSessionIsOpenAsync)); return(_amqpIotSession); }
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(); } }