利用Mono.Cecil制作出枚举类的泛型扩展方法


我的项目的某部分需要经常判断枚举的某个标识的存在,就经常出现类似于“(A & B) == B;”这样的代码。不过我看着觉得不爽,也觉得很繁琐。我知道Enum类有HasFlag()方法,但是这个会导致装箱,智能感知也感知不出来,不好。于是我就写了下面的代码:

1publicstaticboolHas<TEnum>(thisTEnumrpEnum,TEnumrpValue)
2whereTEnum:struct,Enum
3{
4return(rpEnum&rpValue)==rpValue;
5}

可是这代码无法通过编译

因为在编译器的眼中,Enum是个特殊类型,无法作为泛型约束(参考:http://msdn.microsoft.com/zh-cn/library/56b2hk61.aspx

难道这个只能是空想吗?非也

经过搜索,发现在C++/CLI里没有这个限制(参考:http://social.msdn.microsoft.com/Forums/en-US/vs2010ctpvbcs/thread/5844d719-3ef6-4000-913c-27c108269e71)。但我的基础类库不可能全部换用C++/CLI写吧(要知道C++/CLI的语法弄得我头都大了)

不过想想,在C++/CLI可以通过编译,那就说明在IL方面是可以支持枚举作为泛型约束,只是C#编译器有问题而已。于是我就想到了Mono.Cecil这个神器,用它修改编译后的程序集来实现我要的方法(我的基础类库也不可能全部换用IL来写吧)

修改

先弄个伪方法:

View Code
1publicstaticboolHas<TEnum>(thisTEnumrpEnum,TEnumrpValue)
2whereTEnum:struct
3{
4returntrue;
5}

然后新建一个程序,把Mono.Cecil导入进去(下载:http://www.mono-project.com/Cecil

下面是主要代码:

View Code
1staticvoidMain(string[]args)
2{
3//读取程序集
4varrAssembly=AssemblyDefinition.ReadAssembly("Moen.Base.dll");
5//获取指定的类
6varrType=rAssembly.MainModule.GetType("Moen","EnumExtensions");
7//获取指定的方法
8varrMethod=rType.Methods.Single(r=>r.Name=="Has");
9
10//修改泛型约束
11varrGenericParameter=rMethod.GenericParameters[0];
12rGenericParameter.Constraints[0]=rAssembly.MainModule.Import(typeof(Enum));
13//修改最大栈大小(这个功能需要压栈两次)
14rMethod.Body.MaxStackSize=2;
15//清空方法体代码
16rMethod.Body.Instructions.Clear();
17//修改方法体代码
18varrProcessor=rMethod.Body.GetILProcessor();
19rProcessor.Append(rProcessor.Create(OpCodes.Ldarg_0));
20rProcessor.Append(rProcessor.Create(OpCodes.Ldarg_1));
21rProcessor.Append(rProcessor.Create(OpCodes.And));
22rProcessor.Append(rProcessor.Create(OpCodes.Ldarg_1));
23rProcessor.Append(rProcessor.Create(OpCodes.Ceq));
24rProcessor.Append(rProcessor.Create(OpCodes.Stloc_0));
25rProcessor.Append(rProcessor.Create(OpCodes.Ldloc_0));
26rProcessor.Append(rProcessor.Create(OpCodes.Ret));
27
28//保存修改
29rAssembly.Write("Moen.Base.dll");
30}

PS:修改部分的那串代码,是我先编译一个指定枚举的同样功能的方法,然后从反编译出的IL代码中得到的

最后

运行修改程序,类库就被修改了。查看方法的签名,就会像这样。C#编译器可是做不到呀

为什么约束那里除了Enum还有一个struct呢,因为我不想让已装箱的Enum对象在此方法中搞破坏(抛出异常),必须是实际的枚举类

PS:不要吐槽图中的Object2(如果你想知道真相的话,不过很简单)

下面是测试

这样问题解决了。(为了减少操作量,可以在类库的生成后事件中调用这个修改程序)

总结

Mono.Cecil可以很方便地使枚举类成为泛型约束条件,也能做到许多意想不到的东西。当然,我希望下一个版本的C#编译器能去除这个限制

Url: http://moen.cnblogs.com/archive/2011/07/14/use-mono-cecil-to-create-generic-extension-method-of-enum-class.html

优质内容筛选与推荐>>
1、[蓝桥杯历届试题] 海盗比酒量
2、多线程:C#线程同步lock,Monitor,Mutex,同步事件和等待句柄
3、透明度 css
4、ssh
5、ES6 语法笔记


长按二维码向我转账

受苹果公司新规定影响,微信 iOS 版的赞赏功能被关闭,可通过二维码转账支持公众号。

    阅读
    好看
    已推荐到看一看
    你的朋友可以在“发现”-“看一看”看到你认为好看的文章。
    已取消,“好看”想法已同步删除
    已推荐到看一看 和朋友分享想法
    最多200字,当前共 发送

    已发送

    朋友将在看一看看到

    确定
    分享你的想法...
    取消

    分享想法到看一看

    确定
    最多200字,当前共

    发送中

    网络异常,请稍后重试

    微信扫一扫
    关注该公众号





    联系我们

    欢迎来到TinyMind。

    关于TinyMind的内容或商务合作、网站建议,举报不良信息等均可联系我们。

    TinyMind客服邮箱:support@tinymind.net.cn