IOC批量注入再Core框架中还是比较麻烦的,因此写了一个简单的IOC注入通过属性标注服务,再通过service自带的注册服务,扩展了三个注入服务,分别为
AddServiceInjectTransientSetup/AddServiceInjectScopedSetup/AddServiceInjectSingletonSetup下面直接上代码
ServiceInjectSetup类
public static class ServiceInjectSetup
{
public static void AddServiceInjectTransientSetup(this IServiceCollection services, System.Reflection.Assembly assembly)
{
if (services == null)
{
throw new ArgumentNullException(nameof(services));
}
foreach (Type item in assembly.GetTypes())
{
Transient attribute = item.GetCustomAttribute(typeof(Transient)) as Transient;
if (attribute != null)
{
services.AddTransient(attribute.ServiceType, item);
}
}
}
public static void AddServiceInjectScopedSetup(this IServiceCollection services, System.Reflection.Assembly assembly)
{
if (services == null)
{
throw new ArgumentNullException(nameof(services));
}
foreach (Type item in assembly.GetTypes())
{
Scoped attribute = item.GetCustomAttribute(typeof(Scoped)) as Scoped;
if (attribute != null)
{
services.AddScoped(attribute.ServiceType, item);
}
}
}
public static void AddServiceInjectSingletonSetup(this IServiceCollection services, System.Reflection.Assembly assembly)
{
if (services == null)
{
throw new ArgumentNullException(nameof(services));
}
foreach (Type item in assembly.GetTypes())
{
Singleton attribute = item.GetCustomAttribute(typeof(Singleton)) as Singleton;
if (attribute != null)
{
services.AddSingleton(attribute.ServiceType, item);
}
}
}
}
自定义属性
public class Transient : Attribute
{
public Type ServiceType { get; set; }
public Transient(Type serviceType)
{
ServiceType = serviceType;
}
}
public class Singleton : Attribute
{
public Type ServiceType { get; set; }
public Singleton(Type serviceType)
{
ServiceType = serviceType;
}
}
public class Scoped : Attribute
{
public Type ServiceType { get; set; }
public Scoped(Type serviceType)
{
ServiceType = serviceType;
}
}
再startup类调用服务
services.AddServiceInjectScopedSetup(this.GetType().Assembly);
services.AddServiceInjectTransientSetup(this.GetType().Assembly);
services.AddServiceInjectSingletonSetup(this.GetType().Assembly);
标注服务
[Scoped(serviceType: typeof(ItestRepository))]
public class TestServices : ItestRepository
{
public string test()
{
return "张三";
}
}
再控制器中使用即可,有很多种注入方式,我比较喜欢的这种方式
|
请发表评论