Merge pull request #88 from wemogy/87-add-activity-for-tracing-for-az…

…ureservicebushealthcheck fix: #87 add activity to AzureServiceBusHealthCheck
wemogy · Jun 13, 2024 · a45aacc · a45aacc
2 parents 3e1b3c6 + 377298f
commit a45aacc
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/...ureServiceBus/Wemogy.CQRS.Extensions.AzureServiceBus/Health/AzureServiceBusHealthCheck.cs b/...ureServiceBus/Wemogy.CQRS.Extensions.AzureServiceBus/Health/AzureServiceBusHealthCheck.cs
@@ -4,6 +4,7 @@
 using System.Threading.Tasks;
 using Azure.Messaging.ServiceBus;
 using Microsoft.Extensions.Diagnostics.HealthChecks;
+using OpenTelemetry.Trace;
 
 namespace Wemogy.CQRS.Extensions.AzureServiceBus.Health
 {
@@ -30,6 +31,7 @@ public AzureServiceBusHealthCheck(string connectionString, string queueName)
 
         public async Task<HealthCheckResult> CheckHealthAsync(HealthCheckContext context, CancellationToken cancellationToken = default)
         {
+            using var activity = Observability.DefaultActivities.StartActivity();
             try
             {
                 var client = ClientConnections.GetOrAdd(_queueName, _ => new ServiceBusClient(_connectionString));
@@ -39,6 +41,8 @@ public async Task<HealthCheckResult> CheckHealthAsync(HealthCheckContext context
             }
             catch (TaskCanceledException ex)
             {
+                activity?.RecordException(ex);
+
                 // propagate the exception if the cancellation token has been canceled
                 if (cancellationToken.IsCancellationRequested)
                 {
@@ -49,6 +53,7 @@ public async Task<HealthCheckResult> CheckHealthAsync(HealthCheckContext context
             }
             catch (Exception ex)
             {
+                activity?.RecordException(ex);
                 return new HealthCheckResult(context.Registration.FailureStatus, exception: ex);
             }
         }