use std::str::FromStr;
use std::time::Duration;
use sqlx::sqlite::{SqliteConnectOptions, SqliteJournalMode, SqlitePoolOptions, SqliteSynchronous};
use sqlx::SqlitePool;
use crate::config::Config;
pub async fn init_pool(cfg: &Config) -> anyhow::Result<SqlitePool> {
if !cfg.database_url.starts_with("sqlite") {
anyhow::bail!(
"Stage 0 supports sqlite only; got `{}`. Postgres is planned via SQLx.",
cfg.database_url
);
}
let opts = SqliteConnectOptions::from_str(&cfg.database_url)?
.create_if_missing(true)
.journal_mode(SqliteJournalMode::Wal)
.synchronous(SqliteSynchronous::Normal)
.busy_timeout(Duration::from_secs(15))
.foreign_keys(true);
let pool = SqlitePoolOptions::new()
.max_connections(cfg.db_max_connections)
.acquire_timeout(Duration::from_secs(20))
.connect_with(opts)
.await?;
Ok(pool)
}
pub async fn run_migrations(pool: &SqlitePool) -> anyhow::Result<()> {
sqlx::migrate!("./migrations").run(pool).await?;
Ok(())
}
pub async fn clear_segment_read_locks(pool: &SqlitePool) -> anyhow::Result<()> {
let n = sqlx::query("UPDATE segments SET locked = 0 WHERE locked <> 0")
.execute(pool)
.await?
.rows_affected();
if n > 0 {
tracing::info!(cleared = n, "startup: cleared stale segment read-locks");
}
Ok(())
}