在日常开发中,IN
查询是非常常见的需求,你会怎么来处理呢?借助 sqlx 的QueryBuilder
可以方便地实现。
需求
假设有一个需求,需要返回一系列指定ID的会员信息的列表。对应的 SQL 语句大概这样:
注意,IN(???)
条件不是固定的,例中 IN(1,2,3,4)
只是举例,以便理解需求。
错误的实现
有其它语言开发经验的人肯定想到:
- 将参数合并为字符串
- 将合并后的字符串绑定到 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();
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)
}