789981ee92
Tables now contain update_timestamp and update_process and update_timestamp will be automatically set via a trigger. Also the I removed the IDs from the tables as they are not required for the auth system. The names are the identifier.
51 lines
1.6 KiB
Rust
51 lines
1.6 KiB
Rust
use std::sync::Arc;
|
|
|
|
use dao::DaoError;
|
|
use sqlx::{query, SqlitePool};
|
|
|
|
pub struct HelloDaoImpl {
|
|
pool: Arc<SqlitePool>,
|
|
}
|
|
|
|
impl HelloDaoImpl {
|
|
pub fn new(pool: Arc<SqlitePool>) -> Self {
|
|
Self { pool }
|
|
}
|
|
}
|
|
|
|
impl dao::HelloDao for HelloDaoImpl {
|
|
async fn get_hello(&self) -> Result<Arc<str>, dao::DaoError> {
|
|
let result = query!(r"SELECT 'Hello, world!' as message")
|
|
.fetch_all(self.pool.as_ref())
|
|
.await
|
|
.map_err(|err| DaoError::DatabaseQueryError(Box::new(err)))?;
|
|
let message: Arc<str> = result[0].message.clone().into();
|
|
Ok(message)
|
|
}
|
|
}
|
|
|
|
pub struct PermissionDaoImpl {
|
|
pool: Arc<SqlitePool>,
|
|
}
|
|
impl PermissionDaoImpl {
|
|
pub fn new(pool: Arc<SqlitePool>) -> Self {
|
|
Self { pool }
|
|
}
|
|
}
|
|
impl dao::PermissionDao for PermissionDaoImpl {
|
|
async fn has_privilege(&self, user: &str, privilege: &str) -> Result<bool, dao::DaoError> {
|
|
let result = query!(
|
|
r"SELECT count(*) as results FROM user
|
|
INNER JOIN user_role ON user.name = user_role.user_name
|
|
INNER JOIN role ON user_role.role_name = role.name
|
|
INNER JOIN role_privilege ON role.name = role_privilege.role_name
|
|
WHERE role_privilege.privilege_name = ? AND user.name = ?",
|
|
privilege,
|
|
user,
|
|
)
|
|
.fetch_all(self.pool.as_ref())
|
|
.await
|
|
.map_err(|err| DaoError::DatabaseQueryError(Box::new(err)))?;
|
|
Ok(result[0].results > 0)
|
|
}
|
|
}
|