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

!99 Fix PostgreSQL bulk copy identity column handling

Browse Source 
Merge pull request !99 from fluorescence/master
This commit is contained in:
阿妮亚
2025-10-25 05:18:42 +00:00
committed by Gitee
parent 2a34b41a3d 17cb502564
commit b3f7fed97e
2 changed files with 26 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
Src/Asp.NetCore2/SqlSugar/Abstract/FastestProvider/Private.cs
View File

@@ -99,8 +99,27 @@ namespace SqlSugar
dt.Columns.Add(item.ColumnName, item.DataType); dt.Columns.Add(item.ColumnName, item.DataType);
} }
} }
bool supportIdentity = true;
if (this.context.CurrentConnectionConfig.DbType == DbType.PostgreSQL || this.context.CurrentConnectionConfig.DbType == DbType.Vastbase)
{
supportIdentity = false;
}
if (!supportIdentity)
{
// PostgreSQL/Vastbase不支持自增主键导入
foreach (var identityColumnInfo in this.entityInfo.Columns.Where(it => it.IsIdentity))
{
if (dt.Columns.Contains(identityColumnInfo.DbColumnName))
{
dt.Columns.Remove(identityColumnInfo.DbColumnName);
}
}
}
dt.TableName = GetTableName(); dt.TableName = GetTableName();
var columns = entityInfo.Columns; var columns = supportIdentity ? entityInfo.Columns : entityInfo.Columns.Where(it => !it.IsIdentity).ToList();
if (columns.Where(it=>!it.IsIgnore).Count() > tempDataTable.Columns.Count) if (columns.Where(it=>!it.IsIgnore).Count() > tempDataTable.Columns.Count)
{ {
var tempColumns = tempDataTable.Columns.Cast<DataColumn>().Select(it=>it.ColumnName); var tempColumns = tempDataTable.Columns.Cast<DataColumn>().Select(it=>it.ColumnName);

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

@@ -47,11 +47,11 @@ namespace SqlSugar
var columns = this.Context.DbMaintenance.GetColumnInfosByTableName(this.entityInfo.DbTableName); var columns = this.Context.DbMaintenance.GetColumnInfosByTableName(this.entityInfo.DbTableName);
try try
{ {
var identityColumnInfo = this.entityInfo.Columns.FirstOrDefault(it => it.IsIdentity); //var identityColumnInfo = this.entityInfo.Columns.FirstOrDefault(it => it.IsIdentity);
if (identityColumnInfo != null) //if (identityColumnInfo != null)
{ //{
throw new Exception("PgSql bulkcopy no support identity"); // throw new Exception("PgSql bulkcopy no support identity");
} //}
BulkCopy(dt, copyString, conn, columns); BulkCopy(dt, copyString, conn, columns);
} }
catch (Exception ex) catch (Exception ex)