Question

我已经实现了一个基本的（天真的？）LINQ提供程序，它可以用于我的目的，但是我想解决一些怪癖，但我不确定如何。例如：

// performing projection with Linq-to-Objects, since Linq-to-Sage won't handle this:
var vendorCodes = context.Vendors.ToList().Select(e => e.Key);

我的IQueryProvider实施有一个CreateQuery<TResult>实现，如下所示：

public IQueryable<TResult> CreateQuery<TResult>(Expression expression)
{
    return (IQueryable<TResult>)Activator
        .CreateInstance(typeof(ViewSet<>)
        .MakeGenericType(elementType), _view, this, expression, _context);
}

当Expression是MethodCallExpression且TResult是string时，显然这会产生窒息，所以我想我会执行事情：

public IQueryable<TResult> CreateQuery<TResult>(Expression expression)
{
    var elementType = TypeSystem.GetElementType(expression.Type);
    if (elementType == typeof(EntityBase))
    {
        Debug.Assert(elementType == typeof(TResult));
        return (IQueryable<TResult>)Activator.CreateInstance(typeof(ViewSet<>).MakeGenericType(elementType), _view, this, expression, _context);
    }

    var methodCallExpression = expression as MethodCallExpression;
    if(methodCallExpression != null && methodCallExpression.Method.Name == "Select")
    {
        return (IQueryable<TResult>)Execute(methodCallExpression);
    }

    throw new NotSupportedException(string.Format("Expression '{0}' is not supported by this provider.", expression));
}

所以当我运行var vendorCodes = context.Vendors.Select(e => e.Key);时，我最终会进入private static object Execute<T>(Expression,ViewSet<T>)重载，它会打开最里面的过滤器表达式的方法名称并在底层API中进行实际调用。

现在，在这种情况下，我正在传递Select方法调用表达式，因此过滤器表达式为null并且我的switch块被跳过 - 这很好 - 我在哪里我被困在这里：

var method = expression as MethodCallExpression;
if (method != null && method.Method.Name == "Select")
{
    // handle projections
    var returnType = method.Type.GenericTypeArguments[0];
    var expType = typeof (Func<,>).MakeGenericType(typeof (T), returnType);

    var body = method.Arguments[1] as Expression<Func<T,object>>;
    if (body != null)
    {
        // body is null here because it should be as Expression<Func<T,expType>>
        var compiled = body.Compile();
        return viewSet.Select(string.Empty).AsEnumerable().Select(compiled);
    }
}

我需要对MethodCallExpression做什么才能将其传递给LINQ-to-Objects'Select方法？我是否正确接近这个？

Answer 1

（credits to Sergey Litvinov）

以下是有效的代码：

var method = expression as MethodCallExpression;
if (method != null && method.Method.Name == "Select")
{
    // handle projections
    var lambda = ((UnaryExpression)method.Arguments[1]).Operand as LambdaExpression;
    if (lambda != null)
    {
        var returnType = lambda.ReturnType;
        var selectMethod = typeof(Queryable).GetMethods().First(m => m.Name == "Select");
        var typedGeneric = selectMethod.MakeGenericMethod(typeof(T), returnType);
        var result = typedGeneric.Invoke(null, new object[] { viewSet.ToList().AsQueryable(), lambda }) as IEnumerable;
        return result;
    }
}

现在这个：

var vendorCodes = context.Vendors.ToList().Select(e => e.Key);

可以这样：

var vendorCodes = context.Vendors.Select(e => e.Key);

你甚至可以这样做：

var vendors = context.Vendors.Select(e => new { e.Key, e.Name });

关键是直接从Select类型获取Queryable方法，使用lambda的returnType使其成为通用方法，然后从viewSet.ToList().AsQueryable()调用它。

如何使LINQ-to-Objects处理投影？

1 个答案: