标签:
近期工作中发现批量插入的方法需求越来越大。所以在ORM中增加了MYSQL的批量插入方法。由三个方法组成,可以使用在不同情况下。
1、根据传入的实体集合生成批量插入的SQL语名 GetInsertSqlBatch()
2、在方法1的基础上增加一个执行并返回是否成功的功能 ExecuteInsertModelBatch()。
3、上面两个方法都没有控制每次批量插入的最大数量。只适用于小量批量插入情况。如果实体集合一次性传入1万,10万也做一次提交的话好像不太合适吧。所以就有了第三个方法。第三个方法是第二个方法的重载,增加了<param name="batchNum">每个批量插入的数量</param>参数。用于控制每个批次最大插入数量。转入0时使用默认值为100。当插入实体集合大于100或指定的值时,将根据设定的batchNum值的来分批提交。
下面是具体代码的实现。
/// <summary>
/// 获取批量插入SQL
/// </summary>
/// <param name="entitys">插入实体集合</param>
/// <param name="sql">输出SQL</param>
/// <param name="tableName">表名</param>
/// <param name="excludeProperties">过滤属性名称列表</param>
/// <returns>SQL参数集合</returns>
public static List<MySqlParameter> GetInsertSqlBatch<T>(List<T> entitys, out string sql, string tableName, params string[] excludeProperties)
{
List<MySqlParameter> sqlParameters = new List<MySqlParameter>();
sql = string.Empty;
if (entitys != null && entitys.Count > 0)
{
var type = entitys.First().GetType();
tableName = GetTableName(type, tableName);
//INSERT INTO `insert_table` (`datetime`, `uid`, `content`, `type`)
// VALUES (‘0‘, ‘userid_0‘, ‘content_0‘, 0),
// (‘1‘, ‘userid_1‘, ‘content_1‘, 1);
if (!string.IsNullOrEmpty(tableName))
{
StringBuilder InsertSql = new StringBuilder(string.Format("Insert into {0}(", tableName.ToLower()));
StringBuilder ValuesSql = new StringBuilder(" Values(");
var propertieList = GetProperties(type);
var entityItemIndex = 1;
try
{
//循环要插入的实体集合
foreach (var entityItem in entitys)
{
//循环每个实体的属性集合
var propertieIndex = 0;
foreach (var propertieItem in propertieList.Values)
{
//是否添加属性
if (ValidateIsAddPropertie(type, propertieItem, OperationType.Insert, paramsArrayToDictionary(excludeProperties)))
{
if (entityItemIndex == 1)
{
InsertSql.AppendFormat("{0},", propertieItem.Name);
}
var parameterName = string.Format("@{0}{1}", propertieItem.Name, entityItemIndex);
if (entityItemIndex > 1 && propertieIndex == 0)
{
ValuesSql.Append("(");
}
ValuesSql.AppendFormat("{0},", parameterName);
var value = propertieItem.GetValue(entityItem);
sqlParameters.Add(new MySqlParameter(parameterName, value ?? string.Empty));
}
if (propertieIndex == propertieList.Count - 1)
{
if (entityItemIndex == 1)
{
InsertSql.Remove(InsertSql.Length - 1, 1);
InsertSql.Append(")");
}
ValuesSql.Remove(ValuesSql.Length - 1, 1);
ValuesSql.Append("),");
}
propertieIndex++;
}
entityItemIndex++;
}
}
catch (Exception ex)
{
if (sqlParameters != null && sqlParameters.Count > 0)
{
sqlParameters.Clear();
sqlParameters = null;
}
throw ex;
}
sql = string.Format("{0};", InsertSql.Append(ValuesSql).ToString().Trim(‘,‘));
InsertSql.Clear();
ValuesSql.Clear();
}
}
return sqlParameters;
}
/// <summary>
/// 执行批量插入实体方法
/// </summary>
/// <param name="entitys">插入实体集合</param>
/// <param name="dataBaseName">数据名称</param>
/// <param name="tableName">表名</param>
/// <param name="excludeProperties">过滤属性名称列表</param>
/// <returns>SQL参数集合</returns>
public static bool ExecuteInsertModelBatch<T>(List<T> entitys, string dataBaseName, string tableName = "", params string[] excludeProperties)
{
bool isSucceed = false;
List<MySqlParameter> sqlParameters = new List<MySqlParameter>();
string sql = string.Empty;
sqlParameters = GetInsertSqlBatch(entitys, out sql, tableName, excludeProperties);
if (sql.Length > 0 && sqlParameters != null && sqlParameters.Count > 0)
{
try
{
isSucceed = CBDMySqlHelper.ExecuteNonQuery(dataBaseName, sql, sqlParameters.ToArray()) == entitys.Count;
}
catch (Exception ex)
{
throw ex;
}
finally
{
if (sqlParameters != null && sqlParameters.Count > 0)
{
sqlParameters.Clear();
sqlParameters = null;
}
if (entitys != null && entitys.Count > 0)
{
entitys.Clear();
entitys = null;
}
}
}
return isSucceed;
}
/// <summary>
/// 执行批量插入实体方法
/// </summary>
/// <param name="entitys">插入实体集合</param>
/// <param name="batchNum">分批处理数量[<=0时默认值为100]</param>
/// <param name="batchNum">每个批量插入的数量</param>
/// <param name="dataBaseName">数据名称</param>
/// <param name="tableName">表名</param>
/// <param name="excludeProperties">过滤属性名称列表</param>
/// <returns>SQL参数集合</returns>
public static bool ExecuteInsertModelBatch<T>(List<T> entitys, int batchNum, string dataBaseName, string tableName = "", params string[] excludeProperties)
{
bool isSucceed = false;
var insertCount = 0;
List<MySqlParameter> sqlParameters = new List<MySqlParameter>();
if (entitys != null && entitys.Count > 0)
{
//如果分批处理数量为0时,则使用默认值100
if (batchNum <= 0)
{
batchNum = 100;
}
var type = entitys.First().GetType();
tableName = GetTableName(type, tableName);
//INSERT INTO `insert_table` (`datetime`, `uid`, `content`, `type`)
// VALUES (‘0‘, ‘userid_0‘, ‘content_0‘, 0),
// (‘1‘, ‘userid_1‘, ‘content_1‘, 1);
if (!string.IsNullOrEmpty(tableName))
{
StringBuilder InsertSql = new StringBuilder(string.Format("Insert into {0}(", tableName.ToLower()));
StringBuilder ValuesSql = new StringBuilder(" Values(");
var propertieList = GetProperties(type);
var entityItemIndex = 1;
try
{
//循环要插入的实体集合
foreach (var entityItem in entitys)
{
//循环每个实体的属性集合
var propertieIndex = 0;
foreach (var propertieItem in propertieList.Values)
{
//是否添加属性
if (ValidateIsAddPropertie(type, propertieItem, OperationType.Insert, paramsArrayToDictionary(excludeProperties)))
{
if (entityItemIndex == 1)
{
InsertSql.AppendFormat("{0},", propertieItem.Name);
}
var parameterName = string.Format("@{0}{1}", propertieItem.Name, entityItemIndex);
if (entityItemIndex > 1 && propertieIndex == 0)
{
ValuesSql.Append("(");
}
ValuesSql.AppendFormat("{0},", parameterName);
var value = propertieItem.GetValue(entityItem);
sqlParameters.Add(new MySqlParameter(parameterName, value ?? string.Empty));
}
if (propertieIndex == propertieList.Count - 1)
{
if (entityItemIndex == 1)
{
InsertSql.Remove(InsertSql.Length - 1, 1);
InsertSql.Append(")");
}
ValuesSql.Remove(ValuesSql.Length - 1, 1);
ValuesSql.Append("),");
}
propertieIndex++;
}
if ((entityItemIndex % batchNum) == 0 || entityItemIndex == entitys.Count)
{
var sql = string.Format("{0}{1}", InsertSql, ValuesSql).Trim(‘,‘);
insertCount += CBDMySqlHelper.ExecuteNonQuery(dataBaseName, sql, sqlParameters.ToArray());
ValuesSql = new StringBuilder(" Values");
sqlParameters.Clear();
}
entityItemIndex++;
}
if (insertCount == entitys.Count)
{
isSucceed = true;
}
}
catch (Exception ex)
{
InsertSql.Clear();
ValuesSql.Clear();
if (sqlParameters != null && sqlParameters.Count > 0)
{
sqlParameters.Clear();
sqlParameters = null;
}
throw ex;
}
finally
{
if (entitys != null && entitys.Count > 0)
{
entitys.Clear();
entitys = null;
}
}
}
}
return isSucceed;
}
标签:
原文地址:http://www.cnblogs.com/stevenchen2016/p/5840565.html