Csharp EF Core判断是否应用了AsSplitQuery()

发布于04月09日

我正在使用Entity Framework核心查询SQL Server数据库.我想知道是否有一种方法来判断是否在执行一个现有的查询之前应用了AsSplitQuery()？我知道，如果你输出查询ToQueryString()，会有一条消息说:

这个LINQ查询是以拆分查询模式执行的，所显示的SQL是第一个要执行的查询.还可以根据第一查询的结果执行附加查询.

然而，这似乎不是解决我问题的干净方法.任何帮助将不胜感激.

推荐答案

你可以使用简单的扩展方法来判断查询表达式中是否使用了AsSplitQuery:

var isSplitQuery = query.IsUsingSplitQuery();

和执行:

public static class QueryRelationalExtensions
{
    public static bool IsUsingSplitQuery<T>(this IQueryable<T> query)
    {
        var current = query.Expression;

        while (current is MethodCallExpression methodCall)
        {
            if (methodCall.Method.DeclaringType == typeof(RelationalQueryableExtensions))
            {
                if (methodCall.Method.Name == nameof(RelationalQueryableExtensions.AsSplitQuery))
                    return true;
                if (methodCall.Method.Name == nameof(RelationalQueryableExtensions.AsSingleQuery))
                    return false;
            }

            if (methodCall.Arguments.Count == 0)
                break;

            current = UnwrapConvert(methodCall.Arguments[0]);
        }

        return false;
    }

    [return: NotNullIfNotNull(nameof(ex))]
    static Expression? UnwrapConvert(Expression? ex)
    {
        if (ex == null)
            return null;

        switch (ex.NodeType)
        {
            case ExpressionType.ConvertChecked:
            case ExpressionType.Convert:
            {
                if (((UnaryExpression)ex).Method == null)
                    return UnwrapConvert(((UnaryExpression)ex).Operand);
                break;
            }
        }

        return ex;
    }
}

注意，这种方法不判断QuerySplittingBehavior，它也可以添加，但与传递DbContext到IsUsingSplitQuery的方法.