与我对其他问题的回答相同的方法也适用于此。这是一个使用 EF5 的独立测试程序:
using System;
using System.Data.Entity;
using System.Linq;
using System.Linq.Expressions;
using System.Reflection;
namespace ScratchProject
{
public class A
{
public int Id { get; set; }
public string TextA { get; set; }
}
public class B
{
public int Id { get; set; }
public string TextB { get; set; }
}
public class MyContext : DbContext
{
public DbSet<A> As { get; set; }
public DbSet<B> Bs { get; set; }
protected IQueryProvider QueryProvider
{
get
{
IQueryable queryable = As;
return queryable.Provider;
}
}
public IQueryable<TResult> CreateScalarQuery<TResult>(Expression<Func<TResult>> expression)
{
return QueryProvider.CreateQuery<TResult>(
Expression.Call(
method: GetMethodInfo(() => Queryable.Select<int, TResult>(null, (Expression<Func<int, TResult>>)null)),
arg0: Expression.Call(
method: GetMethodInfo(() => Queryable.AsQueryable<int>(null)),
arg0: Expression.NewArrayInit(typeof(int), Expression.Constant(1))),
arg1: Expression.Lambda(body: expression.Body, parameters: new[] { Expression.Parameter(typeof(int)) })));
}
static MethodInfo GetMethodInfo(Expression<Action> expression)
{
return ((MethodCallExpression)expression.Body).Method;
}
}
static class Program
{
static void Main()
{
using (var context = new MyContext())
{
Console.WriteLine(context.CreateScalarQuery(() => context.As.Count(a => a.TextA != "A"))
.Concat(context.CreateScalarQuery(() => context.Bs.Count(b => b.TextB != "B"))));
}
}
}
}
输出:
SELECT
[UnionAll1].[C1] AS [C1]
FROM (SELECT
[GroupBy1].[A1] AS [C1]
FROM ( SELECT
COUNT(1) AS [A1]
FROM [dbo].[A] AS [Extent1]
WHERE N'A' <> [Extent1].[TextA]
) AS [GroupBy1]
UNION ALL
SELECT
[GroupBy2].[A1] AS [C1]
FROM ( SELECT
COUNT(1) AS [A1]
FROM [dbo].[B] AS [Extent2]
WHERE N'B' <> [Extent2].[TextB]
) AS [GroupBy2]) AS [UnionAll1]
是的,实际执行查询也按预期工作。
更新:
根据要求,您可以添加以下内容以使其正常工作Expression<Func<MyContext, TResult>> expression)
:
public IQueryable<TResult> CreateScalarQuery<TResult>(Expression<Func<MyContext, TResult>> expression)
{
var parameterReplacer = new ParameterReplacer(expression.Parameters[0], Expression.Property(Expression.Constant(new Tuple<MyContext>(this)), "Item1"));
return CreateScalarQuery(Expression.Lambda<Func<TResult>>(parameterReplacer.Visit(expression.Body)));
}
class ParameterReplacer : ExpressionVisitor
{
readonly ParameterExpression parameter;
readonly Expression replacement;
public ParameterReplacer(ParameterExpression parameter, Expression replacement)
{
this.parameter = parameter;
this.replacement = replacement;
}
protected override Expression VisitParameter(ParameterExpression node)
{
if (node == parameter)
return replacement;
return base.VisitParameter(node);
}
}
即使从当前上下文中调用,这也有效:
// member of MyContext
public void Test1()
{
Console.WriteLine(this.CreateScalarQuery(ctx => ctx.As.Count(a => a.TextA != "A"))
.Concat(this.CreateScalarQuery(ctx => ctx.Bs.Count(b => b.TextB != "B"))));
}
参数替换将上下文存储在 aTuple<MyContext>
而不是MyContext
直接,因为 EF 不知道如何处理Expression.Constant(this)
。这是 C# 编译器无论如何都不会产生的东西,因此 EF 不需要知道如何处理它。将上下文作为类的成员获取是 C# 编译器确实会产生的东西,因此 EF 已经知道如何处理它。
但是,如果您保存在局部变量中,CreateScalarQuery
也可以使用更简单的版本:this
// member of MyContext
public void Test2()
{
var context = this;
Console.WriteLine(this.CreateScalarQuery(() => context.As.Count(a => a.TextA != "A"))
.Concat(this.CreateScalarQuery(() => context.Bs.Count(b => b.TextB != "B"))));
}