如何获取 IQueryable.Count 的 ToTraceString

Question

我使用((ObjectQuery)IQueryable).ToTraceString() 来获取和调整将由 LINQ 执行的 SQL 代码。

我的问题是，与大多数 IQueryable 方法不同，IQueryable.Count 定义如下：

    public static int Count(this IQueryable source) {
        return (int)source.Provider.Execute(
            Expression.Call(
                typeof(Queryable), "Count",
                new Type[] { source.ElementType }, source.Expression));
    }

执行查询而不编译和返回 IQueryable。 我想通过这样的方式来做到这一点：

public static IQueryable CountCompile(this IQueryable source) {
    return source.Provider.CreateQuery(
        Expression.Call(
            typeof(Queryable), "Count",
            new Type[] { source.ElementType }, source.Expression));
}

但是 CreateQuery 给了我以下异常：

LINQ to Entities query expressions can only be constructed from instances that implement the IQueryable interface.

Answer 1

当我尝试这样做时，我想出了一个实际可行的答案。异常说“只能从实现 IQueryable 接口的实例构造”，所以答案似乎很简单：返回一个可查询的东西。返回.Count() 时可能吗？是的！

public partial class YourObjectContext
{
    private static MethodInfo GetMethodInfo(Expression<Action> expression)
    {
        return ((MethodCallExpression)expression.Body).Method;
    }
    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)) })));
    }
}

使用它：

var query = context.CreateScalarQuery(() => context.Entity.Count());
MessageBox.Show(((ObjectQuery)query).ToTraceString());

基本上，它的作用是将非 IQueryable 查询包装在子选择中。它将查询转换为

from dummy in new int[] { 1 }.AsQueryable()
select context.Entity.Count()

except 让上下文的 QueryProvider 处理查询。生成的 SQL 几乎是您所期望的：

SELECT 
[GroupBy1].[A1] AS [C1]
FROM ( SELECT 
    COUNT(1) AS [A1]
    FROM [dbo].[Entity] AS [Extent1]
)  AS [GroupBy1]

Answer 2

您不能为“Count”创建查询对象，因为它不返回 IQueryable（这是有道理的 - 它返回单个值）。

你有两个选择：

• （推荐）使用 eSQL：

  context.CreateQuery<YourEntity>("select count(1) from YourEntitySet").ToTraceString()
  

• 使用 Reflection 调用不执行 IQueryable 检查的私有方法（由于显而易见的原因，这是错误的，但如果您只是需要它进行调试，它可能会很方便）：

  public static IQueryable CountCompile(this IQueryable source)
  {
      // you should cache this MethodInfo
      return (IQueryable)source.Provider.GetType().GetMethod("CreateQuery", BindingFlags.NonPublic | BindingFlags.Instance, null,
                                          new[] {typeof (Expression), typeof (Type)}, null)
          .Invoke(source.Provider, new object[]
                                       {
                                           Expression.Call(
                                               typeof (Queryable), "Count",
                                               new[] {source.ElementType}, source.Expression),
                                           source.ElementType
                                       });
  }