EF Core中Filtered Include无法使用自定义扩展方法的原因及解决办法

问题背景 #

我尝试限制导航属性的元素数量，根据EF Core文档，导航集合上可以使用Where、OrderBy、OrderByDescending、ThenBy、ThenByDescending、Skip或Take操作。

原有代码运行正常：

public async Task<List<CategoryWithRecipeDto>> GetCategories(int recipeCount)
{
    var categories = _context.Categories
        .Include(cat => cat.Recipes.Take(recipeCount));

    var dtos = await categories
        .Select(cat => _mapper.Map<CategoryWithRecipeDto>(cat))
        .ToListAsync();

    return dtos;
}

现在想让recipeCount参数可选（设为可空类型），于是写了TakeMaybe扩展方法，但在Include中使用时抛出异常：

System.InvalidOperationException:
The expression 'cat.Recetas.TakeMaybe(__queryParameters_CantRecetas_0)' is invalid inside an 'Include' operation, since it does not represent a property access: 't => t.MyProperty'.
To target navigations declared on derived types, use casting ('t => ((Derived)t).MyProperty') or the 'as' operator ('t => (t as Derived).MyProperty').
Collection navigation access can be filtered by composing Where, OrderBy(Descending), ThenBy(Descending), Skip or Take operations.

我很困惑，因为这个扩展方法要么调用Take，要么直接返回集合，推测和EF Core的查询转换机制有关，但不确定具体原因。我可以通过条件构建查询实现需求，但希望复用为常规查询写的扩展方法，想知道：为什么Include里不能用这类自定义扩展？还有哪些其他实现方式？

原因分析 #

EF Core处理Include中的导航集合筛选时，只会识别官方文档列出的那几个内置方法，它需要直接解析表达式树里的方法调用逻辑。而自定义扩展方法对EF Core来说是个“黑盒”——它没法拆解扩展方法内部的代码，不知道你到底是调用了Take还是返回原集合，所以会判定这个表达式不符合要求，抛出异常。

说白了就是：EF Core的查询转换器没能力理解你的自定义扩展，它只认自己内置的那几个方法调用。

替代实现方式 #

1. 条件构建Include表达式 #

直接在查询里根据recipeCount是否有值来决定是否添加Take，这是最直接的方案：

public async Task<List<CategoryWithRecipeDto>> GetCategories(int? recipeCount)
{
    var query = _context.Categories.AsQueryable();

    if (recipeCount.HasValue)
    {
        query = query.Include(cat => cat.Recipes.Take(recipeCount.Value));
    }
    else
    {
        query = query.Include(cat => cat.Recipes);
    }

    var dtos = await query
        .Select(cat => _mapper.Map<CategoryWithRecipeDto>(cat))
        .ToListAsync();

    return dtos;
}

2. 封装成可解析的表达式树方法 #

如果想复用逻辑，可以封装一个返回表达式树的方法（不是扩展方法），这样EF Core能解析内部的Take逻辑：

public static class QueryExtensions
{
    public static Expression<Func<Category, IEnumerable<Recipe>>> TakeRecipes(int? count)
    {
        return count.HasValue 
            ? (Expression<Func<Category, IEnumerable<Recipe>>>)(cat => cat.Recipes.Take(count.Value)) 
            : cat => cat.Recipes;
    }
}

调用时直接传入这个表达式：

public async Task<List<CategoryWithRecipeDto>> GetCategories(int? recipeCount)
{
    var categories = _context.Categories
        .Include(QueryExtensions.TakeRecipes(recipeCount));

    var dtos = await categories
        .Select(cat => _mapper.Map<CategoryWithRecipeDto>(cat))
        .ToListAsync();

    return dtos;
}

这种方式返回的是EF Core能理解的表达式树，而非自定义方法调用，所以可以正常工作。

3. 使用投影替代Include（推荐） #

如果最终是要映射到DTO，完全可以不用Include，直接在Select里做投影并限制数量，这样更灵活，还能避免Include带来的额外数据加载：

public async Task<List<CategoryWithRecipeDto>> GetCategories(int? recipeCount)
{
    var dtos = await _context.Categories
        .Select(cat => new CategoryWithRecipeDto
        {
            // 映射Category的属性
            Id = cat.Id,
            Name = cat.Name,
            // 处理Recipes集合
            Recipes = recipeCount.HasValue 
                ? cat.Recipes.Take(recipeCount.Value).Select(r => _mapper.Map<RecipeDto>(r)).ToList() 
                : cat.Recipes.Select(r => _mapper.Map<RecipeDto>(r)).ToList()
        })
        .ToListAsync();

    return dtos;
}

这种方式不依赖Include的筛选规则，完全由投影控制返回的数据，性能也更好，因为只会加载你需要的字段。

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