Build script to initialize dev db in sqlx project

build.rs

use std::str::FromStr;

use sqlx::{
    sqlite::{SqliteConnectOptions, SqliteJournalMode},
    SqlitePool,
};
use tokio::runtime::Builder;

fn main() {
    let database_url = std::env::var("DATABASE_URL").unwrap_or("sqlite://dev.db".to_string());
    let options = SqliteConnectOptions::from_str(&database_url)
        .expect("Error parsing database URL")
        .create_if_missing(true)
        .journal_mode(SqliteJournalMode::Wal);
    let rt = Builder::new_current_thread()
        .worker_threads(1)
        .enable_time()
        .build()
        .unwrap();
    rt.block_on(async move {
        connect(options).await;
    });
}

async fn connect(options: SqliteConnectOptions) {
    let pool = SqlitePool::connect_with(options)
        .await
        .expect("Error connecting to database");
    migrate(&pool).await;
}

async fn migrate(pool: &SqlitePool) {
    sqlx::migrate!()
        .run(pool)
        .await
        .expect("Error when applying database migrations");
}

This was added in the Cargo.toml:

[build-dependencies]
sqlx = { version = "0.7", features = [
  "sqlite",
] }
tokio = { version = "1", features = [
  "rt",
] }