42 lines
1.3 KiB
Rust
42 lines
1.3 KiB
Rust
//! Globally accessible application state.
|
|
|
|
use axum::extract::FromRef;
|
|
use sqlx::{
|
|
sqlite::{SqliteConnectOptions, SqliteJournalMode, SqlitePoolOptions, SqliteSynchronous},
|
|
SqlitePool,
|
|
};
|
|
|
|
// TODO Occasionally run PRAGMA optimize
|
|
// TODO Open db from path
|
|
async fn pool() -> sqlx::Result<SqlitePool> {
|
|
let options = SqliteConnectOptions::new()
|
|
// https://www.sqlite.org/pragma.html#pragma_journal_mode
|
|
.journal_mode(SqliteJournalMode::Wal)
|
|
// https://www.sqlite.org/pragma.html#pragma_synchronous
|
|
// NORMAL recommended when using WAL, can't cause corruption
|
|
.synchronous(SqliteSynchronous::Normal)
|
|
// https://www.sqlite.org/pragma.html#pragma_foreign_keys
|
|
.foreign_keys(true)
|
|
// https://www.sqlite.org/pragma.html#pragma_trusted_schema
|
|
// The docs recommend always turning this off
|
|
.pragma("trusted_schema", "false")
|
|
.create_if_missing(true)
|
|
.optimize_on_close(true, None);
|
|
|
|
let pool = SqlitePoolOptions::new().connect_with(options).await?;
|
|
|
|
sqlx::migrate!().run(&pool).await?;
|
|
|
|
Ok(pool)
|
|
}
|
|
|
|
#[derive(Clone, FromRef)]
|
|
pub struct AppState {
|
|
pub db: SqlitePool,
|
|
}
|
|
|
|
impl AppState {
|
|
pub async fn new() -> anyhow::Result<Self> {
|
|
Ok(Self { db: pool().await? })
|
|
}
|
|
}
|