src/MassTransit.Abstractions/Configuration/Middleware/DelegateConfigurationExtensions.cs - develop - vshmidt/masstransit

src/MassTransit.Abstractions/Configuration/Middleware/DelegateConfigurationExtensions.cs

44 строки2 KB

Chris Patterson

Integration of all external packages into MassTransit, split out into Abstractions, Middleware, and the core MassTransit assembly.

22 янв 2022, 18:24

22 янв 2022, 18:24b5b4f50

namespace MassTransit
{
    using System;
    using System.Threading.Tasks;
    using Configuration;


    public static class DelegateConfigurationExtensions
    {
        /// <summary>
        /// Executes a synchronous method on the pipe
        /// </summary>
        /// <typeparam name="TContext">The context type</typeparam>
        /// <param name="configurator">The pipe configurator</param>
        /// <param name="callback">The callback to invoke</param>
        public static void UseExecute<TContext>(this IPipeConfigurator<TContext> configurator, Action<TContext> callback)
            where TContext : class, PipeContext
        {
            if (configurator == null)
                throw new ArgumentNullException(nameof(configurator));

            var pipeBuilderConfigurator = new DelegatePipeSpecification<TContext>(callback);

            configurator.AddPipeSpecification(pipeBuilderConfigurator);
        }

        /// <summary>
        /// Executes an asynchronous method on the pipe
        /// </summary>
        /// <typeparam name="TContext">The context type</typeparam>
        /// <param name="configurator">The pipe configurator</param>
        /// <param name="callback">The callback to invoke</param>
        public static void UseExecuteAsync<TContext>(this IPipeConfigurator<TContext> configurator, Func<TContext, Task> callback)
            where TContext : class, PipeContext
        {
            if (configurator == null)
                throw new ArgumentNullException(nameof(configurator));

            var pipeBuilderConfigurator = new AsyncDelegatePipeSpecification<TContext>(callback);

            configurator.AddPipeSpecification(pipeBuilderConfigurator);
        }
    }
}