转: 用 System.Reflection.Emit 来自动生成调用存储过程的实现(接自别人的成果)

来源:互联网 发布:原油eia数据最新 编辑:程序博客网 时间:2024/05/10 18:19
//使用的例子
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
 
原创粉丝点击