70 lines
3.0 KiB
C#
70 lines
3.0 KiB
C#
|
using LY.App.Extensions.DI;
|
|||
|
|
using System.Reflection;
|
|||
|
|
|
|||
|
|
namespace LY.App.Extensions
|
|||
|
|
{
|
|||
|
|
public static class AutoDIExtensions
|
|||
|
|
{
|
|||
|
|
/// <summary>
|
|||
|
|
///
|
|||
|
|
/// </summary>
|
|||
|
|
/// <param name="serviceCollection"></param>
|
|||
|
|
/// <returns></returns>
|
|||
|
|
public static IServiceCollection ServicesAutoInjectionExtension(this IServiceCollection serviceCollection)
|
|||
|
|
{
|
|||
|
|
var directory = AppDomain.CurrentDomain.BaseDirectory;
|
|||
|
|
var types = Directory.GetFiles(directory, "*.dll", SearchOption.TopDirectoryOnly)
|
|||
|
|
.Where(s => !s.ToLower().Contains("sql"))
|
|||
|
|
.Select(Assembly.LoadFrom)
|
|||
|
|
.SelectMany(a => a.GetTypes());
|
|||
|
|
Injection(serviceCollection, types);
|
|||
|
|
return serviceCollection;
|
|||
|
|
}
|
|||
|
|
/// <summary>
|
|||
|
|
/// 服务自动注入
|
|||
|
|
/// </summary>
|
|||
|
|
/// <param name="serviceCollection">需要自动注入服务的服务集合</param>
|
|||
|
|
/// <param name="selector">应用于每个Assembly的筛选函数</param>
|
|||
|
|
/// <exception cref="ArgumentOutOfRangeException">指定的注入类型不在可注入的范围内</exception>
|
|||
|
|
/// <exception cref="NoImplementationInterfaceException">指定注入的类型未实现任何服务</exception>
|
|||
|
|
/// <exception cref="ArgumentException">输入的参数错误：1、注入的类型未实现指定的服务。2、指定的服务不是Interface类型</exception>
|
|||
|
|
/// <returns>自动注入服务后的服务集合</returns>
|
|||
|
|
public static IServiceCollection ServicesAutoInjection(this IServiceCollection serviceCollection, Func<Assembly, bool> selector)
|
|||
|
|
{
|
|||
|
|
var directory = AppDomain.CurrentDomain.BaseDirectory;
|
|||
|
|
var types = Directory.GetFiles(directory, "*.dll", SearchOption.TopDirectoryOnly)
|
|||
|
|
.Select(Assembly.LoadFrom)
|
|||
|
|
.Where(selector)
|
|||
|
|
.SelectMany(a => a.GetTypes());
|
|||
|
|
|
|||
|
|
Injection(serviceCollection, types);
|
|||
|
|
|
|||
|
|
return serviceCollection;
|
|||
|
|
}
|
|||
|
|
// 注入逻辑
|
|||
|
|
private static void Injection(IServiceCollection serviceCollection, IEnumerable<Type> types)
|
|||
|
|
{
|
|||
|
|
foreach (var type in types)
|
|||
|
|
{
|
|||
|
|
var attribute = type.GetCustomAttribute<ServiceInjectionAttribute>();
|
|||
|
|
if (attribute == null) continue;
|
|||
|
|
switch (attribute.InjectionType)
|
|||
|
|
{
|
|||
|
|
case InjectionType.Transient:
|
|||
|
|
serviceCollection.AddTransient(type);
|
|||
|
|
break;
|
|||
|
|
case InjectionType.Scoped:
|
|||
|
|
serviceCollection.AddScoped(type);
|
|||
|
|
break;
|
|||
|
|
case InjectionType.Singleton:
|
|||
|
|
serviceCollection.AddSingleton(type);
|
|||
|
|
break;
|
|||
|
|
default:
|
|||
|
|
throw new ArgumentOutOfRangeException();
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|