Question

我正在使用SQLBulkCopy来移动大量数据。我实现了通知事件，以便每次处理完一定数量的行时通知我，但是当作业完成时，OnSqlRowsCopied事件不会触发。如何在SQLBulkCopy writetoserver完成时获取复制的总行数？

Answer 1

以下hack（使用反射）是一个选项：

    /// <summary>
    /// Helper class to process the SqlBulkCopy class
    /// </summary>
    static class SqlBulkCopyHelper
    {
        static FieldInfo rowsCopiedField = null;

        /// <summary>
        /// Gets the rows copied from the specified SqlBulkCopy object
        /// </summary>
        /// <param name="bulkCopy">The bulk copy.</param>
        /// <returns></returns>
        public static int GetRowsCopied(SqlBulkCopy bulkCopy)
        {
            if (rowsCopiedField == null)
            {
                rowsCopiedField = typeof(SqlBulkCopy).GetField("_rowsCopied", BindingFlags.NonPublic | BindingFlags.GetField | BindingFlags.Instance);
            }

            return (int)rowsCopiedField.GetValue(bulkCopy);
        }
    }

然后按如下方式使用该类：

int rowsCopied = SqlBulkCopyHelper.GetRowsCopied(bulkCopyObjectInYourCode);

希望这有帮助。

Answer 2

为了完整性，我实现了一个扩展方法并包含了命名空间。如果您想要快速解决方案来获取复制的计数，请复制并粘贴此类。注意：当Ignore Duplicates设置为ON时，此计数不考虑实际插入的行数。

namespace System.Data.SqlClient
{    
    using Reflection;

    public static class SqlBulkCopyExtension
    {
        const String _rowsCopiedFieldName = "_rowsCopied";
        static FieldInfo _rowsCopiedField = null;

        public static int RowsCopiedCount(this SqlBulkCopy bulkCopy)
        {
            if (_rowsCopiedField == null) _rowsCopiedField = typeof(SqlBulkCopy).GetField(_rowsCopiedFieldName, BindingFlags.NonPublic | BindingFlags.GetField | BindingFlags.Instance);            
            return (int)_rowsCopiedField.GetValue(bulkCopy);
        }
    }
}

Answer 3

通过使用SqlBulkCopy.SqlRowsCopied Event（每次处理NotifyAfter属性指定的行数时发生），我们可以在完成时实现SQLBulkCopy行计数。

using (SqlBulkCopy s = new SqlBulkCopy(db.Database.Connection as SqlConnection))
{
  s.SqlRowsCopied += new SqlRowsCopiedEventHandler(sqlBulk_SqlRowsCopied);
  s.BatchSize = csvFileData.Rows.Count;//DataTable
  s.NotifyAfter = csvFileData.Rows.Count;
  foreach (var column in csvFileData.Columns)
     s.ColumnMappings.Add(column.ToString(), column.ToString());
  // Set the timeout.
  s.BulkCopyTimeout = 60;
  s.DestinationTableName = "Employee_Data";
  s.WriteToServer(csvFileData);
}

private static void sqlBulk_SqlRowsCopied(object sender, SqlRowsCopiedEventArgs e)
{
    long Count = e.RowsCopied;
}

Answer 4

我认为你必须在完成后对表运行COUNT（）查询，如MSDN示例here中所示。

除此之外，你不能告诉前方吗？例如如果您将DataTable传递给WriteToServer（），那么您可以通过对其执行.Rows.Count来了解多少条记录。

Answer 5

这就是我所做的 - 这是Rahul Modi在这个线程中的解决方案的略微修改（基本上它只是将SqlRowsCopied事件放入内联，我认为在这个实例中比创建新的事件处理程序方法更简洁）： / p>

private long InsetData(DataTable dataTable, SqlConnection connection)
{
   using (SqlBulkCopy copier = new SqlBulkCopy(connection))
   {
      var filesInserted = 0L;

      connection.Open();

      copier.DestinationTableName = "dbo.MyTable";
      copier.NotifyAfter = dataTable.Rows.Count;
      copier.SqlRowsCopied += (s, e) => filesInserted = e.RowsCopied;
      copier.WriteToServer(dataTable);

      connection.Close();

      return filesInserted;
   }
}

Answer 6

将NotifyAfter设置为1。在SqlRowsCopied的处理程序中，增加一个计数器。 WriteToServer完成后，阅读计数器。

Answer 7

扩展方法：

（基于@Benzi的答案）

using System;
using System.Reflection;
using System.Data.SqlClient;
using static System.Reflection.BindingFlags;

namespace Extensions
{
    public static class SqlBulkCopyExtensions
    {
        private static readonly Lazy<FieldInfo> _rowsCopied = new Lazy<FieldInfo>(() => typeof(SqlBulkCopy).GetField("_rowsCopied", NonPublic | GetField | Instance));

        public static int GetRowsCopied(this SqlBulkCopy sqlBulkCopy)
        {
            return (int)_rowsCopied.Value.GetValue(sqlBulkCopy);
        }
    }
}

经过测试并可以在.NET 4.6.1或更高版本上运行

请注意，字段的类型为int（而事件args中的属性的类型为long）。不知道如果复制> int.MaxValue行会发生什么情况。

完成时SQLBulkCopy行计数

7 个答案: