ServiceModule.cs 2.0 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556
  1. using Autofac;
  2. using Autofac.Extras.DynamicProxy;
  3. using System;
  4. using System.Collections.Generic;
  5. using System.Linq;
  6. using System.Reflection;
  7. using Module = Autofac.Module;
  8. using ZhonTai.Plate.Admin.Domain;
  9. using ZhonTai.Common.Configs;
  10. using Microsoft.Extensions.DependencyModel;
  11. namespace ZhonTai.Plate.Admin.HttpApi.Shared.RegisterModules
  12. {
  13. public class ServiceModule : Module
  14. {
  15. private readonly AppConfig _appConfig;
  16. private readonly string _assemblySuffixName;
  17. /// <summary>
  18. /// 服务注入
  19. /// </summary>
  20. /// <param name="appConfig">AppConfig</param>
  21. /// <param name="assemblyName">程序集名称</param>
  22. /// <param name="assemblySuffixName">程序集后缀名</param>
  23. public ServiceModule(AppConfig appConfig, string assemblySuffixName = "Service")
  24. {
  25. _appConfig = appConfig;
  26. _assemblySuffixName = assemblySuffixName;
  27. }
  28. protected override void Load(ContainerBuilder builder)
  29. {
  30. //事务拦截
  31. var interceptorServiceTypes = new List<Type>();
  32. if (_appConfig.Aop.Transaction)
  33. {
  34. builder.RegisterType<TransactionInterceptor>();
  35. builder.RegisterType<TransactionAsyncInterceptor>();
  36. interceptorServiceTypes.Add(typeof(TransactionInterceptor));
  37. }
  38. //服务
  39. Assembly[] assemblies = DependencyContext.Default.RuntimeLibraries
  40. .Where(a => a.Name.EndsWith(_assemblySuffixName))
  41. .Select(o => Assembly.Load(new AssemblyName(o.Name))).ToArray();
  42. builder.RegisterAssemblyTypes(assemblies)
  43. .Where(a => a.Name.EndsWith(_assemblySuffixName))
  44. .AsImplementedInterfaces()
  45. .InstancePerLifetimeScope()
  46. .PropertiesAutowired()// 属性注入
  47. .InterceptedBy(interceptorServiceTypes.ToArray())
  48. .EnableInterfaceInterceptors();
  49. }
  50. }
  51. }