.NET6.0实现IOC容器

.NET6.0实现IOC容器

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

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

1.   public interface IAuthService
2.   {
3.         bool CheckToken();
4.   }

复制代码

3. 实现接口

1. class AuthServiceImpl : IAuthService
2. {
3.         public bool CheckToken()
4.         {
5.             Console.WriteLine("check token");
6.             return true;
7.         }
8. }

复制代码

4. 配置IOC容器

下面是在 program 类中的代码

1. var services = new ServiceCollection();
2. services.AddSingleton<IAuthService, AuthServiceImpl>();

复制代码

5. 获取服务

通过在 Controller的构造函数中注入IAuthService

1.                 private readonly IAuthService _service;  
2.                 public WeatherForecastController(IAuthService service)
3.         {
4.             _service = service;
5.         }
7.         [HttpGet(Name = "test")]
8.         public bool Get()
9.         {
10.            return _service.CheckToken();
11.         }

复制代码

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

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

1.     [AttributeUsage(AttributeTargets.Class, Inherited = false, AllowMultiple = false)]
2.     public class AutowiredAttribute : Attribute
3.     {
4.         /// <summary>
5.         /// 接口
6.         /// </summary>
7.         public Type Iface { get; set; }
8.         /// <summary>
9.         /// 实现类名
10.         /// </summary>
11.         public string ImplClassName { get; set; }
13.         public AutowiredAttribute(Type iface, [CallerMemberName] string implClassName = "")
14.         {
15.             Iface = iface;
16.             ImplClassName = implClassName;
17.         }
18.     }

复制代码

6.2 实现服务加载类

利用IServiceCollection作为服务容器

1.     public class ServiceLoader
2.     {
3.         private static object _lock = new object();
4.         private static AppRuntime _inst;
5.         private readonly IServiceCollection _iocService = new ServiceCollection();
6.         private readonly ICollection<Assembly> _iocAssembly = new HashSet<Assembly>();
7.         private IServiceProvider _iocServiceProvider = null;
10.         public static ServiceLoader Instance
11.         {
12.             get
13.             {
14.                 if (_inst == null)
15.                 {
16.                     lock (_lock)
17.                     {
18.                         _inst = new ServiceLoader();
19.                         _inst.Startup(typeof(ServiceLoader).Assembly);
20.                     }
21.                 }
22.                 return _inst;
23.             }
24.         }
26.         public T GetService<T>()
27.         {
28.             EnsureAutoIoc<T>();
29.             return _iocServiceProvider.GetService<T>();
30.         }
32.         private void EnsureAutoIoc<T>()
33.         {
34.             Startup(typeof(T).Assembly);
35.         }
37.         public void Startup(Assembly ass)
38.         {
39.             if (_iocAssembly.Any(x => x == ass))
40.             {
41.                 return;
42.             }
43.             _iocAssembly.Add(ass);
45.             var types = ass.GetTypes().Where(x => x.GetCustomAttribute<AutowiredAttribute>() != null);
46.             foreach (var item in types)
47.             {
48.                 var autoIocAtt = item.GetCustomAttribute<AutowiredAttribute>();
49.                 AddTransient(autoIocAtt.Iface, item);
50.             }
51.             //_iocServiceProvider = _iocService.BuildServiceProvider();
52.             Interlocked.Exchange(ref _iocServiceProvider, _iocService.BuildServiceProvider());
53.         }
55.         private void AddTransient(Type iface, Type impl)
56.         {
57.             _iocService.AddTransient(iface, impl);
58.         }
59.     }

复制代码

6.3 在实现类加上标记

1.     [Autowired(typeof(IAuthService))]
2.     class AuthServiceImpl : IAuthService
3.     {
4.    
5.         public bool CheckToken()
6.         {
7.             Console.WriteLine("check token");
8.             return true;
9.         }
10.     }

复制代码

6.4 在 Controller 中调用

1.    var svc = ServiceLoader.Instance.GetService<IAuthService>();
2.    svc.CheckToken();

复制代码

至此一个基本的完整IOC容器已实现。

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