如何为C# LINQ的Include方法添加Where条件并扩展IQueryable

要实现你需要的功能，核心是通过表达式树动态构建包含主实体和所有关联实体的无效状态过滤条件，同时完成关联实体的加载。以下是完整的实现方案：

完整代码实现 #

using System;
using System.Linq;
using System.Linq.Expressions;
using System.Reflection;

public static class QueryableExtensions
{
    // 统一配置无效状态属性名，可根据实际修改（比如原代码中的"Invalidated"）
    private const string InvalidPropertyName = "IsInvalid";

    public static IQueryable<TEntity> IncludeValidRelations<TEntity>(
        this IQueryable<TEntity> entities,
        params Expression<Func<TEntity, object>>[] includes)
        where TEntity : class
    {
        var result = entities;

        // 先执行所有关联实体的Include操作
        foreach (var include in includes)
        {
            result = result.Include(include);
        }

        var parameter = Expression.Parameter(typeof(TEntity), "x");
        Expression combinedCondition = null;

        // 构建主实体的无效状态过滤条件
        var mainInvalidProperty = typeof(TEntity).GetProperty(InvalidPropertyName);
        if (mainInvalidProperty != null)
        {
            var mainInvalidExpr = Expression.Property(parameter, InvalidPropertyName);
            combinedCondition = Expression.Equal(mainInvalidExpr, Expression.Constant(false));
        }

        // 遍历每个关联属性，构建对应的过滤条件
        foreach (var include in includes)
        {
            var navExpr = GetNavigationPropertyExpression(include);
            var navType = navExpr.Type;

            // 检查关联实体是否包含无效状态属性
            var navInvalidProperty = navType.GetProperty(InvalidPropertyName);
            if (navInvalidProperty == null)
                continue;

            // 构建：关联属性不为空 且 关联实体未标记为无效
            var nullCheck = Expression.NotEqual(navExpr, Expression.Constant(null, navType));
            var navInvalidExpr = Expression.Property(navExpr, InvalidPropertyName);
            var navInvalidCheck = Expression.Equal(navInvalidExpr, Expression.Constant(false));
            var navCondition = Expression.AndAlso(nullCheck, navInvalidCheck);

            // 合并到总过滤条件中
            combinedCondition = combinedCondition == null 
                ? navCondition 
                : Expression.AndAlso(combinedCondition, navCondition);
        }

        // 应用最终的过滤条件
        if (combinedCondition != null)
        {
            var predicate = Expression.Lambda<Func<TEntity, bool>>(combinedCondition, parameter);
            result = result.Where(predicate);
        }

        return result;
    }

    // 辅助方法：解析Include表达式，处理可能的装箱转换
    private static Expression GetNavigationPropertyExpression<TEntity>(Expression<Func<TEntity, object>> includeExpression)
    {
        var body = includeExpression.Body;
        // 处理值类型导航属性或object转换带来的UnaryExpression包装
        if (body is UnaryExpression unaryExpr && unaryExpr.NodeType == ExpressionType.Convert)
        {
            body = unaryExpr.Operand;
        }
        return body;
    }
}

关键实现细节 #

• 统一属性配置：通过InvalidPropertyName常量管理无效状态的属性名，只需修改一次即可适配不同的属性命名（比如你原代码中的Invalidated）。
• 先Include后过滤：确保所有需要的关联实体被加载，再应用过滤条件。
• 非空检查防护：为每个关联属性添加非空判断，避免空引用异常；如果你的场景中关联实体不可能为null，可以删除nullCheck部分。
• 动态条件合并：自动将主实体和所有关联实体的过滤条件用AndAlso组合，生成等价于手动编写的多条件Where语句。

使用示例 #

// 调用方式完全符合你的需求
var validBooks = _bookSet.IncludeValidRelations(x => x.Author, x => x.anotherProperty);

这段代码等价于手动编写：

var validBooks = _bookSet
    .Include(x => x.Author)
    .Include(x => x.anotherProperty)
    .Where(x => 
        x.IsInvalid == false && 
        (x.Author != null && x.Author.IsInvalid == false) && 
        (x.anotherProperty != null && x.anotherProperty.IsInvalid == false)
    );

补充说明 #

• 如果某个关联实体没有IsInvalid属性，方法会自动跳过该实体的过滤，不会抛出异常。
• 该方法支持所有继承自class的实体类型，只要实体包含指定的无效状态属性即可。

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