当前访客身份:游客 [ 登录  | 注册加入尚学堂]
直播

我来了!

拥有积分:3866
尚学堂雄起!!威武。。。

博客分类

笔记中心

课题中心

提问中心

答题中心

解答题中心

日志系统实战(二)-AOP动态获取运行时数据

我来了! 发表于 2年前 (2014-11-28 22:36:03)  |  评论(0)  |  阅读次数(333)| 0 人收藏此文章,   我要收藏   

 

介绍

这篇距上一篇已经拖3个月之久了,批评自己下。

通过前面一篇介绍。我们通过mono反射代码,可以拿出编译好的静态数据、例如方法参数信息之类的。但实际情况是:我更需要运行时的数据,就是用户输入等外界的动态数据。

既然是动态的,那就是未知的。我们怎么通过提前注入的代码获取呢?  

其实这是一个思路的问题,下面我们具体细看下。

 

实现

一 普通写法

 public static string GetPoint(int x, int y)
 {
    var value=x;
}

哇 好简单啊。其实动态获取和我们普通这样写代码是一样的,我们把要注入的代码,生成一个接收的变量就可以了。  就像上面value 一样接收,然后传递给我们自己函数就好了。

 

二 注入定义

 

  public class WeaveService : Attribute
    {
    }
    public class WeaveAction : Attribute
    {
    }
    public class Log : WeaveAction
    {
        public static void OnActionBefore(MethodBase mbBase, object[] args)
        {
            for (int i = 0; i < args.Length; i++)
            {
                Console.WriteLine(string.Format("{0}方法,第{1}参数是:{2}",mbBase.Name,i, args[i]));
            }
        }
    }

 

WeaveService WeaveAction 2个Attribute是注入的标记,方便我们在注入查找快速定位。

OnActionBefore是我们的接收函数,arg就是函数运行时的参数。

 

三  Weave函数

这块代码在上一篇已经有详细注释了,这里不多描述。

 public static void Weave(string[] assemblyPath)
        {
            foreach (var item in assemblyPath)
            {
                var assembly = AssemblyDefinition.ReadAssembly(item);

                var types = assembly.MainModule.Types.Where(n => n.CustomAttributes.Any(y => y.AttributeType.Resolve().Name == "WeaveService"));

                foreach (var type in types)
                {
                    foreach (var method in type.Methods)
                    {
                        var attrs =
                            method.CustomAttributes.Where(y => y.AttributeType.Resolve().BaseType.Name == "WeaveAction");
                        foreach (var attr in attrs)
                        {
                            var resolve = attr.AttributeType.Resolve();
                            var ilProcessor = method.Body.GetILProcessor();
                            var firstInstruction = ilProcessor.Body.Instructions.First();
                            var onActionBefore = resolve.GetMethods().Single(n => n.Name == "OnActionBefore");
                            var mfReference = assembly.MainModule.Import(typeof(System.Reflection.MethodBase).GetMethod("GetCurrentMethod"));
                            ilProcessor.InsertBefore(firstInstruction, ilProcessor.Create(OpCodes.Call, mfReference));

                            MakeArrayOfArguments(method, firstInstruction, ilProcessor, 0, method.Parameters.Count, assembly);
                            ilProcessor.InsertBefore(firstInstruction, ilProcessor.Create(OpCodes.Call, onActionBefore));
                        }
                    }
                }
                if (types.Any())
                {
                    assembly.Write(item);
                }
            }
        }

  

三  MakeArrayOfArguments 函数。

动态获取函数参数的函数,代码有详细注释。

 1    /// <summary>
 2         /// 构建函数参数
 3         /// </summary>
 4         /// <param name="method">要注入的方法</param>
 5         /// <param name="firstInstruction">函数体内第一行指令认 IL_0000: nop</param>
 6         /// <param name="writer">mono IL处理容器</param>
 7         /// <param name="firstArgument">默认第0个参数开始</param>
 8         /// <param name="argumentCount">函数参数的数量,静态数据可以拿到</param>
 9         /// <param name="assembly">要注入的程序集</param>
