确保 IQueryable<T>.Where() 运行 SQL 查询而不是在内存中过滤

Question

这段代码将我的整个表格下载到内存中并在那里过滤。

class Book 
{
    Id<Book> BookId;
    Id<Author> AuthorId;
}

class Person 
{
    Id<Person> PersonId;
}

class BookLookup 
{
    List<Book> GetBooksForPerson(DbContext db, Id<Person> personId) 
    {
        // yes authorId maps 1-to-1 to personId, legacy
        // codebase, can't change that right now
        return db.Books
                 .Where(b => (int)b.AuthorId == (int)personId)
                 .ToList();
    }
}

此代码在 SQL 中运行，仅下载过滤后的行。

class BookLookup 
{
    List<Book> GetBooksForPerson(DbContext db, Id<Person> personId) 
    {
        // yes authorId maps 1-to-1 to personId, legacy
        // codebase, can't change that right now
        var authorId = (Id<Author>)(int)personId;

        return db.Books
                 .Where(b => b.AuthorId == authorId)
                 .ToList();
    }
}

两种解决方案都可以编译和工作，但一种会导致巨大的性能问题。

除了非常小心之外，如何防止 LINQ 下载我的整个表？

我想要这样的东西，当它无法转换为 sql 时，它会抛出或编译失败。

db.Books.WhereDb(b => (int)b.AuthorId == (int)personId)
db.Books.Where(b => (int)b.AuthorId == (int)personId, 
                FilterOption.MustRunInDb)

已编辑以包含 id 类和 EF 版本：

public struct Id<T> : IEquatable<Id<T>>, IComparable
{
    private readonly int _value;

    private Id(int value)
    {
        _value = value;
    }
    
    public static explicit operator int(Id<T> instance)
    {
        return instance._value;
    }

    public static explicit operator Id<T>(int value)
    {
        return new Id<T>(value);
    }

    public static bool operator ==(Id<T> value1, Id<T> value2)
    {
        return value1.Equals(value2);
    }

    public static bool operator !=(Id<T> value1, Id<T> value2)
    {
        return !(value1 == value2);
    }

    public bool Equals(Id<T> other)
    {
        return _value.Equals(other._value);
    }

    public override bool Equals(object obj)
    {
        if (ReferenceEquals(null, obj)) return false;
        return obj is Id<T> type && Equals(type);
    }

    public override int GetHashCode()
    {
        return _value.GetHashCode();
    }

    /// <summary>
    /// Prints the debug string representation of this object
    /// </summary>
    /// <returns></returns>
    public override string ToString()
    {
        var typeName = GetType().GenericTypeArguments.First().Name;
        return $"Id<{typeName}>:{_value}";
    }

    public int CompareTo(Id<T> obj)
    {
        return _value.CompareTo(obj._value);
    }

    public int CompareTo(object obj)
    {
        if (ReferenceEquals(null, obj)) return 1;

        if(obj is Id<T> type)
        {
            return CompareTo(type);
        }

        return 1;
    }
}

使用 EF Core (v2.1.4) 和自定义助手将 Id<T> 转换为 linq-to-sql 中的 int

Answer 1

在 Entity Framework Core 中，如果它无法评估子句，它将悄悄地评估内存中的所有内容。请注意，我悄悄地说是因为您应该在日志中收到一条消息，告诉您它已经发生了。

您可以通过配置上下文来更改此行为，例如：

protected override void OnConfiguring(DbContextOptionsBuilder optionsBuilder)
{
    optionsBuilder
        .UseSqlServer("<connection string>")
        .ConfigureWarnings(warnings => 
            warnings.Throw(RelationalEventId.QueryClientEvaluationWarning));
}

更多信息请参见this doc。

您也可以在启动代码中执行此操作，例如：

services.AddDbContext<ApplicationDbContext>(options =>
{
    options.UseSqlServer(Configuration.GetConnectionString("DefaultConnection"));
    options.ConfigureWarnings(warnings => 
        warnings.Throw(RelationalEventId.QueryClientEvaluationWarning);
});