从 DbContext 获取 IQueryProvider

Question

我正在尝试编写一个查询提供程序，它在执行前实例化 DbContext，执行查询，然后立即释放生成的上下文对象。

为此，我必须实现IQueryProvider.Execute(....)。在这个方法中，我需要从我的DbContext 对象中获取IQueryProvider 以让它执行查询。

问题是我不知道如何从上下文对象访问提供者。据我所知，它存储在((IObjectContextAdapter)context).ObjectContext.QueryProvider，但该属性是内部的。

我也尝试从 DbSet 中获取它，但我不知道要指定什么类型。 context.Set(type?).AsQueryable().Provider.Execute<TResult>(expression) 或 context.Set<type?>().AsQueryable().Provider.Execute<TResult>(expression)

所以我的问题是如何从给定的DbContext 对象中获取底层的IQueryProvider。

我要解决的根本问题

我想在不同的上下文中针对同一个数据库执行多个查询。但是 DbContext 是 IDisposable 你必须将它包装在 using.Blocks 中。

IEnumerable<T1> result1;
IEnumerable<T2> result2;

using(var context1 = new MyContext())
using(var context2 = new MyContext())
{
    var result1Task = context1.Set<T1>().ToListAsync();
    var result2Task = context2.Set<T2>().ToListAsync();

    await Task.WhenAll(result1Task, result2Task).ConfigureAwait(false);

    result1 = result1Task.Result;
    result2 = result1Task.Result;
}

另一种方法是将其包装在自己的方法中：

// could also be written as generic method GetResultAsync<T>(),
// but you should get the point :)
async Task<IEnumerable<T1>> GetResult1Async()
{
    using(var context = new MyContext())
        return await context.Set<T1>().ToListAsync().ConfigureAwait(false);
}

async Task<IEnumerable<T2>> GetResult2Async()
{
    using(var context = new MyContext())
        return await context.Set<T2>().ToListAsync().ConfigureAwait(false);
}

var result1Task = GetResult1Async();
var result2Task = GetResult2Async();

await Task.WhenAll(result1Task, result2Task).ConfigureAwait(false);

var result1 = result1Task.Result;
var result2 = result1Task.Result;

两者都有很多样板代码，所以我想写一个用于打开/处理上下文对象的概括。

我想把它简化成这样：

var result1Task = repository.Set<T1>().ToListAsync();
var result2Task = repository.Set<T2>().ToListAsync();

await Task.WhenAll(result1Task, result2Task).ConfigureAwait(false);

var result1 = result1Task.Result;
var result2 = result1Task.Result;

诀窍是我希望repository.Set<T>() 返回IQueryable<T>，这样我就可以使用所有Linq 方法。为此，我正在尝试创建自己的IQueryProvider，在其 Execute 方法中包装上下文 using-Block。

这是我目前的课程：

public class RepositoryQueryProvider<TContext> : IQueryProvider
    where TContext : DbContext, new()
{
    // some irrelevant code here

    public TResult Execute<TResult>(Expression expression)
    {
        using (var context = new TContext())
        {
            var efProvider = ?; // how to get the provider from the context?
            return efProvider.Execute<TResult>(expression);
        }
    }
}

Answer 1

这很简单：

dbcontext.GetService<IAsyncQueryProvider>();

而且我认为不需要对 DbContext 使用“使用”