56,681
社区成员
发帖
与我相关
我的任务
分享
using System;
using Microsoft.VisualStudio.Data;
using System.Data.Common;
using Microsoft.VisualStudio.Data.AdoDotNet;
using System.Data;
namespace MySql.Data.VisualStudio
{
class StoredProcedureColumnEnumerator : DataObjectEnumerator
{
public override DataReader EnumerateObjects(string typeName, object[] items,
object[] restrictions, string sort, object[] parameters)
{
DbConnection conn = (DbConnection)Connection.GetLockedProviderObject();
try
{
string spName = String.Format("{0}.{1}", restrictions[1],restrictions[2]);
if (conn.State != ConnectionState.Open)
conn.Open();
string[] parmRest = new string[5];
parmRest[1] = (string)restrictions[1];
parmRest[2] = (string)restrictions[2];
parmRest[3] = (string)restrictions[3];
DataTable parmTable = conn.GetSchema("Procedure Parameters", parmRest);
DbCommand cmd = conn.CreateCommand();
cmd.CommandText = spName;
cmd.CommandType = CommandType.StoredProcedure;
foreach (DataRow row in parmTable.Rows)
{
if (row["ORDINAL_POSITION"].Equals(0)) continue;
DbParameter p = cmd.CreateParameter();
p.ParameterName = row["PARAMETER_NAME"].ToString();
p.Value = GetDefaultValue(row["DATA_TYPE"].ToString());
cmd.Parameters.Add(p);
}
using (IDataReader reader = cmd.ExecuteReader(CommandBehavior.SchemaOnly))
{
DataTable dt = reader.GetSchemaTable();
dt.Columns.Add(new DataColumn("RoutineName", typeof(string)));
foreach (DataRow row in dt.Rows)
{
row["RoutineName"] = restrictions[2];
string basedb = row["BaseSchemaName"] as string;
if (String.IsNullOrEmpty(basedb) || row["BaseSchemaName"] == DBNull.Value)
row["BaseSchemaName"] = cmd.Connection.Database;
}
return new AdoDotNetDataTableReader(dt);
}
}
finally
{
Connection.UnlockProviderObject();
}
}
private object GetDefaultValue(string dataType)
{
if (dataType == "VARCHAR" || dataType == "VARBINARY" ||
dataType == "ENUM" || dataType == "SET" || dataType == "CHAR")
return "";
return 0;
}
}
}