域名 AXUM.RS 将于2025年10月到期。我们无意再对其进行续费,我们希望你能够接续这个域名,让更多 AXUM 开发者继续受益。现在,我们已启用新域名 AXUM.EU.ORG
  • 方案AXUM.RS 域名 = 3000
如果你有意接续这份 AXUM 情怀,请与我们取得联系。
说明:
  1. 如果有人购买 AXUM.RS 域名,或者该域名到期,本站将使用免费域名 AXUM.EU.ORG 继续提供服务。

sqlx优雅地实现IN查询

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

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

错误的实现

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

let ids_str:Vec<String> = ids.iter().map(|id|id.to_string()).collect(); 
let ids_arg:String = ids_str.join(",");
let list:Vec<Member> = sqlx::query_as("SELECT * FROM member WHERE id IN(?)").bind(&ids_arg).fetch_all(conn).await.unwrap();

这种方式不会报错,甚至能工作的很好。但也可能遇到未知的BUG。

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)
}
要查看完整内容,请先登录