假设我有一个插入方法:
public T Add<T>(T t)
{
context.Set<T>().Add(t);
context.SaveChanges();
return t;
}
还有一个通用的更新:
public T Update<T>(T updated,int key)
{
if (updated == null)
return null;
T existing = _context.Set<T>().Find(key);
if (existing != null)
{
context.Entry(existing).CurrentValues.SetValues(updated);
context.SaveChanges();
}
return existing;
}
我想将它们合二为一SaveOrUpdate
接受任何实体方法:
我如何最好地实现这一目标,是否有比使用 context.Set<T>().Find(key)
更有效的方法来避免数据库往返?
最佳答案
我有一个方法有点不同:
- 它不会将现有实体设置为
Modified
,而是设置为Attached
。 - 它不执行
SaveChanges()
。
我会解释原因,但首先是来源:
public static class DbContextExtensions
{
public static void AddOrAttach<T>(this DbContext context, T entity)
where T : class
{
#region leave conditions
if (entity == null) return;
var entry = context.Entry(entity);
var leaveStates = new[]
{
EntityState.Deleted,
EntityState.Modified,
EntityState.Unchanged
};
if (leaveStates.Contains(entry.State)) return;
#endregion
var entityKey = context.GetEntityKey(entity);
if (entityKey == null)
{
entry.State = EntityState.Unchanged;
entityKey = context.GetEntityKey(entity);
}
if (entityKey.EntityKeyValues == null
|| entityKey.EntityKeyValues.Select(ekv => (int)ekv.Value).All(v => v <= 0))
{
entry.State = EntityState.Added;
}
}
public static EntityKey GetEntityKey<T>(this DbContext context, T entity)
where T : class
{
var oc = ((IObjectContextAdapter)context).ObjectContext;
ObjectStateEntry ose;
if (null != entity && oc.ObjectStateManager
.TryGetObjectStateEntry(entity, out ose))
{
return ose.EntityKey;
}
return null;
}
}
如您所见,在 AddOrAttach
方法中有许多我保持不变的状态。
然后有一些逻辑来确定是否应该添加或附加实体。实质是上下文跟踪的每个实体都有一个 EntityKey
对象。如果还没有,我会先附加它,以便它获得一个。
然后,在某些情况下,实体确实具有 EntityKey
,但没有键值。如果是这样,它将被添加
。此外,当它有键值,但它们都是 0 或更小时,它将被 Added
。 (请注意,我假设您使用 int
键字段,可能作为复合主键)。
为什么没有 SaveChanges?
您的方法一个一个地存储实体。但是,通过一次 SaveChanges
调用保存多个对象(对象图)更为常见,即在一次事务中。如果你想通过你的方法来做到这一点,你必须将所有调用包装在 TransactionScope
中(或者以其他方式启动并提交事务)。在一个逻辑工作单元中构建或修改您使用的实体,然后执行一次 SaveChanges
调用要方便得多。这就是为什么我只通过这种方法设置实体状态。
为什么附加?
人们制作了类似的方法来执行“upsert”(添加或更新)。缺点是它将整个实体标记为已修改,而不仅仅是其修改的属性。我更喜欢附加一个实体,然后根据它发生的任何事情继续编写代码,这可能修改它的一个或某些属性。
显然,您很清楚将属性设置为已修改的好处,因为您使用
context.Entry(existing).CurrentValues.SetValues(updated);
这确实是将值复制到现有实体中的推荐方法。每当我使用它时,我都会在我的 AddOrAttach
方法之外(并遵循)它。
但是……
is there a more efficient way that avoids a round trip to the database
CurrentValues.SetValues
仅在当前值为数据库值时才有效。所以你离不开原始实体来使用这个方法。因此,在断开连接的情况下(例如,Web 应用程序),如果要使用此方法,则无法避免数据库往返。另一种方法是将实体状态设置为 Modified
(具有上述缺点)。看我的回答here对此进行更多讨论。
关于c# - Entity Framework 的通用插入或更新,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/39131108/