QYZH.InteractiveMagazine/QYZH.InteractiveMagazine.Infrastructure/Autofac/AutofacExtension.cs

using Autofac;
using Autofac.Extensions.DependencyInjection;
using Microsoft.AspNetCore.Builder;
using Microsoft.Extensions.Configuration;
using QYZH.InteractiveMagazine.Infrastructure.Cache;
using QYZH.InteractiveMagazine.Infrastructure.MessageQueue;
using QYZH.InteractiveMagazine.Models.Settings;
using RabbitMQ.Client;
using StackExchange.Redis;

namespace QYZH.InteractiveMagazine.Infrastructure.Autofacs
{
    public static class AutofacExtension
    {

        /// <summary>
        /// 根据程序集名称获取程序集
        /// </summary>
        /// <param name="builder">Web应用程序构建器</param>
        public static void UseAutofac(this WebApplicationBuilder builder)
        {
            builder.Host.UseServiceProviderFactory(new AutofacServiceProviderFactory())
                .ConfigureContainer<ContainerBuilder>((c, containerBuilder) =>
                {
                    var friendlyName = AppDomain.CurrentDomain.FriendlyName;
                    var source = friendlyName.Split('.');
                    var assemblyNames = string.Join(".", source.Take(source.Length - 1));
                    containerBuilder.RegisterModule(new AutofacModuleRegister(assemblyNames));

                    InitializeRedis(c.Configuration);
                    InitializeRabbitMQ(c.Configuration, containerBuilder);
                });
        }

        private static void InitializeRedis(IConfiguration configuration)
        {
            var redisSettings = configuration.GetSection("RedisSettings").Get<RedisSettings>();
            if (redisSettings != null && !string.IsNullOrWhiteSpace(redisSettings.ConnectionString))
            {
                var multiplexer = ConnectionMultiplexer.Connect(redisSettings.ConnectionString);
                RedisHelper.Connection = multiplexer;
                RedisHelper.SetKeyPrefix(redisSettings.InstanceName ?? string.Empty);
            }
        }

        private static void InitializeRabbitMQ(IConfiguration configuration, ContainerBuilder containerBuilder)
        {
            var rabbitMQSettings = configuration.GetSection("RabbitMQSettings").Get<RabbitMQSettings>();
            if (rabbitMQSettings != null && !string.IsNullOrWhiteSpace(rabbitMQSettings.HostName))
            {
                var factory = new ConnectionFactory
                {
                    HostName = rabbitMQSettings.HostName,
                    Port = rabbitMQSettings.Port,
                    UserName = rabbitMQSettings.UserName ?? string.Empty,
                    Password = rabbitMQSettings.Password ?? string.Empty,
                    VirtualHost = rabbitMQSettings.VirtualHost ?? string.Empty
                };

                var connection = factory.CreateConnectionAsync().GetAwaiter().GetResult();

                containerBuilder.RegisterInstance(connection).As<IConnection>().SingleInstance();
                containerBuilder.RegisterType<RabbitMQPublisher>().InstancePerDependency();
            }
        }
    }
}