程序员的知识教程库

网站首页 > 教程分享 正文

C#从零开始实现一个特性的自动注入功能

henian88 2025-03-19 12:51:38 教程分享 36 ℃ 0 评论

在现代软件开发中,依赖注入(Dependency Injection, DI)是实现松耦合、模块化和可测试代码的一个重要实践。C# 提供了优秀的 DI 容器,如ASP.NET Core中自带的
Microsoft.Extensions.DependencyInjection,但是有时候我们希望自定义一些功能,比如通过特性(Attributes)实现自动注入。本文将详细介绍如何从零开始实现一个通过特性自动注入的功能。

引言

特性(Attributes)是C#提供的一种用于在运行时通过反射向代码中添加声明性信息的方式。通过结合反射和特性,我们可以实现一种自动依赖注入的功能,这样可以简化DI的配置。

Nuget安装

Microsoft.Extensions.DependencyInjection

创建特性

首先,我们需要创建一个特性,用于标记需要注入的类。

using System;

[AttributeUsage(AttributeTargets.Class, Inherited = false)]
public sealed class InjectAttribute : Attribute
{
    public ServiceLifetime Lifetime { get; }

    public InjectAttribute(ServiceLifetime lifetime = ServiceLifetime.Transient)
    {
        Lifetime = lifetime;
    }
}

在这个示例中,InjectAttribute特性有一个可选参数Lifetime,用于指示依赖项的生命周期(如瞬时、作用域内或单例)。

实现自动注入

接下来,我们需要编写代码扫描程序集中的所有类型,并根据标记的特性注入依赖项。

using Microsoft.Extensions.DependencyInjection;
using System;
using System.Linq;
using System.Reflection;

public static class ServiceCollectionExtensions
{
    public static void AddServicesWithInjectAttribute(this IServiceCollection services, params Assembly[] assemblies)
    {
        var typesWithInject = assemblies
            .SelectMany(assembly => assembly.GetTypes())
            .Where(type => type.GetCustomAttribute() != null);

        foreach (var type in typesWithInject)
        {
            var injectAttribute = type.GetCustomAttribute();
            switch (injectAttribute.Lifetime)
            {
                case ServiceLifetime.Singleton:
                    services.AddSingleton(type);
                    break;
                case ServiceLifetime.Scoped:
                    services.AddScoped(type);
                    break;
                case ServiceLifetime.Transient:
                    services.AddTransient(type);
                    break;
            }
        }
    }
}

在这个扩展方法中,
AddServicesWithInjectAttribute会扫描传入的程序集,并根据InjectAttribute将相应的类型添加到DI容器中。

完整示例

为了完整演示如何使用上述功能,我们将创建一些示例代码。

using Microsoft.Extensions.DependencyInjection;
using System.Reflection;
using System;

namespace DependencyInjectionExample
{
    // 定义一个服务接口
    public interface IMyService
    {
        void DoWork();
    }

    // 实现服务并标记 InjectAttribute
    [Inject(ServiceLifetime.Singleton)]
    public class MyService : IMyService
    {
        private Guid id { get; set; }
        public MyService() {
            id = Guid.NewGuid();
        }
    
        public void DoWork()
        {
            Console.WriteLine("Service is working."+this.id);
        }
    }

    // 创建应用程序入口
    class Program
    {
        static void Main(string[] args)
        {
            // 创建服务集合
            var serviceCollection = new ServiceCollection();

            // 自动注入服务
            serviceCollection.AddServicesWithInjectAttribute(Assembly.GetExecutingAssembly());

            // 构建服务提供者
            var serviceProvider = serviceCollection.BuildServiceProvider();

            // 解析服务并调用方法,指定使用 MyService1
            var myService1 = serviceProvider.GetService();
            myService1.DoWork();
            
            var myService2 = serviceProvider.GetService();
            myService2.DoWork();
        }
    }
}

在这个示例中,我们定义了IMyService接口和其实现MyService,并使用[Inject(ServiceLifetime.Singleton)]来标记MyService。在Main方法中,我们使用
AddServicesWithInjectAttribute来自动扫描和注册服务,并最终通过依赖注入解析出服务并调用其方法。

总结

本文介绍了如何通过创建自定义特性与反射结合,实现一个简单的特性自动注入功能。在实际项目中,这种方式可以有效减少DI配置的冗余代码,提升开发效率。当然,这只是一个基础示例,实际应用中可以根据需要进行扩展和优化。

希望这篇文章对你有所帮助!如果有任何问题或建议,欢迎留言讨论。

本文暂时没有评论,来添加一个吧(●'◡'●)

欢迎 发表评论:

最近发表
标签列表