diff --git a/src/ServiceControl.Transports.ASBS/DeadLetterQueueCheck.cs b/src/ServiceControl.Transports.ASBS/DeadLetterQueueCheck.cs index 8aeeaefd34..247b610d53 100644 --- a/src/ServiceControl.Transports.ASBS/DeadLetterQueueCheck.cs +++ b/src/ServiceControl.Transports.ASBS/DeadLetterQueueCheck.cs @@ -3,7 +3,6 @@ using System; using System.Threading; using System.Threading.Tasks; - using Azure.Messaging.ServiceBus.Administration; using Microsoft.Extensions.Logging; using NServiceBus.CustomChecks; @@ -18,9 +17,9 @@ public DeadLetterQueueCheck(TransportSettings settings, ILogger PerformCheck(CancellationToken cancellationToken = default) @@ -31,7 +30,7 @@ public override async Task PerformCheck(CancellationToken cancellat } logger.LogDebug("Checking Dead Letter Queue length"); - var managementClient = new ServiceBusAdministrationClient(connectionString); + var managementClient = authenticationMethod.BuildManagementClient(); var queueRuntimeInfo = await managementClient.GetQueueRuntimePropertiesAsync(stagingQueue, cancellationToken); var deadLetterMessageCount = queueRuntimeInfo.Value.DeadLetterMessageCount; @@ -47,9 +46,9 @@ public override async Task PerformCheck(CancellationToken cancellat return CheckResult.Pass; } - readonly string connectionString; readonly string stagingQueue; readonly bool runCheck; + readonly AuthenticationMethod authenticationMethod; readonly ILogger logger; } } \ No newline at end of file