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

禹翔 · 发表于 2023-12-19 10:42:13

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

// 接口1
public interface IService1
{
void PerformService1();
}
// 接口2
public interface IService2
{
void PerformService2();
}
// 实现类1，实现IService1
public class MyService1 : IService1
{
public void PerformService1()
{
Console.WriteLine("Service 1 performed.");
}
}
// 实现类2，实现IService2
[CustomRegistration] // 带有自定义特性
public class MyService2 : IService2
{
public void PerformService2()
{
Console.WriteLine("Service 2 performed.");
}
}
// 实现类3，实现IService1和IService2
public class MyService3 : IService1, IService2
{
public void PerformService1()
{
Console.WriteLine("Service 3 (Service 1 part) performed.");
}
public void PerformService2()
{
Console.WriteLine("Service 3 (Service 2 part) performed.");
}
}

复制代码

步骤2：创建自定义特性

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

复制代码

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

using System;
using System.Linq;
using System.Reflection;
using Microsoft.Extensions.DependencyInjection;
class Program
{
static void Main()
{
// 创建服务集合
var services = new ServiceCollection();
// 步骤4：注册指定类
services.AddTransient<MyService1>();
// 步骤5：注册带有自定义特性的类
RegisterClassesWithAttribute<CustomRegistrationAttribute>(services);
// 步骤6：注册项目下所有带有接口实现的类（项目下的所有接口）
RegisterAllImplementationsOfInterfaces(services);
// 构建服务提供程序
var serviceProvider = services.BuildServiceProvider();
// 步骤7：使用注册的服务
var myService1 = serviceProvider.GetService<MyService1>();
myService1.PerformService1();
var myService2 = serviceProvider.GetService<MyService2>();
myService2.PerformService2();
var myService3 = serviceProvider.GetService<MyService3>();
myService3.PerformService1();
myService3.PerformService2();
}
// 自动注册带有指定特性的类
static void RegisterClassesWithAttribute<TAttribute>(IServiceCollection services)
where TAttribute : Attribute
{
// 获取当前程序集
var assembly = Assembly.GetExecutingAssembly();
// 获取带有指定特性的所有类
var attributedTypes = assembly.GetTypes()
.Where(type => type.GetCustomAttributes(typeof(TAttribute), true).Any() && type.IsClass);
// 注册这些类
foreach (var attributedType in attributedTypes)
{
services.AddTransient(attributedType);
}
}
// 自动注册项目下所有带有接口实现的类（项目下的所有接口）
static void RegisterAllImplementationsOfInterfaces(IServiceCollection services)
{
// 获取当前程序集
var assembly = Assembly.GetExecutingAssembly();
// 获取项目下所有接口
var interfaceTypes = assembly.GetTypes()
.Where(type => type.IsInterface);
// 获取实现了这些接口的所有类
var implementationTypes = assembly.GetTypes()
.Where(type => interfaceTypes.Any(interfaceType => interfaceType.IsAssignableFrom(type)) && type.IsClass);
// 注册这些类
foreach (var implementationType in implementationTypes)
{
services.AddTransient(implementationType);
}
}
}

复制代码

在上述代码中：

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

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

来源:https://www.cnblogs.com/hanbing81868164/archive/2023/12/19/17912799.html
免责声明：由于采集信息均来自互联网，如果侵犯了您的权益，请联系我们【E-Mail:cb@itdo.tech】我们会及时删除侵权内容，谢谢合作！

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

本帖子中包含更多资源