public static void Execute <T>(DbContext context, IList <T> entities, OperationType operationType, BulkConfig bulkConfig, Action <decimal> progress) where T : class
{
if (operationType != OperationType.Truncate && entities.Count == 0)
{
return;
}
TableInfo tableInfo = TableInfo.CreateInstance(context, entities, operationType, bulkConfig);
if (operationType == OperationType.Insert && !tableInfo.BulkConfig.SetOutputIdentity)
{
SqlBulkOperation.Insert(context, entities, tableInfo, progress);
}
else if (operationType == OperationType.Read)
{
SqlBulkOperation.Read(context, entities, tableInfo, progress);
}
else if (operationType == OperationType.Truncate)
{
SqlBulkOperation.Truncate(context, tableInfo);
}
else
{
SqlBulkOperation.Merge(context, entities, tableInfo, operationType, progress);
}
}
public static Task ExecuteAsync <T>(DbContext context, IList <T> entities, OperationType operationType, BulkConfig bulkConfig, Action <decimal> progress, CancellationToken cancellationToken) where T : class
{
if (operationType != OperationType.Truncate && entities.Count == 0)
{
return(Task.CompletedTask);
}
TableInfo tableInfo = TableInfo.CreateInstance(context, entities, operationType, bulkConfig);
if (operationType == OperationType.Insert && !tableInfo.BulkConfig.SetOutputIdentity)
{
return(SqlBulkOperation.InsertAsync(context, entities, tableInfo, progress, cancellationToken));
}
else if (operationType == OperationType.Read)
{
return(SqlBulkOperation.ReadAsync(context, entities, tableInfo, progress, cancellationToken));
}
else if (operationType == OperationType.Truncate)
{
return(SqlBulkOperation.TruncateAsync(context, tableInfo));
}
else
{
return(SqlBulkOperation.MergeAsync(context, entities, tableInfo, operationType, progress, cancellationToken));
}
}
public void SetSqlBulkCopyConfig <T>(SqlBulkCopy sqlBulkCopy, IList <T> entities, bool setColumnMapping, Action <decimal> progress)
{
sqlBulkCopy.DestinationTableName = InsertToTempTable ? FullTempTableName : FullTableName;
sqlBulkCopy.BatchSize = BulkConfig.BatchSize;
sqlBulkCopy.NotifyAfter = BulkConfig.NotifyAfter ?? BulkConfig.BatchSize;
sqlBulkCopy.SqlRowsCopied += (sender, e) => {
progress?.Invoke(SqlBulkOperation.GetProgress(entities.Count, e.RowsCopied)); // round to 4 decimal places
};
sqlBulkCopy.BulkCopyTimeout = BulkConfig.BulkCopyTimeout ?? sqlBulkCopy.BulkCopyTimeout;
sqlBulkCopy.EnableStreaming = BulkConfig.EnableStreaming;
if (setColumnMapping)
{
foreach (var element in PropertyColumnNamesDict)
{
sqlBulkCopy.ColumnMappings.Add(element.Key, element.Value);
}
}
}