lsm/SqlSugar
1
0
Fork 0
mirror of https://gitee.com/dotnetchina/SqlSugar.git synced 2025-11-08 10:24:55 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

修改海量数据库mysql BulkCopyAsync兼容性问题

Browse Source
This commit is contained in:
陈剑辉
2025-10-28 13:49:35 +08:00
parent 4b4aa7b3be
commit fa7b38119d
1 changed files with 15 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
Src/Asp.NetCore2/SqlSugar/Realization/PostgreSQL/SqlBuilder/PostgreSQLFastBuilder.cs
View File

@@ -11,7 +11,7 @@ namespace SqlSugar
{ {
public class PostgreSQLFastBuilder : FastBuilder, IFastBuilder public class PostgreSQLFastBuilder : FastBuilder, IFastBuilder
{ {
public static Dictionary<string , NpgsqlDbType> PgSqlType = UtilMethods.EnumToDictionary<NpgsqlDbType>(); public static Dictionary<string, NpgsqlDbType> PgSqlType = UtilMethods.EnumToDictionary<NpgsqlDbType>();
private EntityInfo entityInfo; private EntityInfo entityInfo;
public PostgreSQLFastBuilder(EntityInfo entityInfo) public PostgreSQLFastBuilder(EntityInfo entityInfo)
@@ -38,8 +38,10 @@ namespace SqlSugar
{ {
lsColNames.Add($"\"{dt.Columns[i].ColumnName}\""); lsColNames.Add($"\"{dt.Columns[i].ColumnName}\"");
} }
string copyString = $"COPY {dt.TableName} ( {string.Join(",", lsColNames) } ) FROM STDIN (FORMAT BINARY)"; string copyString = $"COPY {dt.TableName} ( {string.Join(",", lsColNames)} ) FROM STDIN (FORMAT BINARY)";
if (this.Context?.CurrentConnectionConfig?.MoreSettings?.DatabaseModel == DbType.OpenGauss) var innertemp = this.Context.CurrentConnectionConfig?.MoreSettings?.InnerTemp as DbType?;
if (this.Context?.CurrentConnectionConfig?.MoreSettings?.DatabaseModel == DbType.OpenGauss ||
(this.Context?.CurrentConnectionConfig?.MoreSettings?.DatabaseModel == DbType.Vastbase && innertemp == DbType.MySql))
{ {
copyString = copyString.Replace("(FORMAT BINARY)", "(FORMAT 'BINARY')"); copyString = copyString.Replace("(FORMAT BINARY)", "(FORMAT 'BINARY')");
} }
@@ -65,7 +67,7 @@ namespace SqlSugar
return await Task.FromResult(dt.Rows.Count); return await Task.FromResult(dt.Rows.Count);
} }
private void BulkCopy(DataTable dt, string copyString, NpgsqlConnection conn, List<DbColumnInfo> columns) private void BulkCopy(DataTable dt, string copyString, NpgsqlConnection conn, List<DbColumnInfo> columns)
{ {
if (conn.State == ConnectionState.Closed) if (conn.State == ConnectionState.Closed)
conn.Open(); conn.Open();
@@ -75,7 +77,7 @@ namespace SqlSugar
ColumnView result = new ColumnView(); ColumnView result = new ColumnView();
result.DbColumnInfo = columns.FirstOrDefault(it => it.DbColumnName.EqualCase(item.ColumnName)); result.DbColumnInfo = columns.FirstOrDefault(it => it.DbColumnName.EqualCase(item.ColumnName));
result.DataColumn = item; result.DataColumn = item;
result.EntityColumnInfo=this.entityInfo.Columns.FirstOrDefault(it => it.DbColumnName.EqualCase(item.ColumnName)); result.EntityColumnInfo = this.entityInfo.Columns.FirstOrDefault(it => it.DbColumnName.EqualCase(item.ColumnName));
var key = result.DbColumnInfo?.DataType?.ToLower(); var key = result.DbColumnInfo?.DataType?.ToLower();
if (result.DbColumnInfo == null) if (result.DbColumnInfo == null)
{ {
@@ -148,7 +150,7 @@ namespace SqlSugar
public override async Task<int> UpdateByTempAsync(string tableName, string tempName, string[] updateColumns, string[] whereColumns) public override async Task<int> UpdateByTempAsync(string tableName, string tempName, string[] updateColumns, string[] whereColumns)
{ {
var sqlquerybulder= this.Context.Queryable<object>().SqlBuilder; var sqlquerybulder = this.Context.Queryable<object>().SqlBuilder;
Check.ArgumentNullException(!updateColumns.Any(), "update columns count is 0"); Check.ArgumentNullException(!updateColumns.Any(), "update columns count is 0");
Check.ArgumentNullException(!whereColumns.Any(), "where columns count is 0"); Check.ArgumentNullException(!whereColumns.Any(), "where columns count is 0");
var sets = string.Join(",", updateColumns.Select(it => $"{sqlquerybulder.GetTranslationColumnName(it)}=TE.{sqlquerybulder.GetTranslationColumnName(it)}")); var sets = string.Join(",", updateColumns.Select(it => $"{sqlquerybulder.GetTranslationColumnName(it)}=TE.{sqlquerybulder.GetTranslationColumnName(it)}"));