76 lines
3.0 KiB
C#
76 lines
3.0 KiB
C#
using System;
|
||
using System.Reflection;
|
||
using Microsoft.Extensions.DependencyInjection;
|
||
using msdi = Microsoft.Extensions.DependencyInjection;
|
||
using System.Linq;
|
||
using System.Collections.Generic;
|
||
|
||
namespace Falcon.DI
|
||
{
|
||
/// <summary>
|
||
/// 服务集合方法扩展
|
||
/// </summary>
|
||
public static class IServiceCollectionExtend
|
||
{
|
||
/// <summary>
|
||
/// 实现自动DI注册
|
||
/// </summary>
|
||
/// <param name="services">服务集合</param>
|
||
/// <param name="assemblies">要注册的程序集集合，如果未指定则为CurrentDomain所有引入的程序集</param>
|
||
public static IServiceCollection UseFalconDI(this IServiceCollection services,params Assembly[] assemblies) {
|
||
if(assemblies == null || assemblies.Length == 0) {
|
||
return services.UseFalconDI(AppDomain.CurrentDomain.GetAssemblies());
|
||
}
|
||
foreach(Assembly ass in assemblies) {
|
||
foreach(Type type in ass.GetTypes()) {
|
||
if(!type.CanInstance()) {
|
||
continue;
|
||
}
|
||
var ra = type.GetCustomAttribute<FalconDIRegisterAttribute>(true);
|
||
if(ra != null) {
|
||
//如果未提供服务类型，注册到所有实现的接口和基类
|
||
if(ra.ServiceTypes == null || ra.ServiceTypes.Length == 0) {
|
||
var sts = new List<Type>();
|
||
//添加所有实现的接口
|
||
sts.AddRange(type.GetInterfaces());
|
||
//添加非object基类
|
||
var baseType = type.BaseType;
|
||
if(baseType != typeof(object)) {
|
||
sts.Add(baseType);
|
||
}
|
||
//添加类型本身
|
||
sts.Add(type);
|
||
|
||
ra.ServiceTypes = sts.ToArray();
|
||
}
|
||
foreach(var ser in ra.ServiceTypes) {
|
||
services.Add(new ServiceDescriptor(ser,type,convertLifetime(ra.Lifetime)));
|
||
}
|
||
}
|
||
}
|
||
}
|
||
return services;
|
||
}
|
||
/// <summary>
|
||
/// 转换生存期枚举
|
||
/// </summary>
|
||
/// <param name="lt">生存期</param>
|
||
/// <returns></returns>
|
||
private static msdi.ServiceLifetime convertLifetime(ServiceLifetime lt) {
|
||
return (msdi.ServiceLifetime)(int)lt;
|
||
}
|
||
|
||
/// <summary>
|
||
/// 获取类型是否可以实例化。
|
||
/// </summary>
|
||
/// <param name="type">类型</param>
|
||
/// <returns>是否可实例化</returns>
|
||
public static bool CanInstance(this Type type) {
|
||
if(!type.IsClass || type.IsAbstract) {
|
||
return false;
|
||
}
|
||
return true;
|
||
}
|
||
}
|
||
}
|