sqlx优雅地实现IN查询

在日常开发中，IN 查询是非常常见的需求，你会怎么来处理呢？借助 sqlx 的QueryBuilder 可以方便地实现。

需求

假设有一个需求，需要返回一系列指定ID的会员信息的列表。对应的 SQL 语句大概这样：

SELECT * FROM member WHERE id IN(1,2,3,4);

错误的实现

• 将参数合并为字符串
• 将合并后的字符串绑定到 sqlx 中

sqlx 的 QueryBuilder 提供了 push_tuples()方法，它能更优雅、可靠地实现这一需求。

正确的实现

pub async fn select_in(conn: &sqlx::MySqlPool, ids: &[u32]) -> Result<Vec<model::member::Member>> {
    let mut q = sqlx::QueryBuilder::new("SELECT * FROM member WHERE id IN");
    q.push_tuples(ids.iter(), |mut b, id| {
        b.push_bind(id);
    });

    let ms = q
        .build_query_as()
        .fetch_all(conn)
        .await
        .map_err(Error::from)?;

    Ok(ms)
}