You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
26 lines
679 B
Rust
26 lines
679 B
Rust
4 years ago
|
use crate::database::Model;
|
||
|
|
use postgres::{Client, Error};
|
||
|
|
use std::sync::{Arc, Mutex};
|
||
|
|
|
||
|
|
#[derive(Clone)]
|
||
|
|
pub struct UserRoles {
|
||
|
|
connection: Arc<Mutex<Client>>,
|
||
|
|
}
|
||
|
|
|
||
|
|
impl Model for UserRoles {
|
||
|
|
fn new(connection: Arc<Mutex<Client>>) -> Self {
|
||
|
|
Self { connection }
|
||
|
|
}
|
||
|
|
|
||
|
|
fn init(&self) -> Result<(), Error> {
|
||
|
|
self.connection.lock().unwrap().batch_execute(
|
||
|
|
"
|
||
|
|
CREATE TABLE IF NOT EXISTS user_roles (
|
||
|
|
user_id INT NOT NULL REFERENCES users(id) ON DELETE CASCADE,
|
||
|
|
role_id INT NOT NULL REFERENCES roles(id) ON DELETE CASCADE,
|
||
|
|
PRIMARY KEY (user_id, role_id)
|
||
|
|
);",
|
||
|
|
)
|
||
|
|
}
|
||
|
|
}
|