src/MassTransit/Monitoring/BusHealthCheck.cs - develop - vshmidt/masstransit

src/MassTransit/Monitoring/BusHealthCheck.cs

83 строки3 KB

Dmitry Golubenkov

Fix: MassTransit ignores IHealthCheckOptionsConfigurator.FailureStatus value despite allowing it to be set (#4788)

28 ноя 2023, 04:38

28 ноя 2023, 04:38b32548f

namespace MassTransit.Monitoring
{
    using System;
    using System.Collections.Generic;
    using System.Linq;
    using System.Threading;
    using System.Threading.Tasks;
    using Microsoft.Extensions.Diagnostics.HealthChecks;
    using Transports;


    public class BusHealthCheck :
        IHealthCheck
    {
        readonly IBusInstance _busInstance;

        public BusHealthCheck(IBusInstance busInstance)
        {
            _busInstance = busInstance;
        }

        public Task<HealthCheckResult> CheckHealthAsync(HealthCheckContext context, CancellationToken cancellationToken)
        {
            var result = _busInstance.BusControl.CheckHealth();

            var data = new Dictionary<string, object>
            {
                ["Endpoints"] = new EndpointDictionary(result.Endpoints.ToDictionary(x => x.Key,
                    x => new Endpoint(Enum.GetName(typeof(BusHealthStatus), x.Value.Status), x.Value.Description)
                ))
            };

            var minimalHealthcheckLevel = context.Registration.FailureStatus switch
            {
                HealthStatus.Healthy => BusHealthStatus.Healthy,
                HealthStatus.Degraded => BusHealthStatus.Degraded,
                _ => BusHealthStatus.Unhealthy
            };

            var usedHealthcheckResult = result.Status < minimalHealthcheckLevel ? minimalHealthcheckLevel : result.Status;

            return Task.FromResult(usedHealthcheckResult switch
            {
                BusHealthStatus.Healthy => HealthCheckResult.Healthy(result.Description, data),
                BusHealthStatus.Degraded => HealthCheckResult.Degraded(result.Description, result.Exception, data),
                _ => HealthCheckResult.Unhealthy(result.Description, result.Exception, data)
            });
        }


        class EndpointDictionary :
            Dictionary<string, Endpoint>
        {
            public EndpointDictionary(IDictionary<string, Endpoint> dictionary)
                : base(dictionary, StringComparer.OrdinalIgnoreCase)
            {
            }

            public override string ToString()
            {
                return string.Join(", ", this.Select(x => $"{x.Key}: {x.Value}"));
            }
        }


        class Endpoint
        {
            public Endpoint(string status, string description)
            {
                Status = status;
                Description = description;
            }

            public string Status { get; set; }
            public string Description { get; set; }

            public override string ToString()
            {
                return $"{Status} - {Description}";
            }
        }
    }
}