c# 批量注入示例代码

1. using Microsoft.Extensions.DependencyInjection;
2. using System;
3. using System.Linq;
4. using System.Reflection;
6. // 自定义属性来标记服务类型
7. [AttributeUsage(AttributeTargets.Class, AllowMultiple = false, Inherited = false)]
8. public class ServiceTypeAttribute : Attribute
9. {
10.     public ServiceLifetime Lifetime { get; }
12.     public ServiceTypeAttribute(ServiceLifetime lifetime)
13.     {
14.         Lifetime = lifetime;
15.     }
16. }
18. public static class DependencyInjectionExtensions
19. {
20.     /// <summary>
21.     /// 批量注册服务
22.     /// </summary>
23.     /// <param name="services">IServiceCollection</param>
24.     /// <param name="servicePrefix">服务类的前缀，例如 "Service"</param>
25.     /// <param name="interfacePrefix">接口的前缀，例如 "I"</param>
26.     public static void RegisterServices(this IServiceCollection services, string servicePrefix, string interfacePrefix)
27.     {
28.         if (services == null)
29.         {
30.             throw new ArgumentNullException(nameof(services));
31.         }
33.         // 获取当前程序集中所有类
34.         var types = Assembly.GetExecutingAssembly().GetTypes();
36.         // 筛选出符合服务命名约定的类
37.         var serviceTypes = types
38.             .Where(t => t.Name.StartsWith(servicePrefix) && !t.IsInterface && !t.IsAbstract)
39.             .ToList();
41.         // 遍历所有服务类
42.         foreach (var serviceType in serviceTypes)
43.         {
44.             // 尝试查找对应的接口
45.             var interfaceType = types.FirstOrDefault(t => t.Name == interfacePrefix + serviceType.Name.Substring(servicePrefix.Length) && t.IsInterface);
47.             if (interfaceType != null)
48.             {
49.                 // 获取服务类型上的 ServiceTypeAttribute
50.                 var serviceTypeAttribute = serviceType.GetCustomAttribute<ServiceTypeAttribute>();
51.                 var lifetime = serviceTypeAttribute?.Lifetime ?? ServiceLifetime.Scoped; // 默认使用 Scoped
53.                 // 根据不同的生命周期注册服务
54.                 switch (lifetime)
55.                 {
56.                     case ServiceLifetime.Singleton:
57.                         services.AddSingleton(interfaceType, serviceType);
58.                         break;
59.                     case ServiceLifetime.Scoped:
60.                         services.AddScoped(interfaceType, serviceType);
61.                         break;
62.                     case ServiceLifetime.Transient:
63.                         services.AddTransient(interfaceType, serviceType);
64.                         break;
65.                     default:
66.                         services.AddScoped(interfaceType, serviceType); // 默认使用 Scoped
67.                         break;
68.                 }
69.             }
70.             else
71.             {
72.                 // 如果没有找到对应的接口，则可以选择只注册服务本身，或者抛出异常
73.                 // 这里选择抛出异常，提示开发者需要有对应的接口
74.                 throw new InvalidOperationException($"Service {serviceType.Name} does not have a corresponding interface (expected interface name: {interfacePrefix}{serviceType.Name.Substring(servicePrefix.Length)})");
75.             }
76.         }
77.     }
78. }
80. // 示例接口和服务
81. public interface IUserService
82. {
83.     void DoSomething();
84. }
86. [ServiceType(ServiceLifetime.Scoped)] // 使用属性标记生命周期
87. public class UserService : IUserService
88. {
89.     public void DoSomething()
90.     {
91.         Console.WriteLine("UserService.DoSomething() called.");
92.     }
93. }
95. public interface IOrderService
96. {
97.      void ProcessOrder();
98. }
100. [ServiceType(ServiceLifetime.Transient)]
101. public class OrderService: IOrderService
102. {
103.     public void ProcessOrder()
104.     {
105.         Console.WriteLine("OrderService.ProcessOrder() called.");
106.     }
107. }
109. public class Program
110. {
111.     public static void Main(string[] args)
112.     {
113.         // 创建 ServiceCollection
114.         IServiceCollection services = new ServiceCollection();
116.         // 注册服务
117.         services.RegisterServices("Service", "I");
119.         // 添加其他必要的服务，例如 MVC
120.         services.AddMvc(); // 如果是 ASP.NET Core MVC 项目
122.         // 构建 ServiceProvider
123.         IServiceProvider serviceProvider = services.BuildServiceProvider();
125.         // 从 ServiceProvider 中获取服务并使用
126.         var userService = serviceProvider.GetService<IUserService>();
127.         userService?.DoSomething();
129.         var orderService = serviceProvider.GetService<IOrderService>();
130.         orderService?.ProcessOrder();
131.         Console.ReadKey();
132.     }
133. }

复制代码

代码说明

• ServiceTypeAttribute: 这是一个自定义属性，用于显式指定服务类的生命周期（Singleton、Scoped 或 Transient）。
  
• RegisterServices 扩展方法:
  
  
  
  • 它扩展了 IServiceCollection，提供了一个便捷的方法来批量注册服务。
    
  • servicePrefix 参数指定服务类的前缀（例如，"Service"）。
    
  • interfacePrefix 参数指定接口的前缀（例如，"I"）。
    
  • 它使用反射来查找程序集中所有符合命名约定的类（例如，以 "Service" 开头的类）。
    
  • 它假定接口的命名约定是接口前缀 + 服务类名（去掉服务前缀）。 例如，如果服务类是 UserService，则对应的接口应该是 IUserService。
    
  • 它使用 GetCustomAttribute 方法获取服务类上的 ServiceTypeAttribute 属性，并根据属性中指定的生命周期注册服务。 如果未找到此属性，则默认使用 Scoped 生命周期。
    
  • 如果找不到与服务类对应的接口，它会抛出一个异常，指示缺少必需的接口。
    
  
  
• 示例:
  
  
  
  • IUserService 和 UserService 演示了如何定义一个服务及其接口。
    
  • IOrderService 和 OrderService 演示了另一个服务及其接口，并使用[ServiceType] 特性标记了生命周期
    
  • 在 Main 方法中，我们创建了一个 ServiceCollection，调用 RegisterServices 方法注册服务，然后从 ServiceProvider 中解析并使用这些服务。
    
  
  

使用说明

• 定义您的服务接口和服务类，并确保它们遵循命名约定（例如，IUserService 和 UserService）。
  
• 使用 ServiceTypeAttribute 属性标记您的服务类，以指定其生命周期。 如果省略此属性，则默认使用 Scoped 生命周期。
  
• 在您的应用程序的启动配置中（例如，Main 方法或 ConfigureServices 方法中），获取 IServiceCollection 的实例，并调用 RegisterServices 扩展方法，传递服务类前缀和接口前缀。
  
• 像往常一样使用依赖注入来获取服务实例。
  

这个方法提供了一种灵活且类型安全的方式来批量注册服务，并允许您通过属性显式控制每个服务的生命周期。
　　

来源：程序园用户自行投稿发布，如果侵权，请联系站长删除
免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！