Addresss PR feedback

Xabaril · Dec 13, 2024 · c5b7238 · c5b7238
1 parent 57221a8
commit c5b7238
Showing 1 changed file with 1 addition and 13 deletions.
diff --git a/src/HealthChecks.Nats/NatsHealthCheck.cs b/src/HealthChecks.Nats/NatsHealthCheck.cs
@@ -8,21 +8,9 @@ namespace HealthChecks.Nats;
 /// </summary>
 public sealed class NatsHealthCheck(INatsConnection connection) : IHealthCheck
 {
-    public async Task<HealthCheckResult> CheckHealthAsync(HealthCheckContext context, CancellationToken cancellationToken = default)
-    {
+    public async Task<HealthCheckResult> CheckHealthAsync(HealthCheckContext context, CancellationToken cancellationToken = default) =>
         await TryConnectAsync(connection).ConfigureAwait(false);
 
-        var result = connection.ConnectionState switch
-        {
-            NatsConnectionState.Open => HealthCheckResult.Healthy(),
-            NatsConnectionState.Connecting or NatsConnectionState.Reconnecting => HealthCheckResult.Degraded(),
-            NatsConnectionState.Closed => await TryConnectAsync(connection).ConfigureAwait(false),
-            _ => new HealthCheckResult(context.Registration.FailureStatus)
-        };
-
-        return result;
-    }
-
     private static async Task<HealthCheckResult> TryConnectAsync(INatsConnection natsConnection)
     {
         try