EF中的那些批量操作

在使用EF的過程中，我們經常會遇到需要批量操作數據的場景，批量操作有的時候不僅能提高性能，比如使用SqlBulkCopy進入批量插入的時候，而且比較方便操作，提高效率。那么這篇文章就來總結EF中的那些批量操作。

插入

1，單個插入

單個插入很簡單，只需要構造好實體，然后調用DbSet的Add方法，最后SaveChanges就行，示例代碼如下。

using (var db = new PackageFHContext()) 
{ 
    // 單個插入 
    var entity = new ProdImage 
    { 
    ProductID = 1807, 
    Type = 7, 
    Priority = 0, 
    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 = 0; i < 50; i++) 
    { 
        prodImages.Add(new ProdImage 
        { 
        ProductID = 1807, 
        Type = 7, 
        Priority = 0, 
        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 == 275695);
    entity.Alt = "ooooooooooo";
    db.SaveChanges();
}

2，批量更新

我們可以先查找要更新的那些記錄，然后遍歷修改字段的值，最后調用SaveChanges提交更新。示例代碼如下。

using (var db = new PackageFHContext())
{
    // 批量更新
    var entitys = db.ProdImageSet.Where(p => p.ProductID == 1807 && p.Type == 7);
    var j = 1;
    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 == 275695);
    db.ProdImageSet.Remove(entity);
    db.SaveChanges();
}

2，批量刪除

示例代碼如下。

using (var db = new PackageFHContext())
{
    // 批量刪除
    var entitys = db.ProdImageSet.Where(p => p.ProductID == 1783 && p.Type == 0);
    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 = 0; i < values.Length; i++)
            {
                values[i] = props[i].GetValue(item);
            }

            table.Rows.Add(values);
        }

        bulkCopy.WriteToServer(table);
    }
}