余烽 发表于 2023-9-5 17:41:00

.NET6.0实现IOC容器

.NET6.0实现IOC容器

IOC的作用这里省略…只对如何使用进行说明。
1. 创建一个.NET6应用程序

这里使用 .NET6.0 WebAPI 应用
2. 声明接口

public interface IAuthService
{
      bool CheckToken();
}3. 实现接口

class AuthServiceImpl : IAuthService
{
      public bool CheckToken()
      {
            Console.WriteLine("check token");
            return true;
      }
}4. 配置IOC容器

下面是在 program 类中的代码
var services = new ServiceCollection();
services.AddSingleton<IAuthService, AuthServiceImpl>();5. 获取服务

通过在 Controller的构造函数中注入IAuthService
                private readonly IAuthService _service;
                public WeatherForecastController(IAuthService service)
      {
            _service = service;
      }

      
      public bool Get()
      {
         return _service.CheckToken();
      }启动后,通过swagger发起请求,验证接口。
基本IOC容器流程已实现。但是这样存在一个弊端,每个接口和实现都要在program中手动注册一遍,还要在Controller构造函数中进行依赖注入,有没有能自动实现注册代替program中的手动注册?
接下来,对上述流程进行改良。
6. 改良思路

定义一个AutowiredAttribute标记,通过Atrribute标记的方式,在实现类上标记其要实现的接口服务,然后实现一个服务加载类ServiceLoader,在这个类中反射获取所有具备AutoIocAttribute的实现类,然后注册到ServiceCollection中。
6.1 定义特性标记AutowiredAttribute

   
    public class AutowiredAttribute : Attribute
    {
      /// <summary>
      /// 接口
      /// </summary>
      public Type Iface { get; set; }
      /// <summary>
      /// 实现类名
      /// </summary>
      public string ImplClassName { get; set; }

      public AutowiredAttribute(Type iface, string implClassName = "")
      {
            Iface = iface;
            ImplClassName = implClassName;
      }
    }6.2 实现服务加载类

利用IServiceCollection作为服务容器
    public class ServiceLoader
    {
      private static object _lock = new object();
      private static AppRuntime _inst;
      private readonly IServiceCollection _iocService = new ServiceCollection();
      private readonly ICollection<Assembly> _iocAssembly = new HashSet<Assembly>();
      private IServiceProvider _iocServiceProvider = null;


      public static ServiceLoader Instance
      {
            get
            {
                if (_inst == null)
                {
                  lock (_lock)
                  {
                        _inst = new ServiceLoader();
                        _inst.Startup(typeof(ServiceLoader).Assembly);
                  }
                }
                return _inst;
            }
      }

      public T GetService<T>()
      {
            EnsureAutoIoc<T>();
            return _iocServiceProvider.GetService<T>();
      }

      private void EnsureAutoIoc<T>()
      {
            Startup(typeof(T).Assembly);
      }

      public void Startup(Assembly ass)
      {
            if (_iocAssembly.Any(x => x == ass))
            {
                return;
            }
            _iocAssembly.Add(ass);

            var types = ass.GetTypes().Where(x => x.GetCustomAttribute<AutowiredAttribute>() != null);
            foreach (var item in types)
            {
                var autoIocAtt = item.GetCustomAttribute<AutowiredAttribute>();
                AddTransient(autoIocAtt.Iface, item);
            }
            //_iocServiceProvider = _iocService.BuildServiceProvider();
            Interlocked.Exchange(ref _iocServiceProvider, _iocService.BuildServiceProvider());
      }

      private void AddTransient(Type iface, Type impl)
      {
            _iocService.AddTransient(iface, impl);
      }
    }6.3 在实现类加上标记

   
    class AuthServiceImpl : IAuthService
    {
   
      public bool CheckToken()
      {
            Console.WriteLine("check token");
            return true;
      }
    }6.4 在 Controller 中调用

   var svc = ServiceLoader.Instance.GetService<IAuthService>();
   svc.CheckToken();至此一个基本的完整IOC容器已实现。

来源:https://www.cnblogs.com/ycit/archive/2023/09/05/17680238.html
免责声明:由于采集信息均来自互联网,如果侵犯了您的权益,请联系我们【E-Mail:cb@itdo.tech】 我们会及时删除侵权内容,谢谢合作!
页: [1]
查看完整版本: .NET6.0实现IOC容器