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

src/MassTransit/Monitoring/ConfigureBusHealthCheckServiceOptions.cs

58 строк2 KB

Chris Patterson

Massive file cleanup

27 мар 2024, 02:43

27 мар 2024, 02:432ac69e8

namespace MassTransit.Monitoring
{
    using System;
    using System.Collections.Generic;
    using System.Linq;
    using System.Reflection;
    using Configuration;
    using Microsoft.Extensions.Diagnostics.HealthChecks;
    using Microsoft.Extensions.Options;
    using Transports;


    public class ConfigureBusHealthCheckServiceOptions :
        IConfigureOptions<HealthCheckServiceOptions>
    {
        readonly IEnumerable<IBusInstance> _busInstances;
        readonly IServiceProvider _provider;
        readonly string[] _tags;

        public ConfigureBusHealthCheckServiceOptions(IEnumerable<IBusInstance> busInstances, IServiceProvider provider)
        {
            _busInstances = busInstances;
            _provider = provider;
            _tags = new[] { "ready", "masstransit" };
        }

        public void Configure(HealthCheckServiceOptions options)
        {
            foreach (var busInstance in _busInstances)
            {
                var type = typeof(MassTransitHealthCheckOptions<>).MakeGenericType(busInstance.InstanceType);
                var optionsType = typeof(IOptions<>).MakeGenericType(type);

                var name = busInstance.Name;
                HealthStatus? minimalFailureStatus = HealthStatus.Unhealthy;
                var tags = new HashSet<string>(_tags, StringComparer.OrdinalIgnoreCase);

                var busOptions = _provider.GetService(optionsType);
                if (busOptions != null)
                {
                    var healthCheckOptions = (IHealthCheckOptions)optionsType.GetProperty("Value", BindingFlags.Instance | BindingFlags.Public)
                        .GetValue(busOptions, null);

                    if (!string.IsNullOrWhiteSpace(healthCheckOptions.Name))
                        name = healthCheckOptions.Name;

                    if (healthCheckOptions.MinimalFailureStatus.HasValue)
                        minimalFailureStatus = healthCheckOptions.MinimalFailureStatus.Value;

                    if (healthCheckOptions.Tags.Any())
                        tags = healthCheckOptions.Tags;
                }

                options.Registrations.Add(new HealthCheckRegistration(name, new BusHealthCheck(busInstance), minimalFailureStatus, tags));
            }
        }
    }
}