C#mysql辅助类 mysqlhelper

public class MSqlHelper: IDisposable
{
private string ConnectionString = "";
/// <summary>
/// 批量操作每批次记录数
/// </summary>
public static int BatchSize = 2000;

/// <summary>
/// 超时时间
/// </summary>
public static int CommandTimeOut = 600;
public SqlHelper(string connectionString = null)
{
if (string.IsNullOrWhiteSpace(connectionString))
ConnectionString = AppSettings.Configuration["DbConnection:ConnectionString"];
else
ConnectionString = connectionString;
}
private MySqlConnection connection { get; set; }
#region ExecuteNonQuery

/// <summary>
/// 执行SQL语句,返回影响的行数
/// </summary>
/// <param name="commandType">命令类型(存储过程,命令文本, 其它.)</param>
/// <param name="commandText">SQL语句或存储过程名称</param>
/// <param name="parms">参数</param>
/// <returns>返回影响的行数</returns>
public int ExecuteNonQuery(string commandText, CommandType commandType = CommandType.Text, MySqlParameter[] parms=null)
{
try
{
using (MySqlConnection connection = GetConnection())
{
Open();

using (var command = GetCommand(connection, commandType, commandText, CommandTimeOut, parms))
{
int retval = command.ExecuteNonQuery();
command.Parameters.Clear();
return retval;

}
}
}
finally
{
Dispose();
}
}
#endregion ExecuteNonQuery
#region ExecuteDataTable

/// <summary>
/// 执行SQL语句,返回结果集中的数据表
/// </summary>
/// <param name="commandType">命令类型(存储过程,命令文本, 其它.)</param>
/// <param name="commandText">SQL语句或存储过程名称</param>
/// <param name="parms">查询参数</param>
/// <returns>返回结果集中的第一个数据表</returns>
public DataTable QueryDataTable(string commandText, CommandType commandType = CommandType.Text, MySqlParameter[] parms = null)
{
try
{
using (MySqlConnection connection = GetConnection())
{
Open();
using (var command = GetCommand(connection, commandType, commandText, CommandTimeOut, parms))
{

using (var adapter = new MySqlDataAdapter(command))
{
using (var ds = new DataSet())
{
adapter.Fill(ds);
if (commandText.IndexOf("@") > 0)
{
commandText = commandText.ToLower();
int index = commandText.IndexOf("where ");
if (index < 0)
{
index = commandText.IndexOf("\nwhere");
}
if (index > 0)
{
ds.ExtendedProperties.Add("SQL", commandText.Substring(0, index - 1)); //将获取的语句保存在表的一个附属数组里,方便更新时生成CommandBuilder
}
else
{
ds.ExtendedProperties.Add("SQL", commandText); //将获取的语句保存在表的一个附属数组里,方便更新时生成CommandBuilder
}
}
else
{
ds.ExtendedProperties.Add("SQL", commandText); //将获取的语句保存在表的一个附属数组里,方便更新时生成CommandBuilder
}

foreach (DataTable dt in ds.Tables)
{
dt.ExtendedProperties.Add("SQL", ds.ExtendedProperties["SQL"]);
}

command.Parameters.Clear();
return ds.Tables[0];
}
}
}
}

}
finally
{
Dispose();
}
}


/// <summary>
/// 查询分页(必须带ID参数)
/// </summary>
/// <param name="sql">sql语句</param>
/// <param name="page">当前页</param>
/// <param name="pageSize">条数</param>
/// <param name="orderby">排序</param>
/// <param name="total">返回总条数</param>
/// <param name="parms">参数</param>
/// <returns></returns>
public DataTable QueryPageList(string sql, int page, int pageSize, string orderby, ref int total, params MySqlParameter[] parms)
{
try
{
using (MySqlConnection connection = GetConnection())
{
string cmdtext = "select * from (" + sql + ") a inner join (select id from (" + sql + ") b order by " + orderby + " limit " + (page - 1) * pageSize + "," + pageSize + ") c using(id) ";

Open();
using (var command = GetCommand(connection, CommandType.Text, cmdtext, CommandTimeOut, parms))
{
total = Count(connection, sql);
using (var adapter = new MySqlDataAdapter(command))
{

using (var ds = new DataSet())
{
adapter.Fill(ds);
if (sql.IndexOf("@") > 0)
{
sql = sql.ToLower();
int index = sql.IndexOf("where ");
if (index < 0)
{
index = sql.IndexOf("\nwhere");
}
if (index > 0)
{
ds.ExtendedProperties.Add("SQL", sql.Substring(0, index - 1)); //将获取的语句保存在表的一个附属数组里,方便更新时生成CommandBuilder
}
else
{
ds.ExtendedProperties.Add("SQL", sql); //将获取的语句保存在表的一个附属数组里,方便更新时生成CommandBuilder
}
}
else
{
ds.ExtendedProperties.Add("SQL", sql); //将获取的语句保存在表的一个附属数组里,方便更新时生成CommandBuilder
}

foreach (DataTable dt in ds.Tables)
{
dt.ExtendedProperties.Add("SQL", ds.ExtendedProperties["SQL"]);
}

command.Parameters.Clear();
return ds.Tables[0];
}
}
}
}
}
finally
{
Dispose();
}
}

 


#endregion ExecuteDataTable
#region private function
/// <summary>
/// 获取连接
/// </summary>
/// <returns></returns>
private MySqlConnection GetConnection()
{
return GetConnection(ConnectionString);
}
private MySqlConnection GetConnection(string ConnectionString)
{
if (connection == null)
{
connection = new MySqlConnection(ConnectionString);
}
return connection;
}
/// <summary>
/// 打开连接
/// </summary>
public void Open()
{
CheckConnection();
}
/// <summary>
/// 关闭连接
/// </summary>
public void Close()
{
if (this.connection != null && this.connection.State == ConnectionState.Open)
{
this.connection.Close();
}
}
/// <summary>
/// 检查连接状态并打开连接
/// </summary>
private void CheckConnection()
{
if (this.connection.State != ConnectionState.Open)
{
try
{
this.connection.Open();
}
catch (Exception ex)
{
throw ex;
}
}
}
/// <summary>
/// 释放资源
/// </summary>
public void Dispose()
{

Close();
if (this.connection != null)
{
this.connection.Dispose();
}
this.connection = null;
}
/// <summary>
/// 获取command
/// </summary>
/// <param name="connection"></param>
/// <param name="commandType"></param>
/// <param name="commandText"></param>
/// <param name="commandTimeOut"></param>
/// <param name="parms"></param>
/// <returns></returns>
private MySqlCommand GetCommand(MySqlConnection connection, CommandType commandType, string commandText, int commandTimeOut, params MySqlParameter[] parms)
{
var mysqlCmd = new MySqlCommand()
{
CommandType = commandType,
CommandText = commandText,
Connection = connection,
CommandTimeout = commandTimeOut
};
mysqlCmd.Parameters.Clear();
if (parms != null && parms.Length > 0)
{
//预处理MySqlParameter参数数组,将为NULL的参数赋值为DBNull.Value;
foreach (MySqlParameter parameter in parms)
{
if ((parameter.Direction == ParameterDirection.InputOutput || parameter.Direction == ParameterDirection.Input) && (parameter.Value == null))
{
parameter.Value = DBNull.Value;
}
}
mysqlCmd.Parameters.AddRange(parms);
}
return mysqlCmd;
}
private int Count(MySqlConnection connection, string commandText)
{
string sql = "select count(*) from (" + commandText + ") a ";
MySqlCommand command = GetCommand(connection, CommandType.Text, sql, CommandTimeOut);
return Convert.ToInt32(command.ExecuteScalar());
}
/// <summary>
/// 执行查询sql,用于自定义事务处理,此处不会释放和断开连接
/// </summary>
/// <param name="connection">(MySqlConnection)数据库连接</param>
/// <param name="sql">执行的sql语句</param>
/// <returns>受影响的行数</returns>
public int ExecuteTransNonQuery(MySqlConnection connection, string sql)
{
MySqlCommand command = GetCommand(connection, CommandType.Text, sql, CommandTimeOut);
return command.ExecuteNonQuery();
}

/// <summary>
/// 执行查询sql返回datatable,用于自定义事务处理,此处不会释放和断开连接
/// </summary>
/// <param name="connection">(MySqlConnection)数据库连接</param>
/// <param name="sql">执行的sql语句</param>
/// <returns>返回datatable</returns>
public DataTable ExecuteTransDataTable(MySqlConnection connection, string sql)
{
MySqlCommand command = GetCommand(connection, CommandType.Text, sql, CommandTimeOut);
var adapter = new MySqlDataAdapter(command);
DataSet ds = new DataSet();
adapter.Fill(ds);
return ds.Tables[0];
}
#endregion
#region 批量操作

/// <summary>
/// 批量更新数据
/// </summary>
/// <param name="table">数据表</param>
public int BatchUpdate(DataTable table)
{
var res = 0;
using (var connection = GetConnection())
{
using (var command = connection.CreateCommand())
{
command.CommandTimeout = CommandTimeOut;
command.CommandType = CommandType.Text;
using (var adapter = new MySqlDataAdapter(command))
{
using (MySqlCommandBuilder commandBulider = new MySqlCommandBuilder(adapter))
{
commandBulider.ConflictOption = ConflictOption.OverwriteChanges;

MySqlTransaction transaction = null;
try
{
Open();
transaction = connection.BeginTransaction();
//设置批量更新的每次处理条数
adapter.UpdateBatchSize = (table.Rows.Count + 1);
//设置事物
adapter.SelectCommand.Transaction = transaction;

if (table.ExtendedProperties["SQL"] != null)
{
adapter.SelectCommand.CommandText = table.ExtendedProperties["SQL"].ToString();
}
res= adapter.Update(table);
transaction.Commit();/////提交事务
}
catch (MySqlException ex)
{
if (transaction != null) transaction.Rollback();
throw ex;
}
finally
{
Dispose();
}
}
}

}
}

return res;
}

/// <summary>
///大批量数据插入,返回成功插入行数
/// </summary>
/// <param name="table">数据表</param>
/// <returns>返回成功插入行数</returns>
public int BulkInsert(DataTable table)
{
if (string.IsNullOrEmpty(table.TableName)) throw new Exception("请给DataTable的TableName属性附上表名称");
if (table.Rows.Count == 0) return 0;
int insertCount = 0;
string tmpPath = Path.GetTempFileName();
string csv = DataTableToCsv(table);
File.WriteAllText(tmpPath, csv);
using (MySqlConnection conn = GetConnection())
{
MySqlTransaction tran = null;
try
{
Open();
tran = conn.BeginTransaction();
MySqlBulkLoader bulk = new MySqlBulkLoader(conn)
{
FieldTerminator = ",",
FieldQuotationCharacter = '"',
EscapeCharacter = '"',
LineTerminator = "\r\n",
FileName = tmpPath,
NumberOfLinesToSkip = 0,
TableName = table.TableName,
};
bulk.Columns.AddRange(table.Columns.Cast<DataColumn>().Select(colum => colum.ColumnName).ToList());
insertCount = bulk.Load();
tran.Commit();
}
catch (MySqlException ex)
{
if (tran != null) tran.Rollback();
throw ex;
}
finally
{
Dispose();
}
}
File.Delete(tmpPath);
return insertCount;
}

#endregion 批量操作
/// <summary>
///将DataTable转换为标准的CSV
/// </summary>
/// <param name="table">数据表</param>
/// <returns>返回标准的CSV</returns>
private static string DataTableToCsv(DataTable table)
{
//以半角逗号(即,)作分隔符,列为空也要表达其存在。
//列内容如存在半角逗号(即,)则用半角引号(即"")将该字段值包含起来。
//列内容如存在半角引号(即")则应替换成半角双引号("")转义,并用半角引号(即"")将该字段值包含起来。
StringBuilder sb = new StringBuilder();
DataColumn colum;
foreach (DataRow row in table.Rows)
{
for (int i = 0; i < table.Columns.Count; i++)
{
colum = table.Columns[i];
if (i != 0) sb.Append(",");
if (colum.DataType == typeof(string) && row[colum].ToString().Contains(","))
{
sb.Append("\"" + row[colum].ToString().Replace("\"", "\"\"") + "\"");
}
else sb.Append(row[colum].ToString());
}
sb.AppendLine();
}

return sb.ToString();
}
}

posted @ 2021-10-21 13:43  龙丶谈笑风声  阅读(1242)  评论(0编辑  收藏  举报