用 System.Reflection.Emit 来自动生成调用储存过程的实现

用 System.Reflection.Emit 来自动生成调用储存过程的实现

- 中国WEB开发者网络 (http://www.webasp.net)
-- 技术教程 (http://www.webasp.net/article/)
--- 用 System.Reflection.Emit 来自动生成调用储存过程的实现 (http://www.webasp.net/article/16/15113.htm)

-- 作者：未知
-- 发布日期： 2004-11-26

/****************************************************************\
*
* 用 System.Reflection.Emit 来自动生成调用储存过程的实现！
*
* By http://lostinet.com
*
* Copyrights : Not-Reversed
*
\****************************************************************/

//使用的例子
namespace Lostinet.Sample
{
using System;
using System.Data;
using System.Data.SqlClient;
using System.Windows.Forms;

//定义一个接口，用于定义存储过程

interface INorthwindStoredProcedures
{
//定义存储过程对应的方法

DataSet CustOrderHist(string CustomerID);

//如果储存过程名字和方法名字不同，应该用SqlAccessAttribute来进行说明
[SqlAccess("Employee Sales By Country")]
DataTable EmployeeSalesByCountry(DateTime Beginning_Date,DateTime Ending_Date);

//...more...

//MORE Ideas..

//直接执行SQL语句？
//[SqlAccess(SqlAccessType.SqlQuery,"SELECT * FROM Employees WHERE EmployeeID=@EmpId")]
//DataTable SelectEmployee(int EmpId);
}

class ConsoleApplication
{
[STAThread]
static void Main(string[] args)
{
using(SqlConnection conn=new SqlConnection("server=(local);trusted_connection=true;database=northwind"))
{
//一句话就把实现创建了！
//需要传如 SqlConnection 和 SqlTransaction
//SqlTransaction可以为null

//这个好就好在，只要能得到SqlConnection/SqlTransaction就能用这个方法了，所以兼容 Lostinet.Data.SqlScope
INorthwindStoredProcedures nsp=(INorthwindStoredProcedures)
StoredProcedure.CreateStoredProcedureInterface(typeof(INorthwindStoredProcedures),conn,null);

//调用储存过程并且显示

ShowData("CustOrderHist ALFKI",nsp.CustOrderHist("ALFKI"));

ShowData("Employee Sales By Country",nsp.EmployeeSalesByCountry(new DateTime(1998,1,1),new DateTime(1999,1,1)));

}
}

static void ShowData(string title,object data)
{
Form f=new Form();
f.Width=600;
f.Height=480;
f.Text=title;

DataGrid grid=new DataGrid();
grid.Dock=DockStyle.Fill;
grid.DataSource=data;

f.Controls.Add(grid);
f.ShowDialog();
}

}
}

#region //实现方法（不完整）
namespace Lostinet.Sample
{
using System;
using System.Collections;
using System.Reflection;
using System.Reflection.Emit;
using System.Data;
using System.Data.SqlClient;

//这个类作为实现的基类，
//目的是提供储存 SqlConnection/SqlTransaction 和公用的一些方法
//这个类必须为public,否则无法继承
//但开发者不会显式访问这个类
public class SPInterfaceBase : IDisposable
{
public SPInterfaceBase()
{
}

public void Dispose()
{
}

//CreateStoredProcedureInterface会把相关的值SqlConnection/SqlTransaction存到这里
public SqlConnection connection;
public SqlTransaction transaction;

//创建一个SqlCommand
public SqlCommand CreateCommand(string spname)
{
SqlCommand cmd=new SqlCommand(spname,connection,transaction);
cmd.CommandType=CommandType.StoredProcedure;
//TODO:
//cmd.Parameters.Add("@ReturnValue",...
return cmd;
}

//由 Type 推算出 SqlDbType ，未完成
SqlDbType GetSqlDbType(Type type)
{
//TODO:switch(type)...

return SqlDbType.NVarChar;
}

//定义参数
public void DefineParameter(SqlCommand cmd,string name,Type type,ParameterDirection direction)
{
SqlParameter param=new SqlParameter("@"+name,GetSqlDbType(type));
param.Direction=direction;
cmd.Parameters.Add(param);
}

//在SqlCommand执行前设置参数值
public void SetParameter(SqlCommand cmd,string name,object value)
{
cmd.Parameters["@"+name].Value=(value==null?DBNull.Value:value);
}
//在SqlCommand执行后取得参数值
public object GetParameter(SqlCommand cmd,string name)
{
return cmd.Parameters["@"+name].Value;
}

//根据不同的返回值执行不同的操作

public SqlDataReader ExecuteDataReader(SqlCommand cmd)
{
return cmd.ExecuteReader();
}
public object ExecuteScalar(SqlCommand cmd)
{
return cmd.ExecuteScalar();
}
public void ExecuteNonQuery(SqlCommand cmd)
{
cmd.ExecuteNonQuery();
}
public DataSet ExecuteDataSet(SqlCommand cmd)
{
DataSet ds=new DataSet();
using(SqlDataAdapter sda=new SqlDataAdapter(cmd))
{
sda.Fill(ds);
}
return ds;
}
public DataTable ExecuteDataTable(SqlCommand cmd)
{
DataTable table=new DataTable();
using(SqlDataAdapter sda=new SqlDataAdapter(cmd))
{
sda.Fill(table);
}
return table;
}
public DataRow ExecuteDataRow(SqlCommand cmd)
{
DataTable table=ExecuteDataTable(cmd);
if(table.Rows.Count==0)
return null;
return table.Rows[0];
}
}

public class StoredProcedure
{
static public object CreateStoredProcedureInterface(Type interfaceType,SqlConnection connection,SqlTransaction transaction)
{
//检查参数
if(interfaceType==null)throw(new ArgumentNullException("interfaceType"));
if(!interfaceType.IsInterface)
throw(new ArgumentException("argument is not interface","interfaceType"));
if(connection==null)throw(new ArgumentNullException("connection"));
if(transaction!=null)
{
if(transaction.Connection!=connection)
throw(new ArgumentException("transaction.Connection!=connection","transaction"));
}

//创建StoredProcedure

StoredProcedure spemit=new StoredProcedure();
spemit.interfaceType=interfaceType;
spemit.connection=connection;
spemit.transaction=transaction;

//创建
return spemit.CreateInstance();
}

//用于储存已创建的类型
static Hashtable EmittedTypes=new Hashtable();

Type interfaceType;
SqlConnection connection;
SqlTransaction transaction;

private StoredProcedure()
{
}

object CreateInstance()
{
lock(interfaceType)
{
//如果没有创建具体的实现，则创建它

if(emittedType==null)
{
emittedType=(Type)EmittedTypes[interfaceType];

if(emittedType==null)
{
CreateType();

//储存已创建类型
EmittedTypes[interfaceType]=emittedType;
}
}
}

//创建具体的实例
SPInterfaceBase spi=(SPInterfaceBase)Activator.CreateInstance(emittedType);

//设置SqlConnection/SqlTransaction
spi.connection=connection;
spi.transaction=transaction;

return spi;
}

Type emittedType;

TypeBuilder typeBuilder;

//创建类型
void CreateType()
{
//创建 Assembly
//AssemblyBuilderAccess.Run-表示只用于运行，不在磁盘上保存
AssemblyName an=new AssemblyName();
an.Name="Assembly."+interfaceType.FullName+".Implementation";
AssemblyBuilder asmBuilder=AppDomain.CurrentDomain.DefineDynamicAssembly(an,AssemblyBuilderAccess.Run);

//创建Module
ModuleBuilder mdlBuilder=asmBuilder.DefineDynamicModule("Module."+interfaceType.FullName+".Implementation");

//创建Type，该类型继承 SPInterfaceBase
typeBuilder=mdlBuilder.DefineType(interfaceType.FullName+".Implementation",TypeAttributes.Class,typeof(SPInterfaceBase));

//实现所有的接口方法
EmitInterface(interfaceType);

//如果interfaceType是基于其他接口的
foreach(Type subinterface in interfaceType.GetInterfaces())
{
//IDisposable不需要实现，由SPInterfaceBase实现了
if(subinterface==typeof(IDisposable))
continue;

EmitInterface(subinterface);
}

emittedType=typeBuilder.CreateType();
}

void EmitInterface(Type type)
{
//实现接口
typeBuilder.AddInterfaceImplementation(type);

//列出接口的成员
foreach(MemberInfo member in type.GetMembers(BindingFlags.Instance|BindingFlags.Public))
{
//约定-成员必须是方法，不能有属性啊，事件之类的
if(member.MemberType!=MemberTypes.Method)
throw(new Exception("Could Not Emit "+member.MemberType+" Automatically!"));

//取得接口中定义的方法
MethodInfo method=(MethodInfo)member;

//计算新方法的属性，在原来方法的属性上复制过来，并且不是Public/Abstract，加上Private
MethodAttributes methodattrs=method.Attributes;
methodattrs&=~(MethodAttributes.Public|MethodAttributes.Abstract);
methodattrs|=MethodAttributes.Private;

ParameterInfo[] paramInfos=method.GetParameters();
int paramlength=paramInfos.Length;

//取得参数的类型数组
Type[] paramTypes=new Type[paramlength];
for(int i=0;i<paramlength;i++)
{
paramTypes[i]=paramInfos[i].ParameterType;
}

//在typeBuilder上建立新方法，参数类型与返回类型都与接口上的方法一致
MethodBuilder mthBuilder=typeBuilder.DefineMethod(method.Name,methodattrs,method.CallingConvention,method.ReturnType,paramTypes);

//复制新方法上的参数的名字和属性
for(int i=0;i<paramlength;i++)
{
ParameterInfo pi=paramInfos[i];
//对于Instance,参数position由1开始
mthBuilder.DefineParameter(i+1,pi.Attributes,pi.Name);
}

//指定新方法是实现接口的方法的。
typeBuilder.DefineMethodOverride(mthBuilder,method);

//在类型上定义一个字段，这个字段用于储存被方法使用的SqlCommand
FieldBuilder field_cmd=typeBuilder.DefineField("_cmd_"+method.Name,typeof(SqlCommand),FieldAttributes.Private);

//ILGenerator 是用于生成实现代码的对象
ILGenerator ilg=mthBuilder.GetILGenerator();

//定义临时变量
LocalBuilder local_res=ilg.DeclareLocal(typeof(object));

//定义一个用于跳转的Label
Label label_cmd_ready=ilg.DefineLabel();

//this._cmd_MethodName
ilg.Emit(OpCodes.Ldarg_0); //this
ilg.Emit(OpCodes.Ldfld,field_cmd);//._cmd_MethodName

//if(this._cmd_MethodName!=null) 跳到 label_cmd_ready
ilg.Emit(OpCodes.Brtrue,label_cmd_ready);

//如果this._cmd_MethodName为null，则运行下面代码来创建SqlCommand

//this._cmd_MethodName=this.CreateCommand("MethodName");
ilg.Emit(OpCodes.Ldarg_0);

//this.CreateCommand
ilg.Emit(OpCodes.Ldarg_0);//参数0
ilg.Emit(OpCodes.Ldstr,SqlAccessAttribute.GetSPName(method));//参数1
//调用
ilg.Emit(OpCodes.Callvirt,typeof(SPInterfaceBase).GetMethod("CreateCommand",BindingFlags.Instance|BindingFlags.Public));

ilg.Emit(OpCodes.Stfld,field_cmd);// ._cmd_MethodName=

//this.DefineParameter(...)
if(paramlength!=0)
{
//取得DefineParameter的引用
MethodInfo method_DefineParameter=typeof(SPInterfaceBase).GetMethod("DefineParameter",BindingFlags.Instance|BindingFlags.Public);

for(int i=0;i<paramlength;i++)
{
//取得各参数
ParameterInfo pi=paramInfos[i];

//this.DefineParameter(this._cmd_MethodName,"ParameterName",typeof(ParameterType),ParameterDirection.Xxx);

//参数0 - this
ilg.Emit(OpCodes.Ldarg_0);

//参数1 - this._cmd_MethodName
ilg.Emit(OpCodes.Ldarg_0);
ilg.Emit(OpCodes.Ldfld,field_cmd);

//参数2 - "ParameterName"
ilg.Emit(OpCodes.Ldstr,pi.Name);

//参数3 - typeof(ParameterType)
ilg.Emit(OpCodes.Ldtoken,pi.ParameterType);

//参数4 - ParameterDirection.Xxx
if(pi.ParameterType.IsByRef)
{
ilg.Emit(OpCodes.Ldc_I4,(int)ParameterDirection.InputOutput);
}
else if(pi.IsOut)
{
ilg.Emit(OpCodes.Ldc_I4,(int)ParameterDirection.Output);
}
else
{
ilg.Emit(OpCodes.Ldc_I4,(int)ParameterDirection.Input);
}

//调用DefineParameter
ilg.Emit(OpCodes.Callvirt,method_DefineParameter);
}
}
//到这里 _cmd_CommandName 已经 OK 了。

//设置label_cmd_ready就指这里
ilg.MarkLabel(label_cmd_ready);

//cmd!=null now.

if(paramlength!=0)
{
//现在要把方法的参数的值设置到SqlParameter上

MethodInfo method_SetParameter=typeof(SPInterfaceBase).GetMethod("SetParameter",BindingFlags.Instance|BindingFlags.Public);

for(int i=0;i<paramlength;i++)
{
ParameterInfo pi=paramInfos[i];

//如果参数是 out 的，则不需要设置
if(!pi.ParameterType.IsByRef&&pi.IsOut)
continue;

//this.SetParameter(this._cmd_MethodName,"ParameterName",ParameterName);

ilg.Emit(OpCodes.Ldarg_0);

ilg.Emit(OpCodes.Ldarg_0);
ilg.Emit(OpCodes.Ldfld,field_cmd);

ilg.Emit(OpCodes.Ldstr,pi.Name);

//取得参数值，如果参数为ValueType，则Box到Object
ilg.Emit(OpCodes.Ldarg,i+1);
if(pi.ParameterType.IsValueType)
ilg.Emit(OpCodes.Box,pi.ParameterType);

ilg.Emit(OpCodes.Callvirt,method_SetParameter);
}
}

//现在要执行储存过程（执行SqlCommand）了

//这里根据返回值类型判断怎样执行SqlCommand

Type returnType=method.ReturnType;

//如果是 void 的，则不需要返回值
bool nores=returnType==typeof(void);

MethodInfo method_Execute=null;

if(nores)
{
//不需要返回值
method_Execute=typeof(SPInterfaceBase).GetMethod("ExecuteNonQuery",BindingFlags.Instance|BindingFlags.Public);
}
else if(returnType==typeof(object))
{
//返回object
method_Execute=typeof(SPInterfaceBase).GetMethod("ExecuteScalar",BindingFlags.Instance|BindingFlags.Public);
}
else if(returnType==typeof(DataSet))
{
//返回DataSet
method_Execute=typeof(SPInterfaceBase).GetMethod("ExecuteDataSet",BindingFlags.Instance|BindingFlags.Public);
}
else if(returnType==typeof(DataTable))
{
//返回DataTable
method_Execute=typeof(SPInterfaceBase).GetMethod("ExecuteDataTable",BindingFlags.Instance|BindingFlags.Public);
}
else if(returnType==typeof(DataRow))
{
//返回DataRow
method_Execute=typeof(SPInterfaceBase).GetMethod("ExecuteDataRow",BindingFlags.Instance|BindingFlags.Public);
}
else
{
//返回其他类型
foreach(Type retInterface in returnType.GetInterfaces())
{
//如果是返回IDataReader
if(retInterface==typeof(IDataReader))
{
//只支持SqlDataReader
if(!returnType.IsAssignableFrom(typeof(SqlDataReader)))
throw(new Exception("SqlDataReader could not convert to "+returnType.FullName));

method_Execute=typeof(SPInterfaceBase).GetMethod("ExecuteDataReader",BindingFlags.Instance|BindingFlags.Public);
break;
}
}
}

//如果找不到适合的策略，
if(method_Execute==null)
{
//TODO:当然，这里应该有返回Int32,String,...的，不过懒得再写了。

//抛出异常，提示不支持该返回类型，要作者改改：）
throw(new NotSupportedException("NotSupport ReturnType:"+returnType.FullName));
}

//this.ExecuteXXX(this._cmd_MethodName)
ilg.Emit(OpCodes.Ldarg_0);
ilg.Emit(OpCodes.Ldarg_0);
ilg.Emit(OpCodes.Ldfld,field_cmd);
ilg.Emit(OpCodes.Callvirt,method_Execute);

//如果有返回值的，则是
//local_res=this.ExecuteXXX(this._cmd_MethodName)
if(!nores)
{
if(returnType.IsValueType)
ilg.Emit(OpCodes.Box,returnType);
ilg.Emit(OpCodes.Stloc,local_res);
}

if(paramlength!=0)
{
//这里处理ref/out的参数
MethodInfo method_GetParameter=typeof(SPInterfaceBase).GetMethod("GetParameter",BindingFlags.Instance|BindingFlags.Public);

for(int i=0;i<paramlength;i++)
{
ParameterInfo pi=paramInfos[i];

//如果不是ref/out则跳过
if(!pi.ParameterType.IsByRef&&!pi.IsOut)
continue;

//ParameterName=this.GetParameter(this._cmd_Methodname,"ParameterName")
ilg.Emit(OpCodes.Ldarg_0);

ilg.Emit(OpCodes.Ldarg_0);
ilg.Emit(OpCodes.Ldfld,field_cmd);

ilg.Emit(OpCodes.Ldstr,pi.Name);

ilg.Emit(OpCodes.Callvirt,method_GetParameter);

//如果类型是值类型，则需要 Unbox
if(pi.ParameterType.IsValueType)
ilg.Emit(OpCodes.Unbox,pi.ParameterType);

ilg.Emit(OpCodes.Starg,i+1);
}
}

//如果是 void , 则直接 return;
//否者是 return local_res ，如果返回值类型是ValueType，则需要Unbox
if(!nores)
{
ilg.Emit(OpCodes.Ldloc,local_res);
if(returnType.IsValueType)
ilg.Emit(OpCodes.Unbox,returnType);
}
ilg.Emit(OpCodes.Ret);

// //throw(new NotImplementedException());
// ilg.Emit(OpCodes.Newobj,typeof(NotImplementedException).GetConstructor(new Type[0]));
// ilg.Emit(OpCodes.Throw);

}
}
}

public enum SqlAccessType
{
StoredProcedure
//TODO:
//,SqlQuery
}

[AttributeUsage(AttributeTargets.Method)]
public class SqlAccessAttribute : Attribute
{
string _sp;

public SqlAccessAttribute(string spname)
{
_sp=spname;
}

public string StoreProcedure
{
get
{
return _sp;
}
}

static public string GetSPName(MethodInfo method)
{
if(method==null)throw(new ArgumentNullException("method"));

object[] attrs=method.GetCustomAttributes(typeof(SqlAccessAttribute),false);
if(attrs==null||attrs.Length==0)
return method.Name;

return ((SqlAccessAttribute)attrs[0]).StoreProcedure;
}

//TODO:
// public SqlAccessAttribute(SqlAccessType type,string text)
// {
//
// }
}

}
#endregion

webasp.net