使用sqlx的QueryBuilder构建复杂、动态的查询

**十分重要:和 query()/query_as()不同,QueryBuilderpush()方法的 SQL 语句里,不需要使用 ? 占位符

使用 QueryBuilder 改造 list()

改造前:

pub async fn list(
    conn: &sqlx::MySqlPool,
    page: u32,
) -> Result<Paginate<Vec<model::member::Member>>> {
    let count: (i64,) = sqlx::query_as("SELECT COUNT(*) FROM member")
        .fetch_one(conn)
        .await
        .map_err(Error::from)?;

    let sql = format!(
        "SELECT * FROM member ORDER BY id DESC LIMIT {} OFFSET {}",
        DEFAULT_PAGE_SIZE,
        page * DEFAULT_PAGE_SIZE
    );
    let data = sqlx::query_as(&sql)
        .fetch_all(conn)
        .await
        .map_err(Error::from)?;

    Ok(Paginate::new(count.0 as u32, page, data))
}

改造后:

我们使用 QueryBuilder 代替了之前使用 format! 拼接 SQL 的做法。

使用 QueryBuilder 改造 exists()

改造前:

pub async fn exists(conn: &sqlx::MySqlPool, name: &str, id: Option<u32>) -> Result<bool> {
    let sql = "SELECT COUNT(*) FROM member WHERE name=?";
    let with_id_sql: String;

    let q = match id {
        Some(id) => {
            with_id_sql = format!("{} AND id<>?", sql);
            sqlx::query_as(&with_id_sql).bind(&name).bind(id)
        }
        None => sqlx::query_as(sql).bind(&name),
    };

    let count: (i64,) = q.fetch_one(conn).await.map_err(Error::from)?;

    Ok(count.0 > 0)
}

改造后:

对比发现,使用 QueryBuilder 改造后的 exists() 逻辑清晰,代码简洁。

再次提醒:**和 query()/query_as()不同,QueryBuilderpush()方法的 SQL 语句里,不需要使用 ? 占位符

再次提醒:**和 query()/query_as()不同,QueryBuilderpush()方法的 SQL 语句里,不需要使用 ? 占位符

本章代码位于06/QueryBuilder分支 。

要查看完整内容,请先登录