如何在 ADO.NET EF 4.1 中插入记录？

Question

我正在尝试完成一些非常简单的事情，但我找不到使用 Entity Framework 4.1 的方法。

我想要一个控制器方法，它接受一个对象，然后执行 UPSERT（插入或更新，具体取决于数据库中是否存在记录）。

我使用的是自然密钥，所以我无法查看我的 POCO 并判断它是否是新的。

这就是我的做法，但对我来说似乎是错误的：

[HttpPost]
public JsonResult SaveMyEntity(MyEntity entity)
{            
    MyContainer db = new MyContainer(); // DbContext
    if (ModelState.IsValid)
    {
        var existing =
            db.MyEntitys.Find(entity.MyKey);
        if (existing == null)
        {
            db.MyEntitys.Add(entity);
        }
        else
        {
            existing.A = entity.A;
            existing.B = entity.B;
            db.Entry(existing).State = EntityState.Modified;
        }
        db.SaveChanges();
        return Json(new { Result = "Success" });
    }
}

理想情况下，整个事情应该是这样的：

db.MyEntities.AddOrModify(entity);

Answer 1

不幸的是，如果不查询数据库或使用存储过程，就无法做到这一点。简约的代码应该是：

public void AddOrModify<T>(T entity, string key) where T : class, IEntity // Implements MyKey 
{
     using (var context = new MyContainer())
     {
         if (context.Set<T>().Any(e => e.MyKey == key))
         {
              context.Entry(entity).State = EntityState.Modified;
         } 
         else
         {
              context.Entry(entity).State = EntityState.Added;
         }

         context.SaveChanges();
     }
}

Answer 2

在大多数情况下，您不需要显式设置 EntityState.Modified，除非您已禁用更改跟踪。

我们采取的解决方案是检查实体标识符的值：

if (entity.Id == default(int)) {
    // transient entity so insert
} else {
    // update
}

Answer 3

实际上有一种方法可以通知数据库上下文，您尝试插入的实体已更改，现在是新的

_context.MyEntity.Attach(entity);
_context.MyEntity(entity).State = System.Data.EntityState.Modified;

Answer 4

要执行 UPSERT 操作，您可能需要考虑创建一个执行 MERGE 的 SP。

http://www.databasejournal.com/features/mssql/article.php/3739131/UPSERT-Functionality-in-SQL-Server-2008.htm

无论您选择何种方式，操作都必须是原子的，否则您将遇到竞争条件。 你的 SP 可能需要一个 HOLDLOCK 来取消这个 ...

http://weblogs.sqlteam.com/dang/archive/2009/01/31/UPSERT-Race-Condition-With-MERGE.aspx

Answer 5

我知道这个问题很老，并且有一个可以接受的答案，但我认为有更好的解决方案：它不需要实现额外的接口或定义键类型。

public static class DbSetExtensions
{
    public static EntityEntry<TEnt> AddIfNotExists<TEnt, TKey>(this DbSet<TEnt> dbSet, TEnt entity, Func<TEnt, TKey> predicate) where TEnt : class
    {
        var exists = dbSet.Any(c => predicate(entity).Equals(predicate(c)));
        return exists
            ? null
            : dbSet.Add(entity);
    }

    public static void AddRangeIfNotExists<TEnt, TKey>(this DbSet<TEnt> dbSet, IEnumerable<TEnt> entities, Func<TEnt, TKey> predicate) where TEnt : class
    {
        var entitiesExist = from ent in dbSet
            where entities.Any(add => predicate(ent).Equals(predicate(add)))
            select ent;

        dbSet.AddRange(entities.Except(entitiesExist));
    }
}

所以以后可以这样使用：

using (var context = new MyDbContext())
{
    var user1 = new User { Name = "Peter", Age = 32 };
    context.Users.AddIfNotExists(user1, u => u.Name);

    var user2 = new User { Name = "Joe", Age = 25 };
    context.Users.AddIfNotExists(user2, u => u.Age);

    // Adds user1 if there is no user with name "Peter"
    // Adds user2 if there is no user with age 25
    context.SaveChanges();
}