C# 当前项目自动服务DI类
思路
我发现 .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);
});
}
}
}
}