1.创建控制器的过程依赖众多不同的提供者和工厂类，但最终是由实现IControllerActivator接口的实例来决定的。实现类只需要实现两个方法：
public interface IControllerActivator
{
//
// 摘要:
// Creates a controller.
//
// 参数:
// context:
// The Microsoft.AspNetCore.Mvc.ControllerContext for the executing action.
object Create(ControllerContext context);
//
// 摘要:
// Releases a controller.
//
// 参数:
// context:
// The Microsoft.AspNetCore.Mvc.ControllerContext for the executing action.
//
// controller:
// The controller to release.
void Release(ControllerContext context, object controller);
}
如你所见，该Create方法传递了用于创建控制器的ControllerContext实例。控制器的创建方式取决于具体的实现。
2.在各大第三方IOC容器中，实现方法注入与属性注入都是先要标记特性所以我们第一步，先新建两个特性
代表属性注入的特性
using System;
using System.Collections.Generic;
using System.Text;
namespace GreenUnity.Common.Attributes
{
/// <summary>
/// 限制特性只能标记在属性上(该特性什么都不做，只用来标记属性注入)
/// </summary>
[AttributeUsage(AttributeTargets.Property)]
public class PropertyInjectionAttribute:Attribute
{
}
}
代表方法注入的特性
using System;
using System.Collections.Generic;
using System.Text;
namespace GreenUnity.Common.Attributes
{
/// <summary>
/// 限制特性只能标记在方法上(该特性什么都不做，只用来标记方法注入)
/// </summary>
[AttributeUsage(AttributeTargets.Method)]
public class MethodInjectionAttribute:Attribute
{
}
}
3.前面我们讲过，控制器的实例化是来自于IControllerActivator接口实列的Create方法,我们新建一个类，继承IControllerActivator接口，实现Create方法，在Create方法里做属性注入与方法注入的实现。注意此扩展方法未解决循环依赖问题
using GreenUnity.Common.Attributes;
using Microsoft.AspNetCore.Mvc;
using Microsoft.AspNetCore.Mvc.Controllers;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Reflection;
using System.Text;
namespace GreenUnity.Common.Extend
{
public class CustomControllerActivator : IControllerActivator
{
/// <summary>
/// 该方法扩展框架IOC属性与方法注入但是未解决循环依赖问题.使用时切勿循环依赖会死循环内存溢出
/// </summary>
/// <param name="context"></param>
/// <returns></returns>
public object Create(ControllerContext context)
{
//获得框架得服务提供对象,用于实列化控制器
IServiceProvider serviceProvider = context.HttpContext.RequestServices;
//获取控制器类型
Type type = context.ActionDescriptor.ControllerTypeInfo.AsType();
//实列化控制器
object Context = serviceProvider.GetService(type);
//属性注入
PropertyInjection(type, serviceProvider, Context);
//方法注入
MethodInjection(type, serviceProvider, Context);
return Context;//把实列化的控制器返回
}
/// <summary>
/// 属性注入
/// </summary>
/// <param name="type"></param>
/// <param name="serviceProvider"></param>
public void PropertyInjection(Type type, IServiceProvider serviceProvider, object Context)
{
try
{
//根据自定义特性确定要属性注入的属性。
//利用反射获取控制器中应用了PropertyInjectionAttribute的属性。
foreach (PropertyInfo propertyInfo in type.GetProperties(BindingFlags.NonPublic | BindingFlags.Instance).Where(m => m.IsDefined(typeof(PropertyInjectionAttribute), true)))
{
Type PropertyType = propertyInfo.PropertyType;//获取当前属性的类型。注：GetType()获取的是当前对象的类型
object property = serviceProvider.GetService(PropertyType);//利用框架的服务提供对象 创建对象
PropertyInjection(PropertyType, serviceProvider, property);//如果有多层注入需要递归创建对象,把当前对象传入。
MethodInjection(PropertyType, serviceProvider, property);//如果有多层注入需要递归创建对象,把当前对象传入。
propertyInfo.SetValue(Context, property);//给属性赋值
}
}
catch (Exception ex)
{
}
}
/// <summary>
/// 方法注入
/// </summary>
/// <param name="type"></param>
/// <param name="serviceProvider"></param>
/// <param name="Context"></param>
public void MethodInjection(Type type, IServiceProvider serviceProvider, object Context)
{
try
{
//根据自定义特性确定要方法注入的方法。
//利用反射获取控制器中应用了MethodInjectionAttribute特性的方法。
foreach (MethodInfo memberInfo in type.GetMethods(BindingFlags.NonPublic | BindingFlags.Instance).Where(m => m.IsDefined(typeof(MethodInjectionAttribute))))
{
ParameterInfo[] parameterInfos = memberInfo.GetParameters();//获取方法参数列表
object[] Parameters = new object[parameterInfos.Length];
//循环实列化方法参数
for (int i = 0; i < parameterInfos.Length; i++)
{
Type ParameterType = parameterInfos[i].ParameterType;//获取当前方法参数的类型
object Parameter = serviceProvider.GetService(ParameterType);//利用框架的服务提供对象 创建对象
MethodInjection(ParameterType, serviceProvider, Parameter);//如果有多层注入需要递归创建对象,把当前对象传入。
PropertyInjection(ParameterType, serviceProvider, Parameter);//如果有多层注入需要递归创建对象,把当前对象传入。
Parameters[i] = Parameter;//添加到参数实列数组中去
}
memberInfo.Invoke(Context, Parameters);//执行方法
}
}
catch (Exception ex)
{
}
}
public void Release(ControllerContext context, object controller)
{
}
}
}
4.在Startup的ConfigureServices中替换掉原有的ControllerActivator实现
#region 替换掉框架自己实现的ControllerActivator //1.把控制器当做服务注册到容器中 services.AddControllersWithViews().AddControllersAsServices(); //2.找出默认的Activator var DefaultActivator = services.FirstOrDefault(m => m.ServiceType == typeof(IControllerActivator)); //3.移除默认Activator services.Remove(DefaultActivator); //4.把自己自定义的Activator添加进去 services.AddTransient<IControllerActivator, CustomControllerActivator>(); #endregion
5.到此已经配置扩展完成，在需要属性注入和方法注入的地方加上特性就行。扩展的方法可以自动注入
/// <summary>
/// 属性注入
/// </summary>
[PropertyInjection]
private UnitOfWork unitOfWork { get; set; }
/// <summary>
/// 方法注入
/// </summary>
private ILogger<UserController> Logger = null;
[MethodInjection]
private void MethodInjection(ILogger<UserController> _logger)
{
Logger = _logger;
}