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

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

 public static bool Has<TEnum>(this TEnum rpEnum, TEnum rpValue)
2     where TEnum : struct, Enum
3 {
4     return (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来写吧）

修改

先弄个伪方法：

 public static bool Has<TEnum>(this TEnum rpEnum, TEnum rpValue)
2     where TEnum : struct
3 {
4     return true;
5 }

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

下面是主要代码：

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