src/MassTransit.Abstractions/Configuration/DependencyInjection/FutureDefinition.cs - develop - vshmidt/masstransit

src/MassTransit.Abstractions/Configuration/DependencyInjection/FutureDefinition.cs
103 строки4 KB
Chris Patterson
Massive file cleanup
27 мар 2024, 02:43
27 мар 2024, 02:432ac69e8
namespace MassTransit
{
    using System;
    using Configuration;


    /// <summary>
    /// A future definition defines the configuration for a future, which can be used by the automatic registration code to
    /// configure the consumer on a receive endpoint.
    /// </summary>
    /// <typeparam name="TFuture"></typeparam>
    public class FutureDefinition<TFuture> :
        IFutureDefinition<TFuture>
        where TFuture : class, SagaStateMachine<FutureState>
    {
        int? _concurrentMessageLimit;
        string? _endpointName;

        protected FutureDefinition()
        {
        }

        /// <summary>
        /// Specify the endpoint name (which may be a queue, or a subscription, depending upon the transport) on which the saga
        /// should be configured.
        /// </summary>
        protected string EndpointName
        {
            set => _endpointName = value;
        }

        public IEndpointDefinition<TFuture>? EndpointDefinition { get; set; }

        IEndpointDefinition? IFutureDefinition.EndpointDefinition => EndpointDefinition;

        /// <summary>
        /// Set the concurrent message limit for the saga, which limits how many saga instances are able to concurrently
        /// consume messages.
        /// </summary>
        public int? ConcurrentMessageLimit
        {
            get => _concurrentMessageLimit;
            protected set => _concurrentMessageLimit = value;
        }

        void IFutureDefinition<TFuture>.Configure(IReceiveEndpointConfigurator endpointConfigurator, ISagaConfigurator<FutureState> sagaConfigurator,
            IRegistrationContext context)
        {
            if (_concurrentMessageLimit.HasValue)
                sagaConfigurator.ConcurrentMessageLimit = _concurrentMessageLimit;

            #pragma warning disable CS0618
            ConfigureSaga(endpointConfigurator, sagaConfigurator);
            #pragma warning restore CS0618
            ConfigureSaga(endpointConfigurator, sagaConfigurator, context);
        }

        Type IFutureDefinition.FutureType => typeof(TFuture);

        string IFutureDefinition.GetEndpointName(IEndpointNameFormatter formatter)
        {
            return string.IsNullOrWhiteSpace(_endpointName)
                ? _endpointName = EndpointDefinition?.GetEndpointName(formatter) ?? formatter.Message<TFuture>()
                : _endpointName!;
        }

        /// <summary>
        /// Called when configuring the saga on the endpoint. Configuration only applies to this saga, and does not apply to
        /// the endpoint.
        /// </summary>
        /// <param name="endpointConfigurator">The receive endpoint configurator for the consumer</param>
        /// <param name="sagaConfigurator">The saga configurator</param>
        [Obsolete("Use the IRegistrationContext overload instead. Visit https://masstransit.io/obsolete for details.")]
        protected virtual void ConfigureSaga(IReceiveEndpointConfigurator endpointConfigurator, ISagaConfigurator<FutureState> sagaConfigurator)
        {
        }

        /// <summary>
        /// Called when configuring the saga on the endpoint. Configuration only applies to this saga, and does not apply to
        /// the endpoint.
        /// </summary>
        /// <param name="endpointConfigurator">The receive endpoint configurator for the consumer</param>
        /// <param name="sagaConfigurator">The saga configurator</param>
        /// <param name="context"></param>
        protected virtual void ConfigureSaga(IReceiveEndpointConfigurator endpointConfigurator, ISagaConfigurator<FutureState> sagaConfigurator,
            IRegistrationContext context)
        {
        }

        /// <summary>
        /// Configure the saga endpoint
        /// </summary>
        /// <param name="configure"></param>
        protected void Endpoint(Action<IEndpointRegistrationConfigurator>? configure = null)
        {
            var configurator = new EndpointRegistrationConfigurator<TFuture>();

            configure?.Invoke(configurator);

            EndpointDefinition = new FutureEndpointDefinition<TFuture>(configurator.Settings);
        }
    }
}