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

开发 后端
在.NET中,进行依赖注入(DI)的自动注册,可以通过反射机制和程序集扫描来实现。以下是详细的步骤以及相应的C#源代码示例,包括注册指定类、注册带有自定义特性的类、以及注册项目下所有带有接口实现的类(项目下的所有接口)。

概述:.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相关的包。这是一个基本的示例,实际应用中可能需要更复杂的配置,具体取决于项目的需求。

责任编辑:姜华 来源: 今日头条
相关推荐

2009-11-12 10:32:47

ADO.NET技术

2009-11-12 10:53:57

ADO.NET连接My

2023-06-28 08:16:50

Autofac应用程序

2010-01-13 17:47:59

VB.NET拖放

2010-01-14 13:59:01

2010-01-14 10:07:08

VB.NET文件名排序

2010-01-18 19:36:52

VB.NET调整控件

2010-10-22 11:31:53

SQL Server自

2009-11-24 15:34:41

DNS服务器组建

2010-01-11 18:40:03

VB.NET操作注册表

2018-03-12 10:02:30

PHP依赖注入

2024-04-15 07:00:00

Python开发Hatch

2009-02-16 15:35:00

2024-02-26 00:04:00

代码zip()开发

2022-12-29 08:54:53

依赖注入JavaScript

2011-05-31 10:00:21

Android Spring 依赖注入

2024-01-02 08:22:01

Koin框架项目

2022-09-16 08:04:25

阿里云权限网络

2019-12-27 15:16:34

设计人工智能开发

2017-05-11 15:01:43

Androidweb布局
点赞
收藏

51CTO技术栈公众号