Как получить ToTraceString для IQueryable.Count

Я использую ((ObjectQuery)IQueryable).ToTraceString() для получения и настройки кода SQL, который будет выполняться LINQ.

Моя проблема в том, что в отличие от большинства 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.

Ответ 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()

за исключением того, что контекст QueryProvider обрабатывает запрос. Сгенерированный SQL - это в значительной степени то, что вы должны ожидать:

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

Ответ 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
                                     });
}