EF中的那些批量操作

在使用EF的过程中，我们经常会遇到需要批量操作数据的场景，批量操作有的时候不仅能提高性能，比如使用SqlBulkCopy进入批量插入的时候，而且比较方便操作，提高效率。那么这篇文章就来总结EF中的那些批量操作。

插入

1，单个插入

单个插入很简单，只需要构造好实体，然后调用DbSet的Add方法，最后SaveChanges就行，示例代码如下。

using (var db = new PackageFHContext())
{
    // 单个插入
    var entity = new ProdImage
    {
    ProductID = ,
    Type = ,
    Priority = ,
    Url = " http://www.baidu.com/abc.png",
    Alt = "test",
    Creator = "system",
    CreateOn = DateTime.Now
    };
    db.ProdImageSet.Add(entity);
    db.SaveChanges();
}

2，批量插入

很多时候我们需要向同一数据表插入大量的数据，如果插入一条就调一次SaveChanges这样性能很低，需要频繁地访问数据库。为了解决这个问题，可以使用SqlBulkCopy来批量插入数据，这样很大地提高性能。示例代码如下。

using (var db = new PackageFHContext())
{
    // 批量插入(使用到SqlBulkCopy)
    var prodImages = new List<ProdImage>();
    for (int i = ; i < ; i++)
    {
        prodImages.Add(new ProdImage
        {
        ProductID = ,
        Type = ,
        Priority = ,
        Url = " http://www.baidu.com/abc.png",
        Alt = "test" + i,
        Creator = "system",
        CreateOn = DateTime.Now
        });
    } 

    if (db.Database.Connection.State != ConnectionState.Open)
    {
        db.Database.Connection.Open();
    } 

    PackageFH.Common.DBHelper.BulkInsert((SqlConnection)db.Database.Connection, "dbo.Pkg_ProdImage", prodImages); 

    if (db.Database.Connection.State != ConnectionState.Closed)
    {
        db.Database.Connection.Close();
    }
}

当然SqlBulkCopy的操作封装在了DBHelper中了，后面会附上DBHelper.BulkInsert的实现。

更新

1，单个更新

单个更新很简单，只需要先查询出那条数据，然后修改字段的值，最后SaveChanges就行。示例代码如下。

using (var db = new PackageFHContext())
{
    // 单个更新
    var entity = db.ProdImageSet.FirstOrDefault(p => p.ImageID == );
    entity.Alt = "ooooooooooo";
    db.SaveChanges();
}

2，批量更新

我们可以先查找要更新的那些记录，然后遍历修改字段的值，最后调用SaveChanges提交更新。示例代码如下。

using (var db = new PackageFHContext())
{
    // 批量更新
    var entitys = db.ProdImageSet.Where(p => p.ProductID ==  && p.Type == );
    var j = ;
    entitys.ToList().ForEach(item =>
    {
        item.Alt = "update test" + j;
        db.Entry(item).State = System.Data.Entity.EntityState.Modified; //不加这句也可以，为什么？
        j++;
    });
    db.SaveChanges();
}

通过SQL Profiler监控可以看到，最后生成的还是很多的UPDATE语句，也就是说每更新一条就是一个UPDATE语句。

删除

1，单个删除

示例代码如下。

using (var db = new PackageFHContext())
{
    // 单个删除
    var entity = db.ProdImageSet.FirstOrDefault(p => p.ImageID == );
    db.ProdImageSet.Remove(entity);
    db.SaveChanges();
}

2，批量删除

示例代码如下。

using (var db = new PackageFHContext())
{
    // 批量删除
    var entitys = db.ProdImageSet.Where(p => p.ProductID ==  && p.Type == );
    entitys.ToList().ForEach(entity => db.Entry(entity).State = System.Data.Entity.EntityState.Deleted); //不加这句也可以，为什么？
    db.ProdImageSet.RemoveRange(entitys);
    db.SaveChanges();
}

通过SQL Profiler监控，结果生成的也是多条DELETE语句。

附：DBHelper.BulkInsert的实现代码

 public static void BulkInsert<T>(SqlConnection conn, string tableName, IList<T> list)
 {
     using (var bulkCopy = new SqlBulkCopy(conn))
     {
         bulkCopy.BatchSize = list.Count;
         bulkCopy.DestinationTableName = tableName;

         var table = new DataTable();
         var props = TypeDescriptor.GetProperties(typeof(T), new Attribute[] { new DatabaseTableColumnAttribute() })
             //Dirty hack to make sure we only have system data types
             //i.e. filter out the relationships/collections
             .Cast<PropertyDescriptor>()
             .Where(propertyInfo => propertyInfo.PropertyType.Namespace.Equals("System"))
             .ToArray();

         foreach (var propertyInfo in props)
         {
             bulkCopy.ColumnMappings.Add(propertyInfo.Name, propertyInfo.Name);
             table.Columns.Add(propertyInfo.Name, Nullable.GetUnderlyingType(propertyInfo.PropertyType) ?? propertyInfo.PropertyType);
         }

         var values = new object[props.Length];
         foreach (var item in list)
         {
             for (var i = ; i < values.Length; i++)
             {
                 values[i] = props[i].GetValue(item);
             }

             table.Rows.Add(values);
         }

         bulkCopy.WriteToServer(table);
     }
 }