2

我有一个查询，它将用户的数据插入到数据库中。它工作正常，但users表在 上具有唯一索引username，因此当尝试使用已经存在的用户名创建行时，它会引发错误：

pub async fn create(user: UserRequest, pool: &PgPool) -> anyhow::Result<User> {
   let new_user = sqlx::query_as!(
       User,
       r#"
       INSERT INTO users (first_name, username)
       VALUES ($1, $2)
       RETURNING id, first_name, username
       "#,
       &user.first_name,
       &user.username,
   )
   .fetch_one(&pool.clone())
   .await?;
    
   Ok(User {
       id: new_user.id,
       first_name: new_user.first_name,
       username: new_user.username,
   })
}

&user是一个函数参数，定义为：

pub struct UserRequest {
    pub first_name: String,
    pub username: String,
}

使用此函数的端点match用于检查它是否是Ok(user)或其他任何东西，并返回其响应：

pub async fn create_user(
    user: web::Json<UserRequest>,
    pool: Data<PgPool>,
) -> impl Responder {
    let new_user = User::create(user.into_inner(), pool.get_ref()).await;
    match new_user {
        Ok(user) => HttpResponse::Ok().json(user),
        _ => HttpResponse::BadRequest().body(ERROR_MESSAGE),
    }
}

当由于该行试图被复制而出现错误时，我该怎么做才能捕捉到这种情况？情况是我需要在发生这种情况时返回不同的 HTTP 状态。

4

1 回答 1

3

您可以使用检查错误代码get_code。

match new_user {
    Ok(user) => HttpResponse::Ok().json(user),
    Err(err) if err.get_code() == "23505" {
    // returns unique constraint violation response.
    },
    _ => HttpResponse::BadRequest().body(ERROR_MESSAGE),
}

正如@MichaelAnderson 在评论中指出的那样，为了完成这项工作，

您需要将 create 的返回类型更改为Result<User, sqlx::postgres::PgDatabaseError>，或者不更改 create 的返回类型，您可以使用anyhow::Error::downcast或其相关函数返回到PgDatabaseError

于 2021-07-21T05:52:20.717 回答