src/Transports/MassTransit.Azure.ServiceBus.Core/Configuration/ServiceBusConfigurationExtensions.cs - develop - vshmidt/masstransit

src/Transports/MassTransit.Azure.ServiceBus.Core/Configuration/ServiceBusConfigurationExtensions.cs
75 строк4 KB
denys.kozhevnikov
Cleanup using
25 янв 2023, 22:00
25 янв 2023, 22:0015c3b8b
namespace MassTransit
{
    using System;
    using AzureServiceBusTransport;
    using DependencyInjection;
    using Microsoft.Extensions.DependencyInjection;
    using Microsoft.Extensions.DependencyInjection.Extensions;
    using Transports;


    public static class ServiceBusConfigurationExtensions
    {
        /// <summary>
        /// Configure and create a bus for Azure Service Bus
        /// </summary>
        /// <param name="selector">Hang off the selector interface for visibility</param>
        /// <param name="configure">The configuration callback to configure the bus</param>
        /// <returns></returns>
        public static IBusControl CreateUsingAzureServiceBus(this IBusFactorySelector selector, Action<IServiceBusBusFactoryConfigurator> configure)
        {
            return AzureBusFactory.CreateUsingServiceBus(configure);
        }

        /// <summary>
        /// Configure MassTransit to use Azure Service Bus for the transport.
        /// </summary>
        /// <param name="configurator">The registration configurator (configured via AddMassTransit)</param>
        /// <param name="configure">The configuration callback for the bus factory</param>
        public static void UsingAzureServiceBus(this IBusRegistrationConfigurator configurator,
            Action<IBusRegistrationContext, IServiceBusBusFactoryConfigurator> configure = null)
        {
            configurator.SetBusFactory(new ServiceBusRegistrationBusFactory(configure));

            configurator.TryAddSingleton(provider =>
            {
                var subscriptionEndpointConnector = provider.GetRequiredService<Bind<IBus, IBusInstance>>().Value as ISubscriptionEndpointConnector;

                return subscriptionEndpointConnector ?? throw new ConfigurationException("The default bus instance is not an Azure Service Bus Instance");
            });
        }

        /// <summary>
        /// Configure MassTransit to use Azure Service Bus for the transport.
        /// </summary>
        /// <param name="configurator">The registration configurator (configured via AddMassTransit)</param>
        /// <param name="configure">The configuration callback for the bus factory</param>
        public static void UsingAzureServiceBus<TBus>(this IBusRegistrationConfigurator<TBus> configurator,
            Action<IBusRegistrationContext, IServiceBusBusFactoryConfigurator> configure = null)
            where TBus : class, IBus
        {
            configurator.SetBusFactory(new ServiceBusRegistrationBusFactory(configure));

            AddSubscriptionEndpointConnector<TBus>(configurator);
        }

        static void AddSubscriptionEndpointConnector<TBus>(IServiceCollection services)
            where TBus : class, IBus
        {
            services.TryAddSingleton(provider =>
            {
                var subscriptionEndpointConnector = provider.GetRequiredService<IBusInstance<TBus>>().BusInstance as ISubscriptionEndpointConnector;

                return subscriptionEndpointConnector ?? throw new ConfigurationException("The default bus instance is not an Azure Service Bus Instance");
            });

            services.TryAddSingleton(provider =>
            {
                var subscriptionEndpointConnector = provider.GetRequiredService<IBusInstance<TBus>>().BusInstance as ISubscriptionEndpointConnector;

                return Bind<TBus>.Create(subscriptionEndpointConnector
                    ?? throw new ConfigurationException("The default bus instance is not an Azure Service Bus Instance"));
            });
        }
    }
}