Handle database restarts
This commit is contained in:
@ -14,16 +14,12 @@ use std::io::ErrorKind;
|
||||
|
||||
/// Get the information of a user, by its id
|
||||
pub async fn get_by_id(id: UserID) -> anyhow::Result<User> {
|
||||
db_connection::execute(|conn| Ok(users::table.filter(users::dsl::id.eq(id.0)).first(conn)?))
|
||||
db_connection::execute(|conn| users::table.filter(users::dsl::id.eq(id.0)).first(conn))
|
||||
}
|
||||
|
||||
/// Get the information of a user, by its mail
|
||||
pub async fn get_by_mail(mail: &str) -> anyhow::Result<User> {
|
||||
db_connection::execute(|conn| {
|
||||
Ok(users::table
|
||||
.filter(users::dsl::email.eq(mail))
|
||||
.first(conn)?)
|
||||
})
|
||||
db_connection::execute(|conn| users::table.filter(users::dsl::email.eq(mail)).first(conn))
|
||||
}
|
||||
|
||||
/// Get the information of a user, by its password reset token
|
||||
@ -36,14 +32,14 @@ pub async fn get_by_pwd_reset_token(token: &str) -> anyhow::Result<User> {
|
||||
}
|
||||
|
||||
db_connection::execute(|conn| {
|
||||
Ok(users::table
|
||||
users::table
|
||||
.filter(
|
||||
users::dsl::reset_password_token.eq(token.to_string()).and(
|
||||
users::dsl::time_gen_reset_token
|
||||
.ge(time() as i64 - PASSWORD_RESET_TOKEN_DURATION.as_secs() as i64),
|
||||
),
|
||||
)
|
||||
.first(conn)?)
|
||||
.first(conn)
|
||||
})
|
||||
}
|
||||
|
||||
@ -57,14 +53,14 @@ pub async fn get_by_account_delete_token(token: &str) -> anyhow::Result<User> {
|
||||
}
|
||||
|
||||
db_connection::execute(|conn| {
|
||||
Ok(users::table
|
||||
users::table
|
||||
.filter(
|
||||
users::dsl::delete_account_token.eq(token.to_string()).and(
|
||||
users::dsl::time_gen_delete_account_token
|
||||
.ge(time() as i64 - ACCOUNT_DELETE_TOKEN_DURATION.as_secs() as i64),
|
||||
),
|
||||
)
|
||||
.first(conn)?)
|
||||
.first(conn)
|
||||
})
|
||||
}
|
||||
|
||||
@ -213,23 +209,20 @@ pub async fn validate_account(user: &mut User) -> anyhow::Result<()> {
|
||||
/// Update account information
|
||||
pub async fn update_account(user: &User) -> anyhow::Result<()> {
|
||||
db_connection::execute(|conn| {
|
||||
Ok(
|
||||
diesel::update(users::dsl::users.filter(users::dsl::id.eq(user.id)))
|
||||
.set((
|
||||
users::dsl::name.eq(user.name.clone()),
|
||||
users::dsl::admin.eq(user.admin),
|
||||
users::dsl::active.eq(user.active),
|
||||
users::dsl::email.eq(user.email.clone()),
|
||||
users::dsl::time_gen_reset_token.eq(user.time_gen_reset_token),
|
||||
users::dsl::reset_password_token.eq(user.reset_password_token.clone()),
|
||||
users::dsl::time_gen_delete_account_token
|
||||
.eq(user.time_gen_delete_account_token),
|
||||
users::dsl::delete_account_token.eq(user.delete_account_token.clone()),
|
||||
users::dsl::time_activate.eq(user.time_activate as i64),
|
||||
users::dsl::password.eq(user.password.clone()),
|
||||
))
|
||||
.execute(conn)?,
|
||||
)
|
||||
diesel::update(users::dsl::users.filter(users::dsl::id.eq(user.id)))
|
||||
.set((
|
||||
users::dsl::name.eq(user.name.clone()),
|
||||
users::dsl::admin.eq(user.admin),
|
||||
users::dsl::active.eq(user.active),
|
||||
users::dsl::email.eq(user.email.clone()),
|
||||
users::dsl::time_gen_reset_token.eq(user.time_gen_reset_token),
|
||||
users::dsl::reset_password_token.eq(user.reset_password_token.clone()),
|
||||
users::dsl::time_gen_delete_account_token.eq(user.time_gen_delete_account_token),
|
||||
users::dsl::delete_account_token.eq(user.delete_account_token.clone()),
|
||||
users::dsl::time_activate.eq(user.time_activate),
|
||||
users::dsl::password.eq(user.password.clone()),
|
||||
))
|
||||
.execute(conn)
|
||||
})?;
|
||||
|
||||
Ok(())
|
||||
|
Reference in New Issue
Block a user