编写带Expression参数的IQueryable扩展方法遇EF查询翻译错误求助

我想要编写一个IQueryable的扩展方法来筛选实体，该方法需包含一个Expression<Func<T, int>>类型的参数用于筛选。以下是我的尝试，但无法正常运行，求原因及解决办法：

尝试代码 #

public static IQueryable<T> AddDefaultQuery<T>(this IQueryable<T> query, Expression<Func<T, int>> countryId, ApplicationDbContext db, ClaimsPrincipal claimsPrincipal)
{
    if (claimsPrincipal.IsEppRole()) {
        var userCountryIds = db.UserCountry.Where(uc => uc.UserId == LoginUser.GetUserID(claimsPrincipal)).Select(uc => uc.CountryId);
        query = query.Where(q => userCountryIds.Contains(countryId.Compile()(q)));
    }
    return query;
}

报错信息 #

发生异常：CLR/System.InvalidOperationException
在Microsoft.EntityFrameworkCore.dll中引发了“System.InvalidOperationException”类型的异常：“无法转换LINQ表达式 'DbSet()
.Where(a => a.Deleted == False)
.Where(a => __userCountryIds_0.Contains(__Compile_1.Invoke(a)))'。请将查询重写为可转换的形式，或通过插入对“AsEnumerable”、“AsAsyncEnumerable”、“ToList”或“ToListAsync”的调用显式切换到客户端计算。”

原因分析 #

问题核心在countryId.Compile()(q)这行代码：

• Compile()会把Expression<Func<T, int>>转换成普通委托Func<T, int>
• EF Core的查询翻译器仅能识别表达式树，无法解析委托调用逻辑，自然无法将这段代码转换成对应SQL，最终抛出无法翻译的异常

解决办法 #

需要直接操作表达式树，将countryId表达式整合到Contains查询逻辑中，避免使用Compile()。可以通过System.Linq.Expressions命名空间的API构建完整查询表达式：

修改后的原生实现代码 #

using System.Linq.Expressions;

public static IQueryable<T> AddDefaultQuery<T>(this IQueryable<T> query, Expression<Func<T, int>> countryId, ApplicationDbContext db, ClaimsPrincipal claimsPrincipal)
{
    if (claimsPrincipal.IsEppRole())
    {
        var userId = LoginUser.GetUserID(claimsPrincipal);
        var userCountryIds = db.UserCountry.Where(uc => uc.UserId == userId).Select(uc => uc.CountryId);
        
        // 构建Where方法的参数表达式（对应lambda中的q）
        var parameter = Expression.Parameter(typeof(T), "q");
        // 将countryId表达式的参数替换为当前parameter，生成获取实体CountryId的表达式
        var countryIdExpr = Expression.Invoke(countryId, parameter);
        // 构建Contains调用的表达式：userCountryIds.Contains(countryIdExpr)
        var containsExpr = Expression.Call(
            typeof(Enumerable), 
            nameof(Enumerable.Contains), 
            new[] { typeof(int) }, 
            userCountryIds.Expression, 
            countryIdExpr);
        
        // 组装完整的Where条件lambda表达式
        var whereLambda = Expression.Lambda<Func<T, bool>>(containsExpr, parameter);
        query = query.Where(whereLambda);
    }
    return query;
}

简化方案（使用LINQKit） #

如果项目允许引入第三方包，可通过LINQKit简化表达式树的合并操作：

// 先安装LINQKit包：Install-Package LINQKit
using LINQKit;

public static IQueryable<T> AddDefaultQuery<T>(this IQueryable<T> query, Expression<Func<T, int>> countryId, ApplicationDbContext db, ClaimsPrincipal claimsPrincipal)
{
    if (claimsPrincipal.IsEppRole())
    {
        var userCountryIds = db.UserCountry.Where(uc => uc.UserId == LoginUser.GetUserID(claimsPrincipal)).Select(uc => uc.CountryId);
        // 调用AsExpandable启用表达式扩展，通过Invoke自动合并表达式树
        query = query.AsExpandable().Where(q => userCountryIds.Contains(countryId.Invoke(q)));
    }
    return query;
}

内容的提问来源于stack exchange，提问作者lRonnyx3