结合两个 linq 表达式来注入导航属性

Question

我有一组相同的条件，它们直接应用于一个类或其他具有相同导航属性的类。

例如，我有家，我有代理，代理与家相关联。

所以我正在寻找一个代理名称为'a'的房子，或者我正在寻找一个名称为'a'的代理，查询如下，

 Expression<<Func<Agent,bool>> ax = x=> x.Name == "a" ;

 Expression<Func<Home,bool>> hx = x=> x.Agent.Name == "a";

我有近 100 个针对 Agent 的搜索查询，而且我还必须将它们应用于 Home 可查询。我不介意再写一遍，但它很难维护，因为我们知道它们在开发过程中会经常改变。

我想要做的是，我想像这样为 Home 查询编写表达式，

 Expression<Func<Home,bool>> hx = Combine( x=>x.Agent , x=>x.Name == "a");

联合将跟随的地方，

 Expression<Func<T,bool>> Combine<T,TNav>( 
     Expression<Func<T,TNav>> parent, 
     Expression<Func<TNav,bool>> nav){

     // combine above to form...


     (x=>x.Agent , x=>x.Name == "a") 
         => x => x.Agent.Name == "a"

     (x=>x.Agent, x=>x.Name.StartsWith("a") || x.Name.EndsWith("a"))
         => x=>x.Agent.Name.StartsWith("a") || x.Agent.Name.EndsWith("a")

     // look carefully, x gets replaced by x.Agent in every node..

     // I know very little about expression rewriting, so I need little help
 }

Answer 1

是的，您确实需要访问者来替换部分原始表达式。你可以这样做：

Expression<Func<T,bool>> Combine<T,TNav>(Expression<Func<T,TNav>> parent, Expression<Func<TNav,bool>> nav)
{
     var param = Expression.Parameter(typeof(T), "x");
     var visitor = new ReplacementVisitor(parent.Parameters[0], param);
     var newParentBody = visitor.Visit(parent.Body);
     visitor = new ReplacementVisitor(nav.Parameters[0], newParentBody);
     var body = visitor.Visit(nav.Body);
     return Expression.Lambda<Func<T, bool>>(body, param);
}

public class ReplacementVisitor : System.Linq.Expressions.ExpressionVisitor
{
    private readonly Expression _oldExpr;
    private readonly Expression _newExpr;
    public ReplacementVisitor(Expression oldExpr, Expression newExpr)
    {
        _oldExpr = oldExpr;
        _newExpr = newExpr;
    }

    public override Expression Visit(Expression node)
    {
        if (node == _oldExpr)
            return _newExpr;
        return base.Visit(node);
    }

}

用法示例：

Expression<Func<Foo, Bar>> expr1 = f => f.Bar;
Expression<Func<Bar, bool>> expr2 = b => b.Baz;
var expr = Combine(expr1, expr2); // f => f.Bar.Baz