110 lines
4.5 KiB
C#
110 lines
4.5 KiB
C#
|
using Microsoft.Extensions.Configuration;
|
|||
|
|
using System.Reflection;
|
|||
|
|
using YY.Admin.Core;
|
|||
|
|
using YY.Admin.Services.Service.Auth;
|
|||
|
|
|
|||
|
|
namespace YY.Admin
|
|||
|
|
{
|
|||
|
|
public static class ServiceExtensions
|
|||
|
|
{
|
|||
|
|
/// <summary>
|
|||
|
|
/// 注册服务
|
|||
|
|
/// </summary>
|
|||
|
|
public static void AddService(this IContainerRegistry containerRegistry, IConfiguration configuration)
|
|||
|
|
{
|
|||
|
|
// 注册配置
|
|||
|
|
containerRegistry.RegisterInstance<IConfiguration>(configuration);
|
|||
|
|
//// 注册日志服务为单例
|
|||
|
|
containerRegistry.RegisterSingleton<ILoggerService, SerilogLoggerService>();
|
|||
|
|
// 自动扫描并注册应用层服务
|
|||
|
|
RegisterServicesByAssembly(containerRegistry, typeof(ISysAuthService).Assembly);
|
|||
|
|
}
|
|||
|
|
/// <summary>
|
|||
|
|
/// 自动注册程序集中所有服务
|
|||
|
|
/// </summary>
|
|||
|
|
private static void RegisterServicesByAssembly(IContainerRegistry containerRegistry, Assembly assembly)
|
|||
|
|
{
|
|||
|
|
// 获取所有公开的非抽象类
|
|||
|
|
var serviceTypes = assembly.GetExportedTypes()
|
|||
|
|
.Where(t => t.IsClass && !t.IsAbstract);
|
|||
|
|
|
|||
|
|
foreach (var implementationType in serviceTypes)
|
|||
|
|
{
|
|||
|
|
// 查找服务接口（名称以"I"开头，去掉首字母后匹配）
|
|||
|
|
var serviceInterface = implementationType.GetInterfaces()
|
|||
|
|
.FirstOrDefault(i => i.Name == $"I{implementationType.Name}");
|
|||
|
|
|
|||
|
|
// 如果没有直接匹配的接口，尝试查找其他接口
|
|||
|
|
if (serviceInterface == null)
|
|||
|
|
{
|
|||
|
|
// 备选方案1：注册所有实现的接口（适合多接口实现）
|
|||
|
|
RegisterAllInterfaces(containerRegistry, implementationType);
|
|||
|
|
|
|||
|
|
// 备选方案2：只注册主接口
|
|||
|
|
// serviceInterface = implementationType.GetInterface($"I{implementationType.Name}");
|
|||
|
|
}
|
|||
|
|
else
|
|||
|
|
{
|
|||
|
|
// 注册找到的接口
|
|||
|
|
RegisterService(containerRegistry, serviceInterface, implementationType);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// 注册服务到容器
|
|||
|
|
/// </summary>
|
|||
|
|
private static void RegisterService(
|
|||
|
|
IContainerRegistry containerRegistry,
|
|||
|
|
Type serviceType,
|
|||
|
|
Type implementationType)
|
|||
|
|
{
|
|||
|
|
// 根据命名约定判断生命周期
|
|||
|
|
bool isSingleton = implementationType.Name.EndsWith("Service") ||
|
|||
|
|
implementationType.Name.EndsWith("Repository");
|
|||
|
|
|
|||
|
|
// 根据基类判断生命周期
|
|||
|
|
bool isSingletonByBase = typeof(ISingletonDependency).IsAssignableFrom(implementationType);
|
|||
|
|
|
|||
|
|
// 根据特性判断生命周期
|
|||
|
|
var attribute = implementationType.GetCustomAttribute<LifecycleAttribute>();
|
|||
|
|
var lifecycle = attribute?.Lifecycle ?? Lifecycle.Transient;
|
|||
|
|
|
|||
|
|
// 确定注册方式
|
|||
|
|
if (isSingleton || isSingletonByBase || lifecycle == Lifecycle.Singleton)
|
|||
|
|
{
|
|||
|
|
containerRegistry.RegisterSingleton(serviceType, implementationType);
|
|||
|
|
Console.WriteLine($"注册单例服务: {serviceType.Name} -> {implementationType.Name}");
|
|||
|
|
}
|
|||
|
|
else
|
|||
|
|
{
|
|||
|
|
containerRegistry.Register(serviceType, implementationType);
|
|||
|
|
Console.WriteLine($"注册瞬时服务: {serviceType.Name} -> {implementationType.Name}");
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// 注册所有实现的接口
|
|||
|
|
/// </summary>
|
|||
|
|
private static void RegisterAllInterfaces(IContainerRegistry containerRegistry, Type implementationType)
|
|||
|
|
{
|
|||
|
|
var interfaces = implementationType.GetInterfaces()
|
|||
|
|
.Where(i => i != typeof(IDisposable) &&
|
|||
|
|
!i.Name.StartsWith("_") && // 排除某些特殊接口
|
|||
|
|
i.Assembly == implementationType.Assembly); // 只注册同程序集接口
|
|||
|
|
|
|||
|
|
foreach (var serviceType in interfaces)
|
|||
|
|
{
|
|||
|
|
RegisterService(containerRegistry, serviceType, implementationType);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if (!interfaces.Any())
|
|||
|
|
{
|
|||
|
|
// 如果没有接口，直接注册具体类型
|
|||
|
|
containerRegistry.Register(implementationType);
|
|||
|
|
Console.WriteLine($"注册具体类型: {implementationType.Name}");
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|