Dionisio
49 lines
1.5 KiB
Rust
49 lines
1.5 KiB
Rust
use sqlx::postgres::PgPoolOptions;
|
|
use std::env;
|
|
use std::path::Path;
|
|
use std::time::Duration;
|
|
|
|
#[tokio::main]
|
|
async fn main() -> Result<(), Box<dyn std::error::Error>> {
|
|
// Configure logging
|
|
tracing_subscriber::fmt::init();
|
|
|
|
// Load environment variables (.env.local first, then .env)
|
|
if let Ok(path) = env::var("DOTENV_PATH") {
|
|
dotenv::from_path(Path::new(&path)).ok();
|
|
} else {
|
|
dotenv::from_filename(".env.local").ok();
|
|
dotenv::dotenv().ok();
|
|
}
|
|
|
|
// Get DATABASE_URL from environment variables
|
|
let database_url = env::var("DATABASE_URL").expect("DATABASE_URL must be configured");
|
|
|
|
println!("Connecting to the database...");
|
|
|
|
// Create connection pool
|
|
let pool = PgPoolOptions::new()
|
|
.max_connections(5)
|
|
.acquire_timeout(Duration::from_secs(10))
|
|
.connect(&database_url)
|
|
.await?;
|
|
|
|
// Run migrations
|
|
println!("Running migrations...");
|
|
|
|
// Get the directory from an environment variable or use a default value
|
|
let migrations_dir = env::var("MIGRATIONS_DIR").unwrap_or_else(|_| "./migrations".to_string());
|
|
println!("Migrations directory: {}", migrations_dir);
|
|
|
|
// Create a migrator
|
|
let migrator = sqlx::migrate::Migrator::new(Path::new(&migrations_dir))
|
|
.await
|
|
.expect("Could not create the migrator");
|
|
|
|
// Run all pending migrations
|
|
migrator.run(&pool).await?;
|
|
|
|
println!("Migrations applied successfully");
|
|
|
|
Ok(())
|
|
} |