你能从 DbSet 中获取 DbContext 吗？

Question

在我的应用程序中，有时需要在一次操作中将 10,000 或更多行保存到数据库中。我发现一次简单地迭代和添加每个项目可能需要半个小时以上。

但是，如果我禁用 AutoDetectChangesEnabled 大约需要 5 秒（这正是我想要的）

我正在尝试为 DbSet 创建一个名为“AddRange”的扩展方法，该方法将禁用 AutoDetectChangesEnabled，然后在完成后重新启用它。

public static void AddRange<TEntity>(this DbSet<TEntity> set, DbContext con, IEnumerable<TEntity> items) where TEntity : class
    {
        // Disable auto detect changes for speed
        var detectChanges = con.Configuration.AutoDetectChangesEnabled;
        try
        {
            con.Configuration.AutoDetectChangesEnabled = false;

            foreach (var item in items)
            {
                set.Add(item);
            }
        }
        finally
        {
            con.Configuration.AutoDetectChangesEnabled = detectChanges;
        }
    }

所以，我的问题是：有没有办法从 DbSet 中获取 DbContext？我不喜欢把它作为一个参数——感觉它应该是不必要的。

Answer 1

使用 Entity Framework Core（使用 2.1 版测试），您可以使用获取当前上下文

// DbSet<MyModel> myDbSet
var context = myDbSet.GetService<ICurrentDbContext>().Context;

How to get a DbContext from a DbSet in EntityFramework Core 2.0

Answer 2

是的，您可以从DbSet<TEntity> 获得DbContext，但解决方案需要大量反射。我在下面提供了如何执行此操作的示例。

我测试了以下代码，它能够成功检索到生成 DbSet 的 DbContext 实例。请注意，虽然它确实回答了您的问题，但几乎可以肯定有更好的解决方案来解决您的问题。

public static class HackyDbSetGetContextTrick
{ 
    public static DbContext GetContext<TEntity>(this DbSet<TEntity> dbSet)
        where TEntity: class
    { 
        object internalSet = dbSet
            .GetType()
            .GetField("_internalSet",BindingFlags.NonPublic|BindingFlags.Instance)
            .GetValue(dbSet);
        object internalContext = internalSet
            .GetType()
            .BaseType
            .GetField("_internalContext",BindingFlags.NonPublic|BindingFlags.Instance)
            .GetValue(internalSet); 
        return (DbContext)internalContext
            .GetType()
            .GetProperty("Owner",BindingFlags.Instance|BindingFlags.Public)
            .GetValue(internalContext,null); 
    } 
}

示例用法：

using(var originalContextReference = new MyContext())
{
   DbSet<MyObject> set = originalContextReference.Set<MyObject>();
   DbContext retrievedContextReference = set.GetContext();
   Debug.Assert(ReferenceEquals(retrievedContextReference,originalContextReference));
}

说明：

根据 Reflector，DbSet<TEntity> 有一个 _internalSet 类型为 InternalSet<TEntity> 的私有字段。该类型在 EntityFramework dll 内部。它继承自InternalQuery<TElement>（其中TEntity : TElement）。 InternalQuery<TElement> 也是 EntityFramework dll 的内部。它有一个私有字段_internalContext，类型为InternalContext。 InternalContext 也是 EntityFramework 内部的。但是，InternalContext 公开了一个名为 Owner 的公共 DbContext 属性。因此，如果您有一个DbSet<TEntity>，则可以通过反射性地访问每个属性并将最终结果转换为DbContext 来获得对DbContext 所有者的引用。

从@LoneyPixel更新

在 EF7 中，在实现 DbSet 的类中直接有一个私有字段 _context。公开这个领域并不难

Answer 3

为什么要在 DbSet 上这样做？尝试在 DbContext 上执行此操作：

public static void AddRangeFast<T>(this DbContext context, IEnumerable<T> items) where T : class
{
    var detectChanges = context.Configuration.AutoDetectChangesEnabled;
    try
    {
        context.Configuration.AutoDetectChangesEnabled = false;
        var set = context.Set<T>();

        foreach (var item in items)
        {
            set.Add(item);
        }
    }
    finally
    {
        context.Configuration.AutoDetectChangesEnabled = detectChanges;
    }
}

那么使用就这么简单：

using (var db = new MyContext())
{
    // slow add
    db.MyObjects.Add(new MyObject { MyProperty = "My Value 1" });
    // fast add
    db.AddRangeFast(new[] {
        new MyObject { MyProperty = "My Value 2" },
        new MyObject { MyProperty = "My Value 3" },
    });
    db.SaveChanges();
}

Answer 4

也许您可以创建一个帮手为您禁用此功能，然后从 AddRange 方法中调用该帮手