常見的 emit 實現 AOP demo

時間 2020-12-17

標籤 java git github 編程 api 框架 async 編程語言 ide 工具欄目 Java 简体版

原文原文鏈接

0. 前言

上接：思想無語言邊界：以 cglib 介紹 AOP 在 java 的一個實現方式java

做爲第四篇，咱們回顧一下 csharp 裏面比較常見動態編織實現方式emitgit

內容安排以下：github

emit aop demo
Norns.Urd

1. emit aop demo

1.1 emit 介紹

emit 是相似 java 中ASM地位的一個底層功能實現，編程

不過不是轉化java字節碼，而是生成dotnet 的 IL代碼，api

生成的IL代碼將由內置的JIT編譯器直接編譯到內存中。框架

官方的介紹文檔async

emit 對你們來講都是很熟悉的api了，動態作什麼事基本都會想到它。編程語言

咱們是可使用emit 作到上篇java 的 cglib 如出一轍的動態編織的AOP效果，因此語言真的只是工具，怎麼玩取決於玩工具的人，demo 以下。ide

1.2 demo

代碼工具

1.2.1 ProxyGenerator 簡單實現

public abstract class MethodInterceptor
{
    public abstract object Invoke(object instance, MethodInfo methodInfo, object[] parameters, object returnValue);
}


public static class ProxyGenerator
{
    private static ModuleBuilder moduleBuilder;
    private static MethodInfo getMethodMethod = typeof(MethodBase).GetMethod("GetMethodFromHandle", new[] { typeof(RuntimeMethodHandle) });
    private static MethodInfo invoke = typeof(MethodInterceptor).GetMethod("Invoke");
    static ProxyGenerator()
    {
        var asmBuilder = AssemblyBuilder.DefineDynamicAssembly(new AssemblyName("EmitAopDemoTest"), AssemblyBuilderAccess.RunAndCollect);
        moduleBuilder = asmBuilder.DefineDynamicModule("Proxy");
    }
    public static T Generate<T>(Type methodInterceptorType)
    {
        var proxyType = GenerateProxyType(typeof(T), methodInterceptorType);
        return (T)Activator.CreateInstance(proxyType);
    }
    public static Type GenerateProxyType(Type type, Type methodInterceptorType)
    {
        var typeBuilder = moduleBuilder.DefineType($"{type.Name}Proxy", TypeAttributes.Class | TypeAttributes.Public, type);
        foreach (var m in type.GetTypeInfo().DeclaredMethods)
        {
            var ps = m.GetParameters().Select(i => i.ParameterType).ToArray();
            var newM = typeBuilder.DefineMethod(m.Name, MethodAttributes.HideBySig | MethodAttributes.NewSlot | MethodAttributes.Virtual, m.CallingConvention, m.ReturnType, ps);
            CreateProxyMethod(methodInterceptorType, m, ps, newM);
            typeBuilder.DefineMethodOverride(newM, m);
        }
        return typeBuilder.CreateType();
    }
    private static void CreateProxyMethod(Type methodInterceptorType, MethodInfo m, Type[] ps, MethodBuilder newM)
    {
        var il = newM.GetILGenerator();
        var argsLocal = il.DeclareLocal(typeof(object[]));
        var returnLocal = il.DeclareLocal(typeof(object));
        // 初始化參數集合
        il.Emit(OpCodes.Ldc_I4, ps.Length);
        il.Emit(OpCodes.Newarr, typeof(object));
        for (var i = 0; i < ps.Length; i++)
        {
            il.Emit(OpCodes.Dup);
            il.Emit(OpCodes.Ldc_I4, i);
            il.Emit(OpCodes.Ldarg, i + 1);
            il.Emit(OpCodes.Box, ps[i]);
            il.Emit(OpCodes.Stelem_Ref);
        }
        il.Emit(OpCodes.Stloc, argsLocal);
        // 調用被代理方法
        il.Emit(OpCodes.Ldarg, 0); // load this
        for (var i = 0; i < ps.Length; i++)
        {
            il.Emit(OpCodes.Ldarg, i + 1);
        }
        il.Emit(OpCodes.Call, m);
        il.Emit(OpCodes.Box, newM.ReturnType);
        il.Emit(OpCodes.Stloc, returnLocal);
        //調用方法後攔截器
        il.Emit(OpCodes.Newobj, methodInterceptorType.GetConstructors().First());
        il.Emit(OpCodes.Ldarg, 0); // load this
        //加載方法信息參數
        il.Emit(OpCodes.Ldtoken, m);
        il.Emit(OpCodes.Call, getMethodMethod);
        il.Emit(OpCodes.Castclass, typeof(MethodInfo));
        il.Emit(OpCodes.Ldloc, argsLocal);
        il.Emit(OpCodes.Ldloc, returnLocal);
        il.Emit(OpCodes.Callvirt, invoke);
        il.Emit(OpCodes.Stloc, returnLocal);
        // return
        il.Emit(OpCodes.Ldloc, returnLocal);
        il.Emit(OpCodes.Unbox_Any, newM.ReturnType);
        il.Emit(OpCodes.Ret);
    }
}

1.2.2 Test

internal class Program
 {
     private static void Main(string[] args)
     {
         RealClass proxy = ProxyGenerator.Generate<RealClass>(typeof(AddOneInterceptor));
         var i = 5;
         var j = 10;
         Console.WriteLine($"{i} + {j} = {(i + j)}, but proxy is {proxy.Add(i, j)}");
     }
 }

結果：