2

我正在关注MSDN 上的一个示例系列,用于创建 LINQ 提供程序并且已经碰壁了。

我期望当我编写以下测试时,下面ExpressionVisitor的源代码中的子类会VisitMethodCall被调用。

[Fact]
public void DatabaseModeler_provides_table_modeler()
{
    var q = new LightmapQuery<AspNetRoles>(new SqliteProvider2());
    q.Where(role => role.Name == "Admin");
    var result = q.ToList();
}

相反,会发生该VisitConstant方法被调用。我认为这是因为当Provider实例化时,它分配了它的Expression属性 a ConstantExpression。我不确定我是否做错了什么,或者 MSDN 上的指南是否存在问题,阻止我获取包含Where方法调用的表达式。

IQueryable<T>这是我拥有的和IQueryProvider实现的源代码。

using System;
using System.Collections;
using System.Collections.Generic;
using System.Linq;
using System.Linq.Expressions;
using System.Reflection;
using System.Text;

namespace Lightmap.Querying
{
    internal class SqliteQueryTranslator : ExpressionVisitor
    {
        internal StringBuilder queryBuilder;

        internal string Translate(Expression expression)
        {
            this.queryBuilder = new StringBuilder();
            this.Visit(expression);
            return this.queryBuilder.ToString();
        }

        public override Expression Visit(Expression node)
        {
            return base.Visit(node);
        }

        protected override Expression VisitMethodCall(MethodCallExpression node)
        {
            if (node.Method.DeclaringType != typeof(IQueryable) && node.Method.Name != nameof(Enumerable.Where))
            {
                throw new NotSupportedException($"The {node.Method.Name} method is not supported.");
            }

            this.queryBuilder.Append($"SELECT * FROM {node.Method.DeclaringType.Name}");
            return node;
        }

        protected override Expression VisitConstant(ConstantExpression node)
        {
            return node;
        }

        private static Expression StripQuotes(Expression expression)
        {
            while (expression.NodeType == ExpressionType.Quote)
            {
                expression = ((UnaryExpression)expression).Operand;
            }

            return expression;
        }
    }

    public abstract class LightmapProvider : IQueryProvider
    {
        public IQueryable CreateQuery(Expression expression)
        {
            Type genericParameter = expression.GetType().GetGenericArguments().First();
            return (IQueryable)Activator.CreateInstance(typeof(LightmapQuery<>)
                .MakeGenericType(genericParameter), new object[] { this, expression });
        }

        public IQueryable<TElement> CreateQuery<TElement>(Expression expression) => new LightmapQuery<TElement>(this, expression);

        object IQueryProvider.Execute(Expression expression)
        {
            return this.Execute(expression);
        }

        public TResult Execute<TResult>(Expression expression)
        {
            return (TResult)this.Execute(expression);
        }

        public abstract string GetQueryText(Expression expression);

        public abstract object Execute(Expression expression);
    }

    public class SqliteProvider2 : LightmapProvider
    {
        public override object Execute(Expression expression)
        {
            var x = new SqliteQueryTranslator().Translate(expression);
            return Activator.CreateInstance(typeof(List<>).MakeGenericType(TypeCache.GetGenericParameter(expression.Type, t => true)));
        }

        public override string GetQueryText(Expression expression)
        {
            throw new NotImplementedException();
        }
    }

    public class LightmapQuery<TTable> : IOrderedQueryable<TTable>
    {
        public LightmapQuery(IQueryProvider provider)
        {
            this.Provider = provider;
            this.Expression = Expression.Constant(this);
        }

        public LightmapQuery(IQueryProvider provider, Expression expression)
        {
            if (!typeof(IQueryable<TTable>).IsAssignableFrom(expression.Type))
            {
                throw new Exception();
            }

            this.Expression = expression;
            this.Provider = provider;
        }

        public Type ElementType => typeof(TTable);

        public Expression Expression { get; }

        public IQueryProvider Provider { get; }

        public IEnumerator<TTable> GetEnumerator()
        {
            return (this.Provider.Execute<IEnumerable<TTable>>(Expression)).GetEnumerator();
        }

        IEnumerator IEnumerable.GetEnumerator()
        {
            return (Provider.Execute<IEnumerable>(Expression)).GetEnumerator();
        }
    }
}

编辑

更新了我的单元测试以实际存储IQueryablefrom .Where,我仍然没有看到我的VisitMethodCall调用。

[Fact]
public void DatabaseModeler_provides_table_modeler()
{
    var q = new LightmapQuery<AspNetRoles>(new SqliteProvider2())
        .Where(role => role.Name == "Admin");
    var result = q.ToList();
}
4

2 回答 2

1

经过一些试验和错误以及 IRC 的一些帮助后,我能够识别问题并解决它。问题是我的单元测试项目是一个 .Net Core 项目,没有对System.Linq.Queryable程序集的引用。

在我的单元测试中

[Fact]
public void DatabaseModeler_provides_table_modeler()
{
    var q = new LightmapQuery<AspNetRoles>(new SqliteProvider2());
    var q2 = q.Where(role => role.Name == "Admin");
    var result = q2.ToList();
}

Typeofq2是这"WhereEnumerableIterator'1"让我们意识到它没有返回一个IQueryable.

在此处输入图像描述

在将上述引用添加到 之后project.jsonTypeofq2变成了"LightmapQuery'1"我所期望的。有了这个,我的VisitMethodCall方法就没有问题了。

在此处输入图像描述

于 2016-02-15T19:37:58.077 回答
0

这个问题是微不足道的 - 您忘记分配应用的结果Where

q.Where(role => role.Name == "Admin");

改用这样的东西

var q = new LightmapQuery<AspNetRoles>(new SqliteProvider2())
    .Where(role => role.Name == "Admin");
var result = q.ToList();
于 2016-02-15T17:48:04.777 回答