使用Unity实现AOP拦截器时，自动完成注册的处理 - ToB企服应用市场:ToB评测及商务社交产业平台

void SetInterceptor()
{
IUnityContainer unityContainer = new UnityContainer();
IUnityContainer interception = unityContainer.AddNewExtension<Interception>();
//interception.RegisterSingleton<CustomViewModel>(new Interceptor<VirtualMethodInterceptor>(), new InterceptionBehavior<ViewModelCommandBehavior>());
//return;
Assembly assem = Assembly.GetExecutingAssembly();
var types = assem.ExportedTypes;
foreach (Type item in types)
{
//具体做哪些限制根据实际情况来
if (item.IsClass && typeof(ViewModelBase).IsAssignableFrom(item))
{
//这里可以加一些其它的限制，目前值用于CustomViewModel
if (item.FullName == typeof(CustomViewModel).FullName)
{
//可以参考上面的RegisterSingleton方法调用，这里的RegisterSingleton是用的扩展类中的扩展方法
//所以在查找方法时是直接在扩展类中找的，而且因为扩展方法是静态方法所以Invoke执行时对象可以设置为null
var method = typeof(UnityContainerExtensions).GetMethods(BindingFlags.Static | BindingFlags.Public)
.Where(x => x.Name == "RegisterSingleton" && x.IsGenericMethod == true && x.GetGenericArguments().Length == 1
&& x.GetParameters().Length == 2 && x.GetParameters()[1].ParameterType.FullName == typeof(InjectionMember[]).FullName).FirstOrDefault();
if (method != null)
{
method = method.MakeGenericMethod(new Type[] { item });
InjectionMember[] injectionMembers = new InjectionMember[2];
injectionMembers[0] = new Interceptor<VirtualMethodInterceptor>();
injectionMembers[1] = new InterceptionBehavior<ViewModelCommandBehavior>();
object[] pars = new object[2];
pars[0] = interception;
pars[1] = injectionMembers;
method.Invoke(null, pars);
}
}
}
}
}

复制代码