如何组合两个成员表达式树?

时间:2009-02-18 06:26:49

标签: c# .net linq-to-sql lambda expression-trees

我正在尝试将以下表达式合并到一个表达式中:item =&gt; item.sub,sub =&gt; sub.key成为item =&gt; item.sub.key。我需要这样做,所以我可以创建一个OrderBy方法,它将项目选择器分别带到键选择器。这可以使用OrderBy上的一个重载并提供IComparer<T>来完成,但它不会转换为SQL。

以下是一个方法签名,以进一步阐明我想要实现的目标,以及一个不起作用的实现,但应说明这一点。

    public static IOrderedQueryable<TEntity> OrderBy<TEntity, TSubEntity, TKey>(
        this IQueryable<TEntity> source, 
        Expression<Func<TEntity, TSubEntity>> selectItem, 
        Expression<Func<TSubEntity, TKey>> selectKey)
        where TEntity : class
        where TSubEntity : class 
    {
        var parameterItem = Expression.Parameter(typeof(TEntity), "item");
        ...
        some magic
        ...
        var selector = Expression.Lambda(magic, parameterItem);
        return (IOrderedQueryable<TEntity>)source.Provider.CreateQuery(
            Expression.Call(typeof(Queryable), "OrderBy", new Type[] { source.ElementType, selector.Body.Type },
                 source.Expression, selector
                 ));
    } 

将被称为:

.OrderBy(item => item.Sub, sub => sub.Key)

这可能吗?有没有更好的办法?我希望以这种方式工作的OrderBy方法的原因是支持适用于许多实体的复杂键选择表达式,尽管它们以不同方式公开。另外,我知道使用深度属性的字符串表示来实现此目的的方法,但我试图保持强类型。

3 个答案:

答案 0 :(得分:4)

由于这是LINQ-to-SQL,您通常可以使用Expression.Invoke来使用子表达式。我会看看能否提出一个例子(更新:完成)。但请注意,EF不支持此功能 - 您需要从头开始重建表达式。我有一些代码可以做到这一点,但它很长......

表达式代码(使用Invoke)非常简单:

var param = Expression.Parameter(typeof(TEntity), "item");
var item = Expression.Invoke(selectItem, param);
var key = Expression.Invoke(selectKey, item);
var lambda = Expression.Lambda<Func<TEntity, TKey>>(key, param);
return source.OrderBy(lambda);

以下是Northwind的示例用法:

using(var ctx = new MyDataContext()) {
    ctx.Log = Console.Out;
    var rows = ctx.Orders.OrderBy(order => order.Customer,
        customer => customer.CompanyName).Take(20).ToArray();
}

使用TSQL(重新格式化以适应):

SELECT TOP (20) [t0].[OrderID], -- snip
FROM [dbo].[Orders] AS [t0]
LEFT OUTER JOIN [dbo].[Customers] AS [t1]
  ON [t1].[CustomerID] = [t0].[CustomerID]
ORDER BY [t1].[CompanyName]

答案 1 :(得分:1)

你有什么东西,然后投射,然后再次排序。

.OrderBy(x => x.Sub)
    .Select(x => x.Sub)
        .OrderBy(x => x.Key)

你的方法可能是这样的:

public static IOrderedQueryable<TSubEntity> OrderBy<TEntity, TSubEntity, TKey>(
    this IQueryable<TEntity> source, 
    Expression<Func<TEntity, TSubEntity>> selectItem, 
    Expression<Func<TSubEntity, TKey>> selectKey)
    where TEntity : class
    where TSubEntity : class 
{
    return (IOrderedQueryable<TSubEntity>)source.
        OrderBy(selectItem).Select(selectItem).OrderBy(selectKey)
}

这将由SQL执行,但您可能已经注意到我必须将返回类型更改为IOrderedQueryable&lt; TSubEntity&gt;。你可以解决这个问题吗?

答案 2 :(得分:1)

I needed the same so made this small extension method:

    /// <summary>
    /// From A.B.C and D.E.F makes A.B.C.D.E.F. D must be a member of C.
    /// </summary>
    /// <param name="memberExpression1"></param>
    /// <param name="memberExpression2"></param>
    /// <returns></returns>
    public static MemberExpression JoinExpression(this Expression memberExpression1, MemberExpression memberExpression2)
    {
        var stack = new Stack<MemberInfo>();
        Expression current = memberExpression2;
        while (current.NodeType != ExpressionType.Parameter)
        {
            var memberAccess = current as MemberExpression;
            if (memberAccess != null)
            {
                current = memberAccess.Expression;
                stack.Push(memberAccess.Member);
            }
            else
            {
                throw new NotSupportedException();
            }
        }


        Expression jointMemberExpression = memberExpression1;
        foreach (var memberInfo in stack)
        {
            jointMemberExpression = Expression.MakeMemberAccess(jointMemberExpression, memberInfo);
        }

        return (MemberExpression) jointMemberExpression;
    }