feat: migrate to diesel and new error type in users
This commit is contained in:
parent
6190d76285
commit
dfe2ca9486
3 changed files with 59 additions and 88 deletions
|
@ -1,15 +1,19 @@
|
|||
use actix_web::{Error, HttpRequest, HttpResponse, get, web};
|
||||
use actix_web::{HttpRequest, HttpResponse, get, web};
|
||||
use diesel::{ExpressionMethods, QueryDsl, Queryable, Selectable, SelectableHelper};
|
||||
use diesel_async::RunQueryDsl;
|
||||
use log::error;
|
||||
use serde::Serialize;
|
||||
use uuid::Uuid;
|
||||
|
||||
use crate::{Data, api::v1::auth::check_access_token, utils::get_auth_header};
|
||||
use crate::{error::Error, api::v1::auth::check_access_token, schema::users::{self, dsl}, utils::get_auth_header, Data};
|
||||
|
||||
#[derive(Serialize, Clone)]
|
||||
#[derive(Serialize, Queryable, Selectable, Clone)]
|
||||
#[diesel(table_name = users)]
|
||||
#[diesel(check_for_backend(diesel::pg::Pg))]
|
||||
struct Response {
|
||||
uuid: String,
|
||||
uuid: Uuid,
|
||||
username: String,
|
||||
display_name: String,
|
||||
display_name: Option<String>,
|
||||
}
|
||||
|
||||
#[get("/{uuid}")]
|
||||
|
@ -22,17 +26,11 @@ pub async fn res(
|
|||
|
||||
let uuid = path.into_inner().0;
|
||||
|
||||
let auth_header = get_auth_header(headers);
|
||||
let auth_header = get_auth_header(headers)?;
|
||||
|
||||
if let Err(error) = auth_header {
|
||||
return Ok(error);
|
||||
}
|
||||
let mut conn = data.pool.get().await?;
|
||||
|
||||
let authorized = check_access_token(auth_header.unwrap(), &data.pool).await;
|
||||
|
||||
if let Err(error) = authorized {
|
||||
return Ok(error);
|
||||
}
|
||||
check_access_token(auth_header, &mut conn).await?;
|
||||
|
||||
let cache_result = data.get_cache_key(uuid.to_string()).await;
|
||||
|
||||
|
@ -42,25 +40,11 @@ pub async fn res(
|
|||
.body(cache_hit));
|
||||
}
|
||||
|
||||
let row = sqlx::query_as(&format!(
|
||||
"SELECT username, display_name FROM users WHERE uuid = '{}'",
|
||||
uuid
|
||||
))
|
||||
.fetch_one(&data.pool)
|
||||
.await;
|
||||
|
||||
if let Err(error) = row {
|
||||
error!("{}", error);
|
||||
return Ok(HttpResponse::InternalServerError().finish());
|
||||
}
|
||||
|
||||
let (username, display_name): (String, Option<String>) = row.unwrap();
|
||||
|
||||
let user = Response {
|
||||
uuid: uuid.to_string(),
|
||||
username,
|
||||
display_name: display_name.unwrap_or_default(),
|
||||
};
|
||||
let user: Response = dsl::users
|
||||
.filter(dsl::uuid.eq(uuid))
|
||||
.select(Response::as_select())
|
||||
.get_result(&mut conn)
|
||||
.await?;
|
||||
|
||||
let cache_result = data
|
||||
.set_cache_key(uuid.to_string(), user.clone(), 1800)
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue