Merge branch 'feat/remote-processing' of https://github.com/wemogy/li…

…bs-cqrs into feat/remote-processing
wemogy · Jun 25, 2024 · 989059c · 989059c
2 parents d4dbce6 + f059ae7
commit 989059c
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);
             }
         }