10         public static void MakeArrayOfArguments(MethodDefinition method, Instruction firstInstruction, ILProcessor writer, int firstArgument,
11                                           int argumentCount, AssemblyDefinition assembly)
12         {
13             //实例函数第一个参数值为this(当前实例对象),所以要从1开始。
14             int thisShift = method.IsStatic ? 0 : 1;
15 
16             if (argumentCount > 0) 
17             {
18                 //我们先创建个和原函数参数,等长的空数组。
19                 writer.InsertBefore(firstInstruction, writer.Create(OpCodes.Ldc_I4, argumentCount - firstArgument));
20                 //然后实例object数组,赋值给我们创建的数组
21                 writer.InsertBefore(firstInstruction, writer.Create(OpCodes.Newarr,
22                                             assembly.MainModule.Import(typeof(object))));
23 
24                 //c#代码描述
25                 //object[] arr=new object[argumentCount - firstArgument] 
26                 for (int i = firstArgument; i < argumentCount; i++)  //遍历参数
27                 {
28                     var parameter = method.Parameters[i];
29 
30                     //在堆栈上复制一个值
31                     writer.InsertBefore(firstInstruction, writer.Create(OpCodes.Dup));
32                     //将常量 i - firstArgument 进行压栈,数组[i - firstArgument] 这个东东。
33                     writer.InsertBefore(firstInstruction, writer.Create(OpCodes.Ldc_I4, i - firstArgument));
34                     //将第i + thisShift个参数 压栈。  
35                     writer.InsertBefore(firstInstruction, writer.Create(OpCodes.Ldarg, (short)(i + thisShift)));
36                     //装箱成object
37                     ToObject(assembly, firstInstruction, parameter.ParameterType, writer);
38                     //压栈给数组 arr[i]赋值
39                     writer.InsertBefore(firstInstruction, writer.Create(OpCodes.Stelem_Ref));
40 
41                     //c#代码描述
42                     // arr[i]=value;
43                 }
44             }
45             else
46             {
47                 writer.InsertBefore(firstInstruction, writer.Create(OpCodes.Ldnull));
48             }
49         }
50         public static void ToObject(AssemblyDefinition assembly, Instruction firstInstruction, TypeReference originalType, ILProcessor writer)
51         {
52             if (originalType.IsValueType)
53             {
54                 //普通值类型进行装箱操作
55                 writer.InsertBefore(firstInstruction, writer.Create(OpCodes.Box, originalType));
56             }
57             else
58             {
59                 if (originalType.IsGenericParameter)
60                 {
61                     //集合装箱
62                     writer.InsertBefore(firstInstruction, writer.Create(OpCodes.Box, assembly.MainModule.Import(originalType)));
63                 }
64 
65             }
66         }

 

 

介绍下mono InsertBefore这个函数。 这个函数是在某个指令之前插入指令。来张图

 

 

通过图我们看出,第一行指令是IL_0000: nop 。 第一行追加了 ldc.i4 2 指令,第二行我们还是nop 之前追加。 自上而下

 

四  业务编写

我定义个要注入的用户类,然后标记下。

  [WeaveService]
    public static class UserManager
    {

        [Log]
        public static string GetUserName(int userId, string memberid)
        {
            return "成功";
        }
        [Log]
        public static string GetPoint(int x, int y)
        {
            var sum = x + y;

            return "用户积分: " + sum;
        }
    }

 

我们平常的业务写法,不需要多余操作。

 public static void Main(string[] args)
        {
          
            UserManager.GetUserName(1,"v123465");
     
            UserManager.GetPoint(2, 3);

            Console.ReadLine();
        }

  

五 注入调用

我把业务类编译输入到D盘,用我们前面的函数Weave进行注入。

  CodeInject.Weave(new string[] { @"D:\test\Test.exe" });
         

 

运行结果如下

 

反编译后的c#

 

总结 

通过静态注入,我们能更好的从用途上去了解IL语言。

拿到动态数据仅仅抛砖引玉。 利用Mono我们可以写自己的AOP静态组件。

 

参考资源

1:postsharp源码

2:http://msdn.microsoft.com/en-us/library/system.reflection.emit.opcodes_fields(v=vs.110).aspx

 

作者:蘑菇先生      

出处:http://www.cnblogs.com/mushroom/p/4124878.html

转载请保留作者和原文地址。

 

分享到:0
关注微信,跟着我们扩展技术视野。每天推送IT新技术文章,每周聚焦一门新技术。微信二维码如下:
微信公众账号:尚学堂(微信号:bjsxt-java)
声明:博客文章版权属于原创作者,受法律保护。如果侵犯了您的权利,请联系管理员,我们将及时删除!
(邮箱:webmaster#sxt.cn(#换为@))
北京总部地址:北京市海淀区西三旗桥东建材城西路85号神州科技园B座三层尚学堂 咨询电话:400-009-1906 010-56233821
Copyright 2007-2015 北京尚学堂科技有限公司 京ICP备13018289号-1 京公网安备11010802015183