实体框架的通用插入或更新

Question

假设我有一个插入方法：

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)

Answer 1

我有一个稍微不同的方法：

• 它不会将现有实体设置为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，但没有键值。如果是这样，那就会了Added。此外，当它有键值，但它们都是 0 或更小时，它将是Added. （请注意，我假设您使用int键字段，可能作为复合主键）。

为什么没有保存更改？

您的方法将实体一一存储。SaveChanges然而，更常见的是通过一次调用（即在一个事务中）保存多个对象（对象图） 。如果您想通过您的方法执行此操作，则必须将所有调用包装在 a 中TransactionScope（或者启动并提交事务，否则）。在一个逻辑工作单元中构建或修改您使用的实体然后进行一次SaveChanges调用要方便得多。这就是为什么我只通过这种方法设置实体状态。

为什么要附加？

人们制作了类似的方法来执行“upsert”（添加或更新）。缺点是它将整个实体标记为已修改，而不仅仅是其已修改的属性。我更喜欢附加一个实体，然后继续执行代码，无论它发生什么，这可能会修改它的一个或一些属性。

显然，您很清楚将属性设置为修改后的好处，因为您使用

context.Entry(existing).CurrentValues.SetValues(updated);

这确实是将值复制到现有实体的推荐方法。每当我使用它时，我都会在我的方法之外（并遵循）我的方法进行AddOrAttach。但...

有没有更有效的方法来避免往返数据库

CurrentValues.SetValues仅当当前值是数据库值时才有效。所以使用这个方法就离不开原始实体。因此，在断开连接的场景（例如，Web 应用程序）中，如果要使用此方法，则无法避免数据库往返。另一种方法是将实体状态设置为Modified（具有上述缺点）。有关于此的更多讨论，请参阅我的回答。

Answer 2

您可以使用接口并执行类似的操作。我使用了显式实现，因此其余代码不必处理它。

// I am not 100% sold on my chosen name for the interface, if you like this idea change it to something more suitable
public interface IIsPersisted {
    bool IsPersistedEntity{get;}
    int Key {get;}
}

public class SomeEntityModel : IIsPersisted{
    public int SomeEntityModelId {get;set;}

    /*some other properties*/

    bool IIsPersisted.IsPersistedEntity{get { return this.SomeEntityModelId > 0;}}
    int IIsPersisted.Key {get{return this.SomeEntityModelId;}}
}



public T UpdateOrCreate<T>(T updated) where T : class, IIsPersisted
{
    if (updated == null)
        return null;

    if(updated.IsPersistedEntity)
    {
        T existing = _context.Set<T>().Find(updated.Key);
        if (existing != null)
        {
            context.Entry(existing).CurrentValues.SetValues(updated);
            context.SaveChanges();
        }
        return existing;
    }
    else
    {
        context.Set<T>().Add(updated);
        context.SaveChanges();
        return updated;
    }
}

编辑

我现在才看到这个：

有没有比使用更有效的方法来避免往返数据库context.Set<T>().Find(key)

如果您希望整个实体从分离状态更新，那么最简单的事情就是这样做。

context.Entry(updated).State = EntityState.Modified;
context.SaveChanges();

这会将整个实体标记为脏并将所有内容保存回数据库。