!99 Fix PostgreSQL bulk copy identity column handling

Merge pull request !99 from fluorescence/master
This commit is contained in:
阿妮亚
2025-10-25 05:18:42 +00:00
committed by Gitee
2 changed files with 26 additions and 7 deletions

View File

@@ -99,8 +99,27 @@ namespace SqlSugar
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();
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)
{
var tempColumns = tempDataTable.Columns.Cast<DataColumn>().Select(it=>it.ColumnName);

View File

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