C#通用的动态代理

目标框架.Net Standard 2.0

安装引用类库System.Reflection.Emit  4.3.0

其他目标框架均适自带此类库,无需安装引用类库

原理:使用反射获取类、方法,将实现的代理类Invoke 通过System.Reflection.Emit改写到需要被改写的接口中,生成继承业务接口的中间类,调用中间类的接口则实现代理的任务。 

    /// <summary>

    /// 对应java的InvocationHandler接口

    /// 使用上应该是差不多的

    /// 注意点是,因为C#的Property的getset也是走这一个方法的

    /// 对于接口来说是全部代理,但是对于类只有虚方法代理

    /// </summary>

    public interface IInvocationHandler

    {

        object Invoke(object proxy, MethodInfo method, object[] args);

    } 

 public interface myIInvocationHandler

    {

        object m(string args, string arg2);

    }

    public class II : IInvocationHandler

    {

        public object Invoke(object proxy, MethodInfo method, object[] args)

        {

            foreach (object o in args)

            {

                Console.WriteLine(o.ToString());

            }

            Console.WriteLine($"hahahaha");

            return args[1];

        }

    }

    /// <summary>

    /// 缓存类,可以无视

    /// </summary>

    class ProxyTypeInfo

    {

        public TypeBuilder TypeBuilder;

        public int Count;

        public MethodInfo[] MethodInfos;

    }

    /// <summary>

    /// 代理类

    /// </summary>

    public static class DynamicProxy

    {

        private static readonly string AssemblyName = "DynamicProxyAssembly";

        private static readonly string ModuleName = "DynamicProxyModule";

        private static readonly string TypeName = "DynamicProxy";

        /// <summary>

        /// 因为有些方法的指令是需要拆箱装箱的

        /// </summary>

        private static readonly HashSet<Type> CanBox = new HashSet<Type>

    {

        typeof(int), typeof(uint),

        typeof(short), typeof(ushort),

        typeof(long), typeof(ulong),

        typeof(float), typeof(double),

        typeof(sbyte), typeof(byte),

        typeof(char),

        typeof(decimal),

    };

        private static readonly Dictionary<Type, ProxyTypeInfo> ProxyDict = new Dictionary<Type, ProxyTypeInfo>();

        private static TypeBuilder CreateDynamicTypeBuilder(Type type, Type parent, Type[] interfaces)

        {

            if (ProxyDict.TryGetValue(type, out var info))

            {

                info.Count++;

            }

            else

            {

                ProxyDict[type] = info = new ProxyTypeInfo

                {

                    Count = 1

                };

            }

            //AssemblyBuilder ab =

            //      AssemblyBuilder.DefineDynamicAssembly(

            //          aName,

            //          AssemblyBuilderAccess.Run);

            var assemblyBuilder = AssemblyBuilder.DefineDynamicAssembly(new AssemblyName(AssemblyName + type.Name),

                AssemblyBuilderAccess.Run);

            var moduleBuilder = assemblyBuilder.DefineDynamicModule(ModuleName + type.Name);

            return info.TypeBuilder = moduleBuilder.DefineType(TypeName + type.Name + info.Count,

                TypeAttributes.Public | TypeAttributes.Class, parent, interfaces);

        }

        private static void ProxyInit(Type type, TypeBuilder typeBuilder, MethodInfo[] methodInfos,

            MethodInfo handlerInvokeMethodInfo)

        {

            //定义两个字段

            var handlerFieldBuilder =

                typeBuilder.DefineField("_handler", typeof(IInvocationHandler), FieldAttributes.Private);

            var methodInfosFieldBuilder =

                typeBuilder.DefineField("_methodInfos", typeof(MethodInfo), FieldAttributes.Private);

            //定义构造函数

            var constructorBuilder = typeBuilder.DefineConstructor(MethodAttributes.Public, CallingConventions.Standard,

                new[] { typeof(IInvocationHandler), typeof(MethodInfo[]) });

            var ilCtor = constructorBuilder.GetILGenerator();

            ilCtor.Emit(OpCodes.Ldarg_0);

            ilCtor.Emit(OpCodes.Call,

                typeof(object).GetConstructor(new Type[0]) ?? throw new Exception("不可能的错误:object.GetConstructor"));

            ilCtor.Emit(OpCodes.Ldarg_0);

            ilCtor.Emit(OpCodes.Ldarg_1);

            ilCtor.Emit(OpCodes.Stfld, handlerFieldBuilder);

            ilCtor.Emit(OpCodes.Ldarg_0);

            ilCtor.Emit(OpCodes.Ldarg_2);

            ilCtor.Emit(OpCodes.Stfld, methodInfosFieldBuilder);

            ilCtor.Emit(OpCodes.Ret);

            for (var i = 0; i < methodInfos.Length; i++)

            {

                var methodInfo = methodInfos[i];

                var parameterTypes = methodInfo.GetParameters().Select(p => p.ParameterType).ToArray();

                var methodBuilder = typeBuilder.DefineMethod(methodInfo.Name,

                    MethodAttributes.Public | MethodAttributes.Virtual,

                    methodInfo.CallingConvention, methodInfo.ReturnType, parameterTypes);

                var ilMethod = methodBuilder.GetILGenerator();

                ilMethod.Emit(OpCodes.Ldarg_0);

                ilMethod.Emit(OpCodes.Ldfld, handlerFieldBuilder);

                ilMethod.Emit(OpCodes.Ldarg_0);

                ilMethod.Emit(OpCodes.Ldarg_0);

                ilMethod.Emit(OpCodes.Ldfld, methodInfosFieldBuilder);

                ilMethod.Emit(OpCodes.Ldc_I4, i);

                ilMethod.Emit(OpCodes.Ldelem_Ref);

                ilMethod.Emit(OpCodes.Ldc_I4, parameterTypes.Length);

                ilMethod.Emit(OpCodes.Newarr, typeof(object));

                for (var j = 0; j < parameterTypes.Length; j++)

                {

                    ilMethod.Emit(OpCodes.Dup);

                    ilMethod.Emit(OpCodes.Ldc_I4_S, (short)j);

                    ilMethod.Emit(OpCodes.Ldarg_S, (short)(j + 1));

                    if (CanBox.Contains(parameterTypes[j]))

                    {

                        ilMethod.Emit(OpCodes.Box, parameterTypes[j]);

                    }

                    ilMethod.Emit(OpCodes.Stelem_Ref);

                }

                ilMethod.Emit(OpCodes.Callvirt, handlerInvokeMethodInfo);

                ilMethod.Emit(CanBox.Contains(methodInfo.ReturnType) ? OpCodes.Unbox_Any : OpCodes.Castclass,

                    methodInfo.ReturnType);

                ilMethod.Emit(OpCodes.Ret);

            }

        }

        /// <summary>

        /// 通过接口创建动态代理

        /// </summary>

        public static T CreateProxyByInterface<T>(IInvocationHandler handler, bool userCache = true)

        {

            return (T)CreateProxyByInterface(typeof(T), handler, userCache);

        }

        public static object CreateProxyByInterface(Type type, IInvocationHandler handler, bool userCache = true)

        {

            if (!userCache || !ProxyDict.TryGetValue(type, out var info))

            {

                var handlerInvokeMethodInfo = typeof(IInvocationHandler).GetMethod("Invoke") ??

                                              throw new Exception("不可能的错误:handlerInvokeMethodInfo");

                var typeBuilder = CreateDynamicTypeBuilder(type, null, new[] { type });

                var methodInfos = type.GetMethods();

                ProxyInit(type, typeBuilder, methodInfos, handlerInvokeMethodInfo);

                info = ProxyDict[type];

                if (info.Count == 1)

                {

                    info.MethodInfos = methodInfos;

                }

            }

            //Type t = info.TypeBuilder.CreateTypeInfo();

            return Activator.CreateInstance(info.TypeBuilder.CreateTypeInfo(), handler, info.MethodInfos) ??

                                              throw new Exception("不同环境此处可能需要改写");

        }

        /// <summary>

        /// 通过类创建动态代理

        /// </summary>

        public static T CreateProxyByType<T>(IInvocationHandler handler, bool userCache = true)

        {

            return (T)CreateProxyByType(typeof(T), handler, userCache);

        }

        public static object CreateProxyByType(Type type, IInvocationHandler handler, bool userCache = true)

        {

            if (!userCache || !ProxyDict.TryGetValue(type, out var info))

            {

                var handlerInvokeMethodInfo = typeof(IInvocationHandler).GetMethod("Invoke") ??

                                              throw new Exception("不可能的错误:handlerInvokeMethodInfo");

                var typeBuilder = CreateDynamicTypeBuilder(type, type, null);

                var methodInfos = type.GetMethods().Where(methodInfo => methodInfo.IsVirtual || methodInfo.IsAbstract)

                    .ToArray();

                ProxyInit(type, typeBuilder, methodInfos, handlerInvokeMethodInfo);

                info = ProxyDict[type];

                if (info.Count == 1)

                {

                    info.MethodInfos = methodInfos;

                }

            }

            return Activator.CreateInstance(info.TypeBuilder.CreateTypeInfo(), handler, info.MethodInfos) ??

                                              throw new Exception("不同环境此处可能需要改写");

        }

    }

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 221,198评论 6 514
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 94,334评论 3 398
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 167,643评论 0 360
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 59,495评论 1 296
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 68,502评论 6 397
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 52,156评论 1 308
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,743评论 3 421
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 39,659评论 0 276
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 46,200评论 1 319
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 38,282评论 3 340
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 40,424评论 1 352
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 36,107评论 5 349
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 41,789评论 3 333
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 32,264评论 0 23
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 33,390评论 1 271
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 48,798评论 3 376
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 45,435评论 2 359

推荐阅读更多精彩内容