- A+
所属分类:.NET技术
思路
我发现 .NET Core WebAPi项目有一个与Springboot的不同之处,就是Springboot项目有自动装配机制,他可以将在src下面与启动类在同一级包目录下的类进行扫描注册
而之前我了解到Springboot的自动装配机制本质上也就是通过扫描对应包,然后进行通过它自身进行服务注册,我虽然写不出那么牛掰的东西,但是我也打算大致仿照一下
步骤如下:
- 准备几个装饰类Server、Config等
- 扫描当前程序集以及引用程序集被改装饰(注解)描述了的类和接口
- 这里要注意注册的顺序,注册的顺序如果出错,就会爆出异常(本人已经经历过)
- 同时还需要注意,我们读取配置注册配置类必须是在最前的,因为配置加载最好是在程序启动前全部加载完(不过想想也有懒加载的情况,这里我没有难么多考虑)
代码实现
我这里加载文件配置的代码和注册类的代码写到了一起,有点耦合,大家看可用适当的做拆分,OK希望我的这个小工具类可用帮助到大家,如果大家有更好的想法,希望写出来
using gobangBack.Annotation; using System; using System.Reflection; namespace gobangBack.Utils { // 这个扩展方法仅仅对于本项目 /// <summary> /// 步骤: /// 1.对于当前项目的Service目录下的所有带有Service结尾的类型进行父子关系判断 /// 2.进行服务类型和实现类型的一次性注入 /// </summary> public static class ServiceAutoDI { public static readonly Queue<Type> RegisterQueue = new Queue<Type>(); public static void AutoAllDI(this IServiceCollection service) { AutoServerDIFromService(service); } // 1.自动注册服务类 public static void AutoServerDIFromService(this IServiceCollection service) { // 1.获取当前程序集中有被Server修饰过的所有的类(并且不能是抽象类) var classTypes = AppDomain.CurrentDomain.GetAssemblies() .SelectMany(assembly => assembly.GetTypes()) .Where(type => type.IsClass && !type.IsAbstract) .Where(type => Attribute.IsDefined(type, typeof(Server))) .ToList(); // 2.获得当前程序集中被Server修饰过的所有接口 var interfaceTypes = AppDomain.CurrentDomain.GetAssemblies() .SelectMany(assembly => assembly.GetTypes()) .Where(type => type.IsInterface) .Where(type => Attribute.IsDefined(type, typeof(Server))) .ToList(); // 3.获取当前程序集中被Config修饰过的配置类 var configClassTypes = AppDomain.CurrentDomain.GetAssemblies() .SelectMany(assembly => assembly.GetTypes()) .Where(type => type.IsClass && !type.IsAbstract) .Where(type => Attribute.IsDefined(type, typeof(Config))) .ToList(); // 3.判断类型的构造方法中是否有被Server修饰的接口或者是被Config修饰的类 // 3.2获取到所有Server修饰类的构造函数中的所有参数 var quoteInterface = classTypes .SelectMany(classType => classType.GetConstructors()) .SelectMany(constructor => constructor.GetParameters()) // 4.2 判断参数中是否有interfaceTypes的值,如果有,就将其找出放入注入队列中 .Where(param => interfaceTypes.Contains(param.ParameterType)) .Select(param => param.ParameterType).ToList(); foreach (var interfaceType in quoteInterface) { RegisterQueue.Enqueue(interfaceType); interfaceTypes.Remove(interfaceType); } // 5.进行配置读取 // 5.1.先加载好配置文件 var configuration = new ConfigurationBuilder() .AddJsonFile("appsettings.json") .Build(); // 5.2.进行配置文件的读取,配置的名称必须和类名一致 configClassTypes.ForEach(config => { // 1.获取到配置类结尾的部分 string configName = config.FullName.Substring(config.FullName.LastIndexOf(".")+1); // 2.将配置类的配置添加到容器中 // 2.1.获取到OptionsConfigurationServiceCollectionExtensions类中名称为Configure参数为IServiceCollection,IConfiguration的方法 // 2.2.讲它的泛型类型设置config类型 MethodInfo configureMethod = typeof(OptionsConfigurationServiceCollectionExtensions) .GetMethod("Configure", new Type[] { typeof(IServiceCollection), typeof(IConfiguration) }) .MakeGenericMethod(config); // 3.读取配置文件中的名称为对应名称的部分,进行加载 configureMethod.Invoke(null, new object[] { service, configuration.GetSection(configName) }); // 4.注册配置类 service.AddScoped(config); }); // 6.优先注册在队列中的接口 var prioityInterface = RegisterQueue.ToArray(); foreach(var interfaceType in prioityInterface) { var implenmentTypes = classTypes.Where(classType => classType.IsAssignableTo(interfaceType)) .ToList(); implenmentTypes.ForEach(type => { service.AddScoped(interfaceType, type); }); } // 7.注册其它的接口类型 foreach(var interfaceType in interfaceTypes) { var implenmentTypes = classTypes.Where(classType => classType.IsAssignableTo(interfaceType)) .ToList(); implenmentTypes.ForEach(type => { service.AddScoped(interfaceType, type); }); } } } }