public async Task <AmqpIoTConnection> EnsureConnection(TimeSpan timeout)
{
if (Logging.IsEnabled)
{
Logging.Enter(this, timeout, $"{nameof(EnsureConnection)}");
}
AmqpIoTConnection amqpIoTConnection = null;
IAmqpIoTAuthenticationRefresher amqpAuthenticationRefresher = null;
AmqpIoTCbsLink amqpIoTCbsLink = null;
bool gain = await _lock.WaitAsync(timeout).ConfigureAwait(false);
if (!gain)
{
throw new TimeoutException();
}
try
{
if (_amqpIoTConnection == null)
{
if (Logging.IsEnabled)
{
Logging.Info(this, "Creating new AmqpConnection", $"{nameof(EnsureConnection)}");
}
// Create AmqpConnection
amqpIoTConnection = await _amqpIoTConnector.OpenConnectionAsync(timeout).ConfigureAwait(false);
if (_deviceIdentity.AuthenticationModel != AuthenticationModel.X509)
{
if (_amqpIoTCbsLink == null)
{
if (Logging.IsEnabled)
{
Logging.Info(this, "Creating new AmqpCbsLink", $"{nameof(EnsureConnection)}");
}
amqpIoTCbsLink = amqpIoTConnection.CreateCbsLink(_deviceIdentity, timeout);
}
else
{
amqpIoTCbsLink = _amqpIoTCbsLink;
}
if (_deviceIdentity.AuthenticationModel == AuthenticationModel.SasGrouped)
{
if (Logging.IsEnabled)
{
Logging.Info(this, "Creating connection width AmqpAuthenticationRefresher", $"{nameof(EnsureConnection)}");
}
amqpAuthenticationRefresher = new AmqpAuthenticationRefresher(_deviceIdentity, amqpIoTCbsLink);
await amqpAuthenticationRefresher.InitLoopAsync(timeout).ConfigureAwait(false);
}
}
_amqpIoTConnection = amqpIoTConnection;
_amqpIoTCbsLink = amqpIoTCbsLink;
_amqpAuthenticationRefresher = amqpAuthenticationRefresher;
_amqpIoTConnection.Closed += OnConnectionClosed;
if (Logging.IsEnabled)
{
Logging.Associate(this, _amqpIoTConnection, $"{nameof(_amqpIoTConnection)}");
}
if (Logging.IsEnabled)
{
Logging.Associate(this, _amqpIoTCbsLink, $"{nameof(_amqpIoTCbsLink)}");
}
}
else if (_amqpIoTConnection.IsClosing())
{
throw new IotHubCommunicationException("AMQP connection is closing.");
}
else
{
amqpIoTConnection = _amqpIoTConnection;
}
}
catch (Exception ex) when(!ex.IsFatal())
{
amqpIoTCbsLink?.Close();
amqpAuthenticationRefresher?.StopLoop();
amqpIoTConnection?.SafeClose();
throw;
}
finally
{
_lock.Release();
}
if (Logging.IsEnabled)
{
Logging.Exit(this, timeout, $"{nameof(EnsureConnection)}");
}
return(amqpIoTConnection);
}
