精准掌握.NET依赖注入：DI自动注册服务轻松搞定

 
概述：.NET依赖注入（DI）通过反射自动注册服务，示例展示了注册指定类、带特性类、项目下所有接口实现的类。简化配置，提高可维护性。
在.NET中，进行依赖注入（DI）的自动注册，可以通过反射机制和程序集扫描来实现。以下是详细的步骤以及相应的C#源代码示例，包括注册指定类、注册带有自定义特性的类、以及注册项目下所有带有接口实现的类（项目下的所有接口）：
步骤1：创建接口和实现类

1. // 接口1
2. public interface IService1
3. {
4.     void PerformService1();
5. }
7. // 接口2
8. public interface IService2
9. {
10.     void PerformService2();
11. }
13. // 实现类1，实现IService1
14. public class MyService1 : IService1
15. {
16.     public void PerformService1()
17.     {
18.         Console.WriteLine("Service 1 performed.");
19.     }
20. }
22. // 实现类2，实现IService2
23. [CustomRegistration] // 带有自定义特性
24. public class MyService2 : IService2
25. {
26.     public void PerformService2()
27.     {
28.         Console.WriteLine("Service 2 performed.");
29.     }
30. }
32. // 实现类3，实现IService1和IService2
33. public class MyService3 : IService1, IService2
34. {
35.     public void PerformService1()
36.     {
37.         Console.WriteLine("Service 3 (Service 1 part) performed.");
38.     }
40.     public void PerformService2()
41.     {
42.         Console.WriteLine("Service 3 (Service 2 part) performed.");
43.     }
44. }

复制代码

步骤2：创建自定义特性

1. // 自定义特性
2. [AttributeUsage(AttributeTargets.Class, Inherited = false, AllowMultiple = false)]
3. sealed class CustomRegistrationAttribute : Attribute
4. {
5. }

复制代码

步骤3：创建自动注册方法

1. using System;
2. using System.Linq;
3. using System.Reflection;
4. using Microsoft.Extensions.DependencyInjection;
6. class Program
7. {
8.     static void Main()
9.     {
10.         // 创建服务集合
11.         var services = new ServiceCollection();
13.         // 步骤4：注册指定类
14.         services.AddTransient<MyService1>();
16.         // 步骤5：注册带有自定义特性的类
17.         RegisterClassesWithAttribute<CustomRegistrationAttribute>(services);
19.         // 步骤6：注册项目下所有带有接口实现的类（项目下的所有接口）
20.         RegisterAllImplementationsOfInterfaces(services);
22.         // 构建服务提供程序
23.         var serviceProvider = services.BuildServiceProvider();
25.         // 步骤7：使用注册的服务
26.         var myService1 = serviceProvider.GetService<MyService1>();
27.         myService1.PerformService1();
29.         var myService2 = serviceProvider.GetService<MyService2>();
30.         myService2.PerformService2();
32.         var myService3 = serviceProvider.GetService<MyService3>();
33.         myService3.PerformService1();
34.         myService3.PerformService2();
35.     }
37.     // 自动注册带有指定特性的类
38.     static void RegisterClassesWithAttribute<TAttribute>(IServiceCollection services)
39.         where TAttribute : Attribute
40.     {
41.         // 获取当前程序集
42.         var assembly = Assembly.GetExecutingAssembly();
44.         // 获取带有指定特性的所有类
45.         var attributedTypes = assembly.GetTypes()
46.             .Where(type => type.GetCustomAttributes(typeof(TAttribute), true).Any() && type.IsClass);
48.         // 注册这些类
49.         foreach (var attributedType in attributedTypes)
50.         {
51.             services.AddTransient(attributedType);
52.         }
53.     }
55.     // 自动注册项目下所有带有接口实现的类（项目下的所有接口）
56.     static void RegisterAllImplementationsOfInterfaces(IServiceCollection services)
57.     {
58.         // 获取当前程序集
59.         var assembly = Assembly.GetExecutingAssembly();
61.         // 获取项目下所有接口
62.         var interfaceTypes = assembly.GetTypes()
63.             .Where(type => type.IsInterface);
65.         // 获取实现了这些接口的所有类
66.         var implementationTypes = assembly.GetTypes()
67.             .Where(type => interfaceTypes.Any(interfaceType => interfaceType.IsAssignableFrom(type)) && type.IsClass);
69.         // 注册这些类
70.         foreach (var implementationType in implementationTypes)
71.         {
72.             services.AddTransient(implementationType);
73.         }
74.     }
75. }

复制代码

在上述代码中：

• 使用AddTransient方法注册了特定的MyService1类。
  
• 使用RegisterClassesWithAttribute方法注册了带有CustomRegistrationAttribute特性的类。这里使用了反射机制来动态获取所有带有指定特性的类的类型，并将它们注册到DI容器中。
  
• 使用RegisterAllImplementationsOfInterfaces方法注册了项目下所有实现接口的类。
  

请确保在项目中引用了
Microsoft.Extensions.DependencyInjection相关的包。这是一个基本的示例，实际应用中可能需要更复杂的配置，具体取决于项目的需求。
 


免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！