Browse Source

Optimization of code

pull/2/head
skx 4 years ago
parent
commit
f7a11a9855
  1. 84
      Src/Asp.Net/SqlSugar/Realization/SqlServer/SqlBuilder/SqlServerBlueCopy.cs

84
Src/Asp.Net/SqlSugar/Realization/SqlServer/SqlBuilder/SqlServerBlueCopy.cs

@ -14,7 +14,7 @@ namespace SqlSugar
internal SqlSugarProvider Context { get; set; } internal SqlSugarProvider Context { get; set; }
internal ISqlBuilder Builder { get; set; } internal ISqlBuilder Builder { get; set; }
internal InsertBuilder InsertBuilder { get; set; } internal InsertBuilder InsertBuilder { get; set; }
public object[] Inserts { get; internal set; } internal object[] Inserts { get; set; }
public int ExecuteBlueCopy() public int ExecuteBlueCopy()
{ {
@ -24,10 +24,9 @@ namespace SqlSugar
{ {
return WriteToServer(); return WriteToServer();
} }
DataTable dt = GetCopyData();
DataTable dt; SqlBulkCopy bulkCopy = GetBulkCopyInstance();
SqlBulkCopy bulkCopy; bulkCopy.DestinationTableName = InsertBuilder.GetTableNameString;
SetCopyData(out dt, out bulkCopy);
try try
{ {
bulkCopy.WriteToServer(dt); bulkCopy.WriteToServer(dt);
@ -51,10 +50,9 @@ namespace SqlSugar
{ {
return WriteToServer(); return WriteToServer();
} }
DataTable dt=GetCopyData();
DataTable dt; SqlBulkCopy bulkCopy = GetBulkCopyInstance();
SqlBulkCopy bulkCopy; bulkCopy.DestinationTableName = InsertBuilder.GetTableNameString;
SetCopyData(out dt, out bulkCopy);
try try
{ {
await bulkCopy.WriteToServerAsync(dt); await bulkCopy.WriteToServerAsync(dt);
@ -72,6 +70,41 @@ namespace SqlSugar
} }
private int WriteToServer() private int WriteToServer()
{
var dt = this.Inserts.First() as DataTable;
if (dt == null)
return 0;
Check.Exception(dt.TableName == "Table", "dt.TableName can't be null ");
dt = GetCopyWriteDataTable(dt);
SqlBulkCopy copy = GetBulkCopyInstance();
copy.DestinationTableName = this.Builder.GetTranslationColumnName(dt.TableName);
copy.WriteToServer(dt);
return dt.Rows.Count;
}
private DataTable GetCopyWriteDataTable(DataTable dt)
{
var result = this.Context.Ado.GetDataTable("select top 0 * from " + this.Builder.GetTranslationColumnName(dt.TableName));
foreach (DataRow item in dt.Rows)
{
DataRow dr= result.NewRow();
foreach (DataColumn column in result.Columns)
{
if (dt.Columns.Cast<DataColumn>().Select(it => it.ColumnName.ToLower()).Contains(column.ColumnName.ToLower()))
{
dr[column.ColumnName] = item[column.ColumnName];
if (dr[column.ColumnName] == null)
{
dr[column.ColumnName] = DBNull.Value;
}
}
}
result.Rows.Add(dr);
}
result.TableName = dt.TableName;
return result;
}
private SqlBulkCopy GetBulkCopyInstance()
{ {
SqlBulkCopy copy; SqlBulkCopy copy;
if (this.Context.Ado.Transaction == null) if (this.Context.Ado.Transaction == null)
@ -82,23 +115,15 @@ namespace SqlSugar
{ {
copy = new SqlBulkCopy((SqlConnection)this.Context.Ado.Connection, SqlBulkCopyOptions.CheckConstraints, (SqlTransaction)this.Context.Ado.Transaction); copy = new SqlBulkCopy((SqlConnection)this.Context.Ado.Connection, SqlBulkCopyOptions.CheckConstraints, (SqlTransaction)this.Context.Ado.Transaction);
} }
var dt = this.Inserts.First() as DataTable;
if (dt == null)
{
return 0;
}
if (this.Context.Ado.Connection.State == ConnectionState.Closed) if (this.Context.Ado.Connection.State == ConnectionState.Closed)
{ {
this.Context.Ado.Connection.Open(); this.Context.Ado.Connection.Open();
} }
Check.Exception(dt.TableName=="Table","dt.TableName can't be null "); return copy;
copy.DestinationTableName =this.Builder.GetTranslationColumnName(dt.TableName);
copy.WriteToServer(dt);
return dt.Rows.Count;
} }
private void SetCopyData(out DataTable dt, out SqlBulkCopy bulkCopy) private DataTable GetCopyData()
{ {
dt = this.Context.Ado.GetDataTable("select top 0 * from " + InsertBuilder.GetTableNameString); var dt = this.Context.Ado.GetDataTable("select top 0 * from " + InsertBuilder.GetTableNameString);
foreach (var rowInfos in DbColumnInfoList) foreach (var rowInfos in DbColumnInfoList)
{ {
var dr = dt.NewRow(); var dr = dt.NewRow();
@ -127,25 +152,8 @@ namespace SqlSugar
} }
dt.Rows.Add(dr); dt.Rows.Add(dr);
} }
bulkCopy = null; return dt;
if (this.Context.Ado.Transaction != null)
{
var sqlTran = this.Context.Ado.Transaction as SqlTransaction;
bulkCopy = new SqlBulkCopy(this.Context.Ado.Connection as SqlConnection, SqlBulkCopyOptions.CheckConstraints, sqlTran);
}
else
{
bulkCopy = new SqlBulkCopy(this.Context.Ado.Connection as SqlConnection);
}
//获取目标表的名称
bulkCopy.DestinationTableName = InsertBuilder.GetTableNameString;
//写入DataReader对象
if (this.Context.Ado.Connection.State == ConnectionState.Closed)
{
this.Context.Ado.Connection.Open();
}
} }
private object AddParameter(int i,string dbColumnName, object value) private object AddParameter(int i,string dbColumnName, object value)
{ {
var name =Builder.SqlParameterKeyWord+dbColumnName+i; var name =Builder.SqlParameterKeyWord+dbColumnName+i;

Loading…
Cancel
Save