1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
|
// Copyright 2008, 2010, Oracle and/or its affiliates. All rights reserved.
//
// MySQL Connector/NET is licensed under the terms of the GPLv2
// <http://www.gnu.org/licenses/old-licenses/gpl-2.0.html>, like most
// MySQL Connectors. There are special exceptions to the terms and
// conditions of the GPLv2 as it is applied to this software, see the
// FLOSS License Exception
// <http://www.mysql.com/about/legal/licensing/foss-exception.html>.
//
// This program is free software; you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published
// by the Free Software Foundation; version 2 of the License.
//
// This program is distributed in the hope that it will be useful, but
// WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
// or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
// for more details.
//
// You should have received a copy of the GNU General Public License along
// with this program; if not, write to the Free Software Foundation, Inc.,
// 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA using System;
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();
if (dt == null)
dt = new DataTable();
else
{
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;
}
}
}
|
