遍历表达式树并提取参数

Gun*_*Gun 5 c# linq reflection lambda traversal

我正在编写一种映射工具。我有一个看起来像这样的方法(简化):

   public void RegisterMapping<TTarget, TSource>(string propertyName, 
                                                 Expression<Func<TSource, object>> memberMap)
Run Code Online (Sandbox Code Playgroud)

是memberMap一个表达式,定义如何将属性从 转换TSource为TTarget。对于业务逻辑,我需要TSource从中提取对属性的所有引用。例如,从

x => x.Customers.Where(c => c.Orders.Any())
Run Code Online (Sandbox Code Playgroud)

我想得到Customers, 并从

x => x.FirstName + " " + x.LastName
Run Code Online (Sandbox Code Playgroud)

FirstName并且LastName(可以是 string[],PropertyInfo转换起来很简单)。

我该怎么做?我的第一种方法是手动遍历树,检查节点类型并根据节点类型检查不同的属性(例如,Operand对于一元表达式,Arguments对于函数调用)以确定其中任何一个是 的属性TSource。然后我发现了表达式种类列表,我放弃了——即使我只支持最常见的种类,这仍然是大量的工作。然后我找到了ExpressionVisitor。它看起来更好,但重写访问者方法仍然需要做很多工作,在我投入时间之前,我想知道是否还有其他选择,可能使用更专业的框架。

Mat*_*igh 6

我认为正如您所说,使用ExpressionVisitor是一种很好的方法。您不需要实现所有Visit...方法,因为它们已经有默认实现。据我了解,你想要的是找到 lambda 函数内某种类型的所有属性访问

public class MemberAccessVisitor : ExpressionVisitor
{
    private readonly Type declaringType;
    private IList<string> propertyNames = new List<string>();

    public MemberAccessVisitor(Type declaringType)
    {
        this.declaringType = declaringType;
    }

    public IEnumerable<string> PropertyNames { get { return propertyNames; } }

    public override Expression Visit(Expression expr)
    {
        if (expr != null && expr.NodeType == ExpressionType.MemberAccess)
        {
            var memberExpr = (MemberExpression)expr;
            if (memberExpr.Member.DeclaringType == declaringType)
            {
                propertyNames.Add(memberExpr.Member.Name);
            }
        }

        return base.Visit(expr);
    }
}
Run Code Online (Sandbox Code Playgroud)

通过检查成员是否是属性以及获取而PropertyInfo不是字符串,可以进一步改进您想要的内容

它可以如下使用:

var visitor = new MemberAccessVisitor(typeof(TSource));

visitor.Visit(memberMap);

var propertyNames = visitor.PropertyNames;
Run Code Online (Sandbox Code Playgroud)