public async Task CheckHasIdentityAsync(DbContext context)
{
int hasIdentity = 0;
if (HasSinglePrimaryKey)
{
var sqlConnection = context.Database.GetDbConnection();
var currentTransaction = context.Database.CurrentTransaction;
try
{
if (currentTransaction == null)
{
if (sqlConnection.State != ConnectionState.Open)
{
await sqlConnection.OpenAsync().ConfigureAwait(false);
}
}
using (var command = sqlConnection.CreateCommand())
{
if (currentTransaction != null)
{
command.Transaction = currentTransaction.GetDbTransaction();
}
command.CommandText = SqlQueryBuilder.SelectIsIdentity(FullTableName, PropertyColumnNamesDict[PrimaryKeys[0]]);
using (var reader = await command.ExecuteReaderAsync().ConfigureAwait(false))
{
if (reader.HasRows)
{
while (await reader.ReadAsync().ConfigureAwait(false))
{
hasIdentity = (int)reader[0];
}
}
}
}
}
finally
{
if (currentTransaction == null)
{
sqlConnection.Close();
}
}
}
HasIdentity = hasIdentity == 1;
}
public void CheckHasIdentity(DbContext context)
{
var hasIdentity = 0;
if (HasSinglePrimaryKey)
{
var sqlConnection = context.Database.GetDbConnection();
var currentTransaction = context.Database.CurrentTransaction;
try
{
if (currentTransaction == null)
{
if (sqlConnection.State != ConnectionState.Open)
{
sqlConnection.Open();
}
}
using (var command = sqlConnection.CreateCommand())
{
if (currentTransaction != null)
{
command.Transaction = currentTransaction.GetDbTransaction();
}
command.CommandText = SqlQueryBuilder.SelectIsIdentity(FullTableName, PropertyColumnNamesDict[PrimaryKeys[0]]);
using (var reader = command.ExecuteReader())
{
if (reader.HasRows)
{
while (reader.Read())
{
hasIdentity = reader[0] == DBNull.Value ? 0 : (int)reader[0];
}
}
}
}
}
finally
{
if (currentTransaction == null)
{
sqlConnection.Close();
}
}
}
HasIdentity = hasIdentity == 1;
}
public async Task LoadOutputDataAsync <T>(DbContext context, IList <T> entities) where T : class
{
if (BulkConfig.SetOutputIdentity && HasSinglePrimaryKey)
{
var sqlQuery = SqlQueryBuilder.SelectFromOutputTable(this);
var entitiesWithOutputIdentity = await QueryOutputTableAsync <T>(context, sqlQuery).ToListAsync().ConfigureAwait(false);
UpdateEntitiesIdentity(entities, entitiesWithOutputIdentity);
}
if (BulkConfig.CalculateStats)
{
var numberUpdated = await GetNumberUpdatedAsync(context);
BulkConfig.StatsInfo = new StatsInfo
{
StatsNumberUpdated = numberUpdated,
StatsNumberInserted = entities.Count - numberUpdated
};
}
}
public static async Task MergeAsync <T>(DbContext context, IList <T> entities, TableInfo tableInfo, OperationType operationType, Action <decimal> progress) where T : class
{
tableInfo.InsertToTempTable = true;
await tableInfo.CheckHasIdentityAsync(context).ConfigureAwait(false);
await context.Database.ExecuteSqlCommandAsync(SqlQueryBuilder.CreateTableCopy(tableInfo.FullTableName, tableInfo.FullTempTableName, tableInfo)).ConfigureAwait(false);
if (tableInfo.CreatedOutputTable)
{
await context.Database.ExecuteSqlCommandAsync(SqlQueryBuilder.CreateTableCopy(tableInfo.FullTableName, tableInfo.FullTempOutputTableName, tableInfo, true)).ConfigureAwait(false);
}
try
{
await InsertAsync(context, entities, tableInfo, progress).ConfigureAwait(false);
await context.Database.ExecuteSqlCommandAsync(SqlQueryBuilder.MergeTable(tableInfo, operationType)).ConfigureAwait(false);
if (tableInfo.CreatedOutputTable)
{
try
{
await tableInfo.LoadOutputDataAsync(context, entities).ConfigureAwait(false);
}
finally
{
if (!tableInfo.BulkConfig.UseTempDB)
{
await context.Database.ExecuteSqlCommandAsync(SqlQueryBuilder.DropTable(tableInfo.FullTempOutputTableName)).ConfigureAwait(false);
}
}
}
}
finally
{
if (!tableInfo.BulkConfig.UseTempDB)
{
await context.Database.ExecuteSqlCommandAsync(SqlQueryBuilder.DropTable(tableInfo.FullTempTableName)).ConfigureAwait(false);
}
}
}
// Compiled queries created manually to avoid EF Memory leak bug when using EF with dynamic SQL:
// https://github.com/borisdj/EFCore.BulkExtensions/issues/73
// Once the following Issue gets fixed(expected in EF 3.0) this can be replaced with code segment: DirectQuery
// https://github.com/aspnet/EntityFrameworkCore/issues/12905
#region CompiledQuery
public void LoadOutputData <T>(DbContext context, IList <T> entities) where T : class
{
if (BulkConfig.SetOutputIdentity && HasSinglePrimaryKey)
{
var sqlQuery = SqlQueryBuilder.SelectFromOutputTable(this);
var entitiesWithOutputIdentity = QueryOutputTable <T>(context, sqlQuery).ToList();
UpdateEntitiesIdentity(entities, entitiesWithOutputIdentity);
}
if (!BulkConfig.CalculateStats)
{
return;
}
SqlQueryBuilder.SelectCountIsUpdateFromOutputTable(this);
var numberUpdated = GetNumberUpdated(context);
BulkConfig.StatsInfo = new StatsInfo
{
StatsNumberUpdated = numberUpdated,
StatsNumberInserted = entities.Count - numberUpdated
};
}
