使用Unity实现AOP拦截器时，自动完成注册的处理

说明：这里只是VirtualMethodInterceptor拦截器的处理，下面代码中用到的ViewModelCommandBehavior是对IInterceptionBehavior接口的实现
主要就是用反射获取到相应的方法来执行指定的操作
问题：通常是用下面注释掉的 interception.RegisterSingleton 来注册，但是当需要注册的类很多或者不确定时会很不方便，所以改用下面的方法来自动完成相关类的注册

1.         void SetInterceptor()
2.         {
3.             IUnityContainer unityContainer = new UnityContainer();
4.             IUnityContainer interception = unityContainer.AddNewExtension<Interception>();
5.             //interception.RegisterSingleton<CustomViewModel>(new Interceptor<VirtualMethodInterceptor>(), new InterceptionBehavior<ViewModelCommandBehavior>());
6.             //return;
7.             Assembly assem = Assembly.GetExecutingAssembly();
8.             var types = assem.ExportedTypes;
9.             foreach (Type item in types)
10.             {
11.                 //具体做哪些限制根据实际情况来
12.                 if (item.IsClass && typeof(ViewModelBase).IsAssignableFrom(item))
13.                 {
14.                     //这里可以加一些其它的限制，目前值用于CustomViewModel
15.                     if (item.FullName == typeof(CustomViewModel).FullName)
16.                     {
17.                         //可以参考上面的RegisterSingleton方法调用，这里的RegisterSingleton是用的扩展类中的扩展方法
18.                         //所以在查找方法时是直接在扩展类中找的，而且因为扩展方法是静态方法所以Invoke执行时对象可以设置为null
19.                         var method = typeof(UnityContainerExtensions).GetMethods(BindingFlags.Static | BindingFlags.Public)
20.                             .Where(x => x.Name == "RegisterSingleton" && x.IsGenericMethod == true && x.GetGenericArguments().Length == 1
21.                             && x.GetParameters().Length == 2 && x.GetParameters()[1].ParameterType.FullName == typeof(InjectionMember[]).FullName).FirstOrDefault();
22.                         if (method != null)
23.                         {
24.                             method = method.MakeGenericMethod(new Type[] { item });
25.                             InjectionMember[] injectionMembers = new InjectionMember[2];
26.                             injectionMembers[0] = new Interceptor<VirtualMethodInterceptor>();
27.                             injectionMembers[1] = new InterceptionBehavior<ViewModelCommandBehavior>();
28.                             object[] pars = new object[2];
29.                             pars[0] = interception;
30.                             pars[1] = injectionMembers;
31.                             method.Invoke(null, pars);
32.                         }
33.                     }
34.                 }
35.             }
36.         }

复制代码

 

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