src/MassTransit/Configuration/JobServiceRegistrationExtensions.cs - develop - vshmidt/masstransit

src/MassTransit/Configuration/JobServiceRegistrationExtensions.cs
91 строка4 KB
Chris Patterson
Added the IsJobServiceEndpoint extension method for use in AddConfigureEndpointsCallback to avoid adding filters to the job service endpoints
19 май 2025, 23:04
19 май 2025, 23:043031bb7
#nullable enable
namespace MassTransit
{
    using System;
    using Configuration;
    using DependencyInjection.Registration;
    using Microsoft.Extensions.DependencyInjection;
    using Microsoft.Extensions.DependencyInjection.Extensions;


    public static class JobServiceRegistrationExtensions
    {
        /// <summary>
        /// Set the job consumer options (optional, not required to use job consumers)
        /// </summary>
        /// <param name="configurator"></param>
        /// <param name="configure">Configure the job consumer options using this callback</param>
        /// <returns></returns>
        public static IJobServiceRegistrationConfigurator SetJobConsumerOptions(this IBusRegistrationConfigurator configurator,
            Action<JobConsumerOptions>? configure = null)
        {
            var registration = configurator.RegisterJobService(configurator.Registrar);

            registration.AddConfigureAction(configure);

            var registrationConfigurator = new JobServiceRegistrationConfigurator(configurator, registration);

            return registrationConfigurator;
        }

        /// <summary>
        /// Add registrations for the job service saga state machines
        /// </summary>
        /// <param name="configurator"></param>
        /// <param name="configure">Configure the job saga options</param>
        public static IJobSagaRegistrationConfigurator AddJobSagaStateMachines(this IBusRegistrationConfigurator configurator,
            Action<JobSagaOptions>? configure = null)
        {
            var registrationConfigurator = new JobSagaRegistrationConfigurator(configurator, configure);

            return registrationConfigurator;
        }

        /// <summary>
        /// Register a custom job distribution strategy for the job saga state machines
        /// </summary>
        /// <param name="services"></param>
        /// <typeparam name="T"></typeparam>
        /// <returns></returns>
        public static IServiceCollection TryAddJobDistributionStrategy<T>(this IServiceCollection services)
            where T : class, IJobDistributionStrategy
        {
            services.TryAddScoped<IJobDistributionStrategy, T>();

            return services;
        }

        /// <summary>
        /// Compares the <paramref name="name" /> to the known job saga endpoints and returns true if the name matches.
        /// Use this inside an AddConfigureEndpointsCallback to avoid adding filters to the job saga endpoints.
        /// </summary>
        /// <param name="context">The registration context</param>
        /// <param name="name">The endpoint name</param>
        /// <returns>true if matched, otherwise false</returns>
        public static bool IsJobServiceEndpoint(this IRegistrationContext context, string name)
        {
            var selector = context.GetRequiredService<IContainerSelector>();

            var formatter = selector.GetEndpointNameFormatter(context);

            IEndpointDefinition? endpointDefinition = selector.GetEndpointDefinition<JobSaga>(context);
            if (string.Equals(endpointDefinition?.GetEndpointName(formatter), name, StringComparison.OrdinalIgnoreCase))
                return true;

            endpointDefinition = selector.GetEndpointDefinition<JobTypeSaga>(context);
            if (string.Equals(endpointDefinition?.GetEndpointName(formatter), name, StringComparison.OrdinalIgnoreCase))
                return true;

            endpointDefinition = selector.GetEndpointDefinition<JobAttemptSaga>(context);
            if (string.Equals(endpointDefinition?.GetEndpointName(formatter), name, StringComparison.OrdinalIgnoreCase))
                return true;

            if (string.Equals(formatter.Saga<JobTypeSaga>(), name, StringComparison.OrdinalIgnoreCase)
                || string.Equals(formatter.Saga<JobSaga>(), name, StringComparison.OrdinalIgnoreCase)
                || string.Equals(formatter.Saga<JobAttemptSaga>(), name, StringComparison.OrdinalIgnoreCase))
                return true;

            return false;
        }
    }
}