ASP.NET CORE 學習之原生DI實現批量注冊

以前使用Autofac的時候，只需一句AsImplementInterfaces()就可以很輕松實現批量注冊功能。而asp.net core內置的DI框架沒有現成的批量注冊方法，考慮到替換Autofac框架過程有些繁瑣，於是自己寫擴展實現了一個簡易的原生DI批量注冊功能

Startup.cs擴展

public static class StartUpExtenions
    {
        /// <summary>
        /// 批量注冊服務
        /// </summary>
        /// <param name="services">DI服務</param>
        /// <param name="assemblys">需要批量注冊的程序集集合</param>
        /// <param name="baseType">基礎類/接口</param>
        /// <param name="serviceLifetime">服務生命周期</param>
        /// <returns></returns>
        public static IServiceCollection BatchRegisterService(this IServiceCollection services, Assembly[] assemblys, Type baseType, ServiceLifetime serviceLifetime = ServiceLifetime.Singleton)
        {
            List<Type> typeList = new List<Type>();  //所有符合注冊條件的類集合
            foreach (var assembly in assemblys)
            {
                //篩選當前程序集下符合條件的類
                var types = assembly.GetTypes().Where(t => !t.IsInterface && !t.IsSealed && !t.IsAbstract && baseType.IsAssignableFrom(t));
                if (types != null && types.Count() > 0)
                    typeList.AddRange(types);
            }
            if (typeList.Count() == 0)
                return services;

            var typeDic = new Dictionary<Type, Type[]>(); //待注冊集合
            foreach (var type in typeList)
            {
                var interfaces = type.GetInterfaces();   //獲取接口
                typeDic.Add(type, interfaces);
            }
            if (typeDic.Keys.Count() > 0)
            {
                foreach (var instanceType in typeDic.Keys)
                {
                    foreach (var interfaceType in typeDic[instanceType])
                    {
                        //根據指定的生命周期進行注冊
                        switch (serviceLifetime)
                        {
                            case ServiceLifetime.Scoped:
                                services.AddScoped(interfaceType, instanceType);
                                break;
                            case ServiceLifetime.Singleton:
                                services.AddSingleton(interfaceType, instanceType);
                                break;
                            case ServiceLifetime.Transient:
                                services.AddTransient(interfaceType, instanceType);
                                break;
                        }
                    }
                }
            }
            return services;
        }
    }

 

在ConfigureServices方法中調用批量注冊

 services.BatchRegisterService(new Assembly[] { Assembly.GetExecutingAssembly(), Assembly.Load("Test.DAL") }, typeof(IDependency)); 

 

經測試 ，使用擴展批量注冊的方式注冊的服務類正常工作