-
Notifications
You must be signed in to change notification settings - Fork 33
/
Copy pathMySQLDatabase.cs
225 lines (203 loc) · 7.48 KB
/
MySQLDatabase.cs
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
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
using MySql.Data.MySqlClient;
using System;
using System.Collections.Generic;
using System.Configuration;
using System.Data;
using System.Threading;
namespace AspNet.Identity.MySQL
{
/// <summary>
/// Class that encapsulates a MySQL database connections
/// and CRUD operations
/// </summary>
public class MySQLDatabase : IDisposable
{
private MySqlConnection _connection;
/// Default constructor which uses the "DefaultConnection" connectionString
/// </summary>
public MySQLDatabase()
: this("DefaultConnection")
{
}
/// <summary>
/// Constructor which takes the connection string name
/// </summary>
/// <param name="connectionStringName"></param>
public MySQLDatabase(string connectionStringName)
{
string connectionString = ConfigurationManager.ConnectionStrings[connectionStringName].ConnectionString;
_connection = new MySqlConnection(connectionString);
}
/// <summary>
/// Executes a non-query MySQL statement
/// </summary>
/// <param name="commandText">The MySQL query to execute</param>
/// <param name="parameters">Optional parameters to pass to the query</param>
/// <returns>The count of records affected by the MySQL statement</returns>
public int Execute(string commandText, Dictionary<string, object> parameters)
{
int result = 0;
if (String.IsNullOrEmpty(commandText))
{
throw new ArgumentException("Command text cannot be null or empty.");
}
try
{
EnsureConnectionOpen();
var command = CreateCommand(commandText, parameters);
result = command.ExecuteNonQuery();
}
finally
{
_connection.Close();
}
return result;
}
/// <summary>
/// Executes a MySQL query that returns a single scalar value as the result.
/// </summary>
/// <param name="commandText">The MySQL query to execute</param>
/// <param name="parameters">Optional parameters to pass to the query</param>
/// <returns></returns>
public object QueryValue(string commandText, Dictionary<string, object> parameters)
{
object result = null;
if (String.IsNullOrEmpty(commandText))
{
throw new ArgumentException("Command text cannot be null or empty.");
}
try
{
EnsureConnectionOpen();
var command = CreateCommand(commandText, parameters);
result = command.ExecuteScalar();
}
finally
{
EnsureConnectionClosed();
}
return result;
}
/// <summary>
/// Executes a SQL query that returns a list of rows as the result.
/// </summary>
/// <param name="commandText">The MySQL query to execute</param>
/// <param name="parameters">Parameters to pass to the MySQL query</param>
/// <returns>A list of a Dictionary of Key, values pairs representing the
/// ColumnName and corresponding value</returns>
public List<Dictionary<string, string>> Query(string commandText, Dictionary<string, object> parameters)
{
List<Dictionary<string, string>> rows = null;
if (String.IsNullOrEmpty(commandText))
{
throw new ArgumentException("Command text cannot be null or empty.");
}
try
{
EnsureConnectionOpen();
var command = CreateCommand(commandText, parameters);
using (MySqlDataReader reader = command.ExecuteReader())
{
rows = new List<Dictionary<string, string>>();
while (reader.Read())
{
var row = new Dictionary<string, string>();
for (var i = 0; i < reader.FieldCount; i++)
{
var columnName = reader.GetName(i);
var columnValue = reader.IsDBNull(i) ? null : reader.GetString(i);
row.Add(columnName, columnValue);
}
rows.Add(row);
}
}
}
finally
{
EnsureConnectionClosed();
}
return rows;
}
/// <summary>
/// Opens a connection if not open
/// </summary>
private void EnsureConnectionOpen()
{
var retries = 3;
if (_connection.State == ConnectionState.Open)
{
return;
}
else
{
while (retries >= 0 && _connection.State != ConnectionState.Open)
{
_connection.Open();
retries--;
Thread.Sleep(30);
}
}
}
/// <summary>
/// Closes a connection if open
/// </summary>
public void EnsureConnectionClosed()
{
if (_connection.State == ConnectionState.Open)
{
_connection.Close();
}
}
/// <summary>
/// Creates a MySQLCommand with the given parameters
/// </summary>
/// <param name="commandText">The MySQL query to execute</param>
/// <param name="parameters">Parameters to pass to the MySQL query</param>
/// <returns></returns>
private MySqlCommand CreateCommand(string commandText, Dictionary<string, object> parameters)
{
MySqlCommand command = _connection.CreateCommand();
command.CommandText = commandText;
AddParameters(command, parameters);
return command;
}
/// <summary>
/// Adds the parameters to a MySQL command
/// </summary>
/// <param name="commandText">The MySQL query to execute</param>
/// <param name="parameters">Parameters to pass to the MySQL query</param>
private static void AddParameters(MySqlCommand command, Dictionary<string, object> parameters)
{
if (parameters == null)
{
return;
}
foreach (var param in parameters)
{
var parameter = command.CreateParameter();
parameter.ParameterName = param.Key;
parameter.Value = param.Value ?? DBNull.Value;
command.Parameters.Add(parameter);
}
}
/// <summary>
/// Helper method to return query a string value
/// </summary>
/// <param name="commandText">The MySQL query to execute</param>
/// <param name="parameters">Parameters to pass to the MySQL query</param>
/// <returns>The string value resulting from the query</returns>
public string GetStrValue(string commandText, Dictionary<string, object> parameters)
{
string value = QueryValue(commandText, parameters) as string;
return value;
}
public void Dispose()
{
if (_connection != null)
{
_connection.Dispose();
_connection = null;
}
}
}
}