using System.Reflection; using Microsoft.Extensions.DependencyInjection; namespace Common; /// /// 服务依赖注入扩展 /// public static class ServiceExtension { /// /// 自动扫描并注册服务和仓储 /// public static IServiceCollection AddServices(this IServiceCollection services) { // 扫描程序集 var serviceAssembly = Assembly.Load("Service"); var repositoryAssembly = Assembly.Load("Repository"); // 注册所有服务实现 RegisterServices(services, serviceAssembly); // 注册所有仓储实现 RegisterRepositories(services, repositoryAssembly); return services; } private static void RegisterServices(IServiceCollection services, Assembly assembly) { var types = assembly.GetTypes() .Where(t => t.IsClass && !t.IsAbstract); foreach (var type in types) { var interfaces = type.GetInterfaces() .Where(i => i.Name.StartsWith("I") && i.Namespace != null && i.Namespace.StartsWith("Service")); foreach (var @interface in interfaces) { // EmailBackgroundService 必须是 Singleton(后台服务),其他服务可用 Transient if (type.Name == "EmailBackgroundService") { services.AddSingleton(@interface, type); } else if (type.Name == "EmailFetchService") { // EmailFetchService 用 Transient,避免连接冲突 services.AddTransient(@interface, type); } else { services.AddSingleton(@interface, type); } } } } private static void RegisterRepositories(IServiceCollection services, Assembly assembly) { var types = assembly.GetTypes() .Where(t => t.IsClass && !t.IsAbstract); foreach (var type in types) { var interfaces = type.GetInterfaces() .Where(i => i.Name.StartsWith("I") && i.Namespace == "Repository" && !i.IsGenericType); // 排除泛型接口如 IBaseRepository foreach (var @interface in interfaces) { services.AddSingleton(@interface, type); Console.WriteLine($"注册 Repository: {@interface.Name} -> {type.Name}"); } } } }