.NET 6 整合 Autofac 依赖注入容器

前言

一行业务代码还没写，框架代码一大堆，不利于学习。
常看到java的学习资料或博客，标题一般为《SpringBoot 整合 XXX》，所以仿照着写了《.NET 6 整合 Autofac 依赖注入容器》这样一个标题。
以下是我自己的用法，可能不是最佳实践。
一. 引用包

NuGet搜索并安装：
Autofac
Autofac.Extensions.DependencyInjection
二. 配置代码

1. var builder = WebApplication.CreateBuilder(args);
3. ...省略
5. // ASP.NET Core整合Autofac
6. builder.Host.UseServiceProviderFactory(new AutofacServiceProviderFactory());//通过工厂替换，把Autofac整合进来
7. builder.Host.ConfigureContainer<ContainerBuilder>(containerBuilder =>
8. {
9.     ServiceFactory.SetBuilder(containerBuilder);
10.     ServiceFactory.RegisterAssembly(Assembly.GetExecutingAssembly()); //注册服务
11. });
13. var app = builder.Build();
15. ServiceFactory.SetContainer((app.Services as AutofacServiceProvider).LifetimeScope as IContainer);
16. Task.Run(async () => await ServiceFactory.StartAllService()); //启动服务，注意：服务启动完成之前，调用接口会异常
18. ...省略

复制代码

三. 服务的自动注册、启动、停止

IService接口

1. /// <summary>
2. /// 服务接口
3. /// </summary>
4. public interface IService
5. {
6.     /// <summary>
7.     /// 服务启动
8.     /// </summary>
9.     Task OnStart();
11.     /// <summary>
12.     /// 服务停止
13.     /// </summary>
14.     Task OnStop();
15. }

复制代码

RegisterServiceAttribute特性

1. /// <summary>
2. /// 注册为服务
3. /// </summary>
4. public class RegisterServiceAttribute : Attribute
5. {
6. }

复制代码

ServiceBase抽像类

1. [RegisterService]
2. public abstract class ServiceBase : IService
3. {
4.     #region OnStart
5.     public virtual Task OnStart()
6.     {
7.         return Task.CompletedTask;
8.     }
9.     #endregion
11.     #region OnStop
12.     public virtual Task OnStop()
13.     {
14.         return Task.CompletedTask;
15.     }
16.     #endregion
18. }

复制代码

ServiceFactory服务工厂类

1. public class ServiceFactory
2. {
3.     private static ContainerBuilder _builder;
5.     private static IContainer _container;
7.     private static bool _isRunning; //服务是否正在运行
9.     public static void SetBuilder(ContainerBuilder builder)
10.     {
11.         _builder = builder;
12.     }
14.     public static void SetContainer(IContainer container)
15.     {
16.         _container = container;
17.     }
19.     /// <summary>
20.     /// Autofac.ContainerBuilder
21.     /// </summary>
22.     public static ContainerBuilder Builder
23.     {
24.         get
25.         {
26.             return _builder;
27.         }
28.     }
30.     /// <summary>
31.     /// 获取服务
32.     /// </summary>
33.     /// <typeparam name="T">接口类型</typeparam>
34.     public static T Get<T>()
35.     {
36.         if (_isRunning)
37.         {
38.             return _container.Resolve<T>();
39.         }
40.         else
41.         {
42.             throw new Exception("服务尚未启动完成");
43.         }
44.     }
46.     /// <summary>
47.     /// 获取服务
48.     /// </summary>
49.     /// <param name="type">接口类型</param>
50.     public static object Get(Type type)
51.     {
52.         if (_isRunning)
53.         {
54.             return _container.Resolve(type);
55.         }
56.         else
57.         {
58.             throw new Exception("服务尚未启动完成");
59.         }
60.     }
62.     #region 注册程序集
63.     /// <summary>
64.     /// 注册程序集
65.     /// </summary>
66.     /// <param name="serviceAssembly">服务程序集</param>
67.     public static void RegisterAssembly(Assembly serviceAssembly)
68.     {
69.         Type[] typeArr = serviceAssembly.GetTypes();
71.         foreach (Type type in typeArr)
72.         {
73.             if (type.GetCustomAttribute<RegisterServiceAttribute>() != null && !type.IsAbstract)
74.             {
75.                 _builder.RegisterType(type).SingleInstance().AsImplementedInterfaces().AsSelf();
76.             }
77.             else
78.             {
79.                 Type[] interfaceTypes = type.GetInterfaces();
80.                 foreach (Type interfaceType in interfaceTypes)
81.                 {
82.                     if (interfaceType.GetCustomAttribute<RegisterServiceAttribute>() != null && !type.IsAbstract)
83.                     {
84.                         _builder.RegisterType(type).SingleInstance().AsImplementedInterfaces().AsSelf();
85.                         break;
86.                     }
87.                 }
88.             }
89.         }
91.         //_container = _builder.Build();
92.     }
93.     #endregion
95.     #region 启动所有服务
96.     /// <summary>
97.     /// 启动所有服务
98.     /// </summary>
99.     public static async Task StartAllService()
100.     {
101.         Type iServiceInterfaceType = typeof(IService);
102.         IEnumerable<Type> types = _container.ComponentRegistry.Registrations.Select(a => a.Activator.LimitType);
103.         await Parallel.ForEachAsync(types, async (type, c) =>
104.         {
105.             if (iServiceInterfaceType.IsAssignableFrom(type))
106.             {
107.                 object obj = _container.Resolve(type);
109.                 try
110.                 {
111.                     IService service = obj as IService;
112.                     await service.OnStart();
113.                     LogUtil.Info("服务 " + obj.GetType().FullName + " 已启动");
114.                 }
115.                 catch (Exception ex)
116.                 {
117.                     LogUtil.Error(ex, "服务 " + obj.GetType().FullName + " 启动失败");
118.                 }
119.             }
120.         });
121.         _isRunning = true;
122.     }
123.     #endregion
125.     #region 停止所有服务
126.     /// <summary>
127.     /// 停止所有服务
128.     /// </summary>
129.     public static async Task StopAllService()
130.     {
131.         Type iServiceInterfaceType = typeof(IService);
132.         IEnumerable<Type> types = _container.ComponentRegistry.Registrations.Select(a => a.Activator.LimitType);
133.         await Parallel.ForEachAsync(types, async (type, c) =>
134.         {
135.             if (iServiceInterfaceType.IsAssignableFrom(type))
136.             {
137.                 object obj = _container.Resolve(type);
138.                 IService service = obj as IService;
140.                 try
141.                 {
142.                     await service.OnStop();
143.                     LogUtil.Info("服务 " + obj.GetType().FullName + " 已停止");
144.                 }
145.                 catch (Exception ex)
146.                 {
147.                     LogUtil.Error(ex, "服务 " + obj.GetType().FullName + " 停止失败");
148.                 }
149.             }
150.         });
151.         _isRunning = false;
152.     }
153.     #endregion
155. }

复制代码

ServiceFactory的简写类

1. /// <summary>
2. /// ServiceFactory的简写
3. /// </summary>
4. public class SF
5. {
6.     /// <summary>
7.     /// 获取服务
8.     /// </summary>
9.     /// <typeparam name="T">接口类型</typeparam>
10.     public static T Get<T>()
11.     {
12.         return ServiceFactory.Get<T>();
13.     }
15.     /// <summary>
16.     /// 获取服务
17.     /// </summary>
18.     /// <param name="type">接口类型</param>
19.     public static object Get(Type type)
20.     {
21.         return ServiceFactory.Get(type);
22.     }
23. }

复制代码

编写服务类

服务类示例1

1. /// <summary>
2. /// 测试服务
3. /// </summary>
4. public class TestService : ServiceBase
5. {
6.     public async Task<string> GetValue()
7.     {
8.         return await Task.FromResult("测试数据");
9.     }
10. }

复制代码

服务类示例2

重写了OnStart和OnStop以实现服务的启动和停止。

1. /// <summary>
2. /// 定时任务管理
3. /// </summary>
4. public class ScheduleJobs : ServiceBase
5. {
6.     #region 变量
7.     private IScheduler _scheduler;
8.     #endregion
10.     #region OnStart
11.     public override async Task OnStart()
12.     {
13.         try
14.         {
15.             NameValueCollection options = new NameValueCollection();
16.             string schedulerName = "DefaultQuartzScheduler";
17.             options.Add("quartz.scheduler.instanceName", schedulerName);
18.             StdSchedulerFactory schedulerFactory = new StdSchedulerFactory(options);
19.             _scheduler = await schedulerFactory.GetScheduler(schedulerName);
20.             if (_scheduler == null)
21.             {
22.                 _scheduler = await schedulerFactory.GetScheduler();
23.             }
24.             await _scheduler.Start();
25.             AddJobs(_scheduler);
26.         }
27.         catch (Exception ex)
28.         {
29.             Console.WriteLine(ex.ToString());
30.         }
31.     }
32.     #endregion
34.     #region OnStop
35.     public override async Task OnStop()
36.     {
37.         await _scheduler.Shutdown();
38.     }
39.     #endregion
41.     #region ScheduleJob
42.     private async Task ScheduleJob<T>(IScheduler scheduler, string cronString) where T : IJob
43.     {
44.         IJobDetail jobDetail = JobBuilder.Create<T>().Build();
45.         ITrigger trigger = TriggerBuilder.Create().WithCronSchedule(cronString).Build();
46.         await scheduler.ScheduleJob(jobDetail, trigger);
47.     }
48.     #endregion
50.     private async void AddJobs(IScheduler scheduler)
51.     {
52.         await ScheduleJob<TestJob>(scheduler, "0/30 * * * * ?");
53.     }
55. }

复制代码

构造注入示例

1. [ApiController]
2. [Route("[controller]")]
3. public class TestController : ControllerBase
4. {
5.     private readonly TestService _testService;
7.     public TestController(TestService testService)
8.     {
9.         _testService = testService;
10.     }
12.     ...省略
13. }

复制代码

也可以这样使用

1. var value = await SF.Get<TestService>().GetValue();

复制代码

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