使用ExpressionVisitor在SelectMany后无法找到Where表达式的问题

看起来你的CommonExpressionVisitor在处理SelectMany时的逻辑有漏洞，导致无法定位到关联的Where表达式，下面具体说原因和修复方式：

核心问题 #

你的当前实现只处理了SelectMany（以及Where/Select）的第一个参数（源查询），但忽略了两个关键部分：

• SelectMany的其他参数：SelectMany有多个重载，通常第二个参数是集合选择器（比如x => x.Children），如果这个选择器内部包含Where表达式（比如x => x.Children.Where(y => y.IsActive)），你的代码根本不会遍历到这个子表达式。
• 未调用基类的默认遍历逻辑：ExpressionVisitor的base.VisitMethodCall(node)会自动递归遍历方法调用的所有子节点（包括方法的实例、所有参数），你直接return node跳过了这一步，导致很多深层或后续的表达式节点（比如链式调用里SelectMany之后的Where）没有被访问到。

修复方案 #

方案1：手动遍历所有必要参数 #

如果你需要对这些Queryable方法做自定义处理，可以手动遍历所有相关参数，确保没有遗漏：

class CommonExpressionVisitor : ExpressionVisitor 
{
    protected override Expression VisitMethodCall(MethodCallExpression node) 
    {
        if (node.Method.DeclaringType == typeof(Queryable)) 
        {
            switch (node.Method.Name) 
            {
                case nameof(Queryable.Where): 
                    // 遍历源查询和Where的谓词表达式
                    Visit(node.Arguments[0]);
                    Visit(node.Arguments[1]);
                    return node;
                case nameof(Queryable.Select): 
                    // 遍历源查询和Select的投影表达式
                    Visit(node.Arguments[0]);
                    Visit(node.Arguments[1]);
                    return node;
                case nameof(Queryable.SelectMany): 
                    // 遍历源查询、集合选择器，若有结果选择器也一并处理
                    Visit(node.Arguments[0]);
                    Visit(node.Arguments[1]);
                    if (node.Arguments.Count > 2)
                    {
                        Visit(node.Arguments[2]);
                    }
                    return node;
                // 其他Queryable方法可按需添加处理逻辑
            }
        }
        // 非Queryable方法调用，交给基类处理所有子节点
        return base.VisitMethodCall(node);
    }
}

方案2：依赖基类遍历逻辑（更简洁） #

如果你的主要需求是找到所有Where表达式，不需要对Queryable方法做特殊修改，只需要在识别到Where时做记录，然后依赖基类自动遍历整个表达式树即可：

class CommonExpressionVisitor : ExpressionVisitor 
{
    // 可以添加一个列表来收集找到的Where表达式
    public List<MethodCallExpression> WhereExpressions { get; } = new List<MethodCallExpression>();

    protected override Expression VisitMethodCall(MethodCallExpression node) 
    {
        if (node.Method.DeclaringType == typeof(Queryable)) 
        {
            switch (node.Method.Name) 
            {
                case nameof(Queryable.Where): 
                    WhereExpressions.Add(node);
                    Console.WriteLine("找到Where表达式");
                    break;
                case nameof(Queryable.SelectMany): 
                    Console.WriteLine("找到SelectMany表达式");
                    break;
            }
        }
        // 关键：调用基类方法，自动递归遍历所有子节点
        return base.VisitMethodCall(node);
    }
}

验证逻辑 #

比如对于链式查询：

var query = dbContext.Parents
    .SelectMany(p => p.Children.Where(c => c.Age > 10))
    .Where(pc => pc.IsActive);

使用修复后的Visitor，会：

1. 先遍历最外层的Where，添加到收集列表
2. 遍历Where的源参数（也就是SelectMany节点）
3. 遍历SelectMany的源参数（Parents）和集合选择器（p => p.Children.Where(...)）
4. 遍历集合选择器里的Where，同样添加到收集列表

这样不管Where是在SelectMany之前、之后还是内部，都会被正确找到。

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