使用动态数量的 .and() 创建 Diesel.rs 查询

Posted 2023-03-21

【中文标题】使用动态数量的 .and() 创建 Diesel.rs 查询

【英文标题】：Creating Diesel.rs queries with a dynamic number of .and()'s

【发布时间】：2018-07-19 15:51:27

【问题描述】：

在玩 Diesel 时，我在编写一个函数时遇到了困难，该函数将 Strings 的向量作为输入并执行以下操作：

将所有Strings 组合到一个大型查询中 在Connection 上运行查询 处理结果 返回Vec

如果我一步构建查询，它就可以正常工作：

fn get_books(authors: Vec<String>, connection: SqliteConnection) 
    use schema::ebook::dsl::*;

    let inner = author
        .like(format!("%%", authors[0]))
        .and(author.like(format!("%%", authors[1])))
        .and(author.like(format!("%%", authors[2])));

    ebook
        .filter(inner)
        .load::<Ebook>(&connection)
        .expect("Error loading ebook");

如果我尝试以更多步骤生成查询（需要使用输入向量的可变长度），我无法编译它：

fn get_books(authors: Vec<String>, connection: SqliteConnection) 
    use schema::ebook::dsl::*;

    let mut inner = author
        .like(format!("%%", authors[0]))
        .and(author.like(format!("%%", authors[1]))); // <1>

    inner = inner.and(author.like(format!("%%", authors[2]))); // <2>

    ebook
        .filter(inner)
        .load::<Ebook>(&connection)
        .expect("Error loading ebook");

这会产生以下错误：

inner = inner.and(author.like(format!("%%",authors[2])));
        ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ expected struct `diesel::expression::operators::Like`, found struct `diesel::expression::operators::And`

我不明白为什么 Rust 需要 Like 而不是 And。标记为<1> 的函数返回And，因此And 存储在inner 中。

我错过了什么？为什么第一个代码可以编译而第二个不能？生成这种查询的正确方法是什么？

【参考方案1】：

您需要做的第一件事是查看完整错误消息：

error[E0308]: mismatched types
  --> src/main.rs:23:13
   |
23 |     inner = inner.and(author.like(format!("%%", authors[2])));//<2>
   |             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ expected struct `diesel::expression::operators::Like`, found struct `diesel::expression::operators::And`
   |
   = note: expected type `diesel::expression::operators::And<diesel::expression::operators::Like<_, _>, _>`
              found type `diesel::expression::operators::And<diesel::expression::operators::And<diesel::expression::operators::Like<_, _>, diesel::expression::operators::Like<schema::ebook::columns::author, diesel::expression::bound::Bound<diesel::sql_types::Text, std::string::String>>>, _>`

它很长，但那是因为它完全合格。让我们缩短最后一部分：

expected type `And<Like<_, _>, _>`
   found type `And<And<Like<_, _>, Like<author, Bound<Text, String>>>, _>`

如果您查看and 的文档，您会发现每次调用and 都会消耗接收器并返回一个全新的类型——And：

fn and<T: AsExpression<Bool>>(self, other: T) -> And<Self, T::Expression>

这是 Diesel 生成强类型 SQL 表达式而没有运行时开销的核心能力。所有的工作都委托给类型系统。其实Diesel的创建者有an entire talk where he shows how far Diesel pushes the type system and what benefits it has。

回到您的问题，不可能将And<_, _> 与And<And<_, _>, _> 存储在同一位置，因为它们将具有不同的大小并且实际上是不同的类型。从根本上讲，这与尝试将整数存储在布尔值中相同。

事实上，没有没有方法知道你需要什么具体类型，因为你甚至不知道你将有多少条件——这取决于向量。

在这种情况下，我们必须放弃静态分派并通过 trait 对象 转向动态分派。 Diesel 在这种情况下有一个特定的特征（也有很好的例子）：BoxableExpression。

剩下的部分是将您的作者转换为like 表达式并将它们组合起来。然而，当authors 为空时，我们需要一个基本情况。为此，我们构建了一个微不足道的真实陈述 (author = author)。

#[macro_use]
extern crate diesel;

use diesel::SqliteConnection;
use diesel::prelude::*;
use diesel::sql_types::Bool;

mod schema 
    table! 
        ebook (id) 
            id -> Int4,
            author -> Text,
        
    


fn get_books(authors: Vec<String>, connection: SqliteConnection) 
    use schema::ebook::dsl::*;

    let always_true = Box::new(author.eq(author));
    let query: Box<BoxableExpression<schema::ebook::table, _, SqlType = Bool>> = authors
        .into_iter()
        .map(|name| author.like(format!("%%", name)))
        .fold(always_true, |query, item| 
            Box::new(query.and(item))
        );

    ebook
        .filter(query)
        .load::<(i32, String)>(&connection)
        .expect("Error loading ebook");


fn main() 

---

如果没有更好的 SQL 方法来执行此操作，我也不会感到惊讶。例如，PostgreSQL 似乎有 WHERE col LIKE ANY( subselect ) 和 WHERE col LIKE ALL( subselect ) 形式。