24 lines
712 B
Rust
24 lines
712 B
Rust
mod schema;
|
|
pub mod users;
|
|
|
|
use diesel::sqlite::SqliteConnection;
|
|
use diesel::connection::SimpleConnection;
|
|
use diesel::QueryResult;
|
|
use diesel::r2d2::{ConnectionManager, Pool};
|
|
|
|
fn initialize_db(conn: &mut SqliteConnection) -> QueryResult<()> {
|
|
// Enable WAL mode and enforce foreign keys
|
|
conn.batch_execute("PRAGMA journal_mode = WAL; PRAGMA synchronous = NORMAL; PRAGMA foreign_keys = ON;")
|
|
}
|
|
|
|
pub fn initialize_pool(url: &str) -> Pool<ConnectionManager<SqliteConnection>> {
|
|
let manager = ConnectionManager::new(url);
|
|
|
|
let pool = Pool::builder().test_on_check_out(true).build(manager).expect("oops");
|
|
|
|
let mut conn = pool.get().unwrap();
|
|
initialize_db(&mut conn).unwrap();
|
|
|
|
pool
|
|
}
|