You've already forked AstralRinth
c4fb7b7928
* Merged mod file upload in version creation, mod creation and
version file add to one function; This makes sure that they are
consistent
* Made some fields on `User` optional: `github_id`, `avatar_url`, `bio`.
* We may not want to publicly show the `github_id` to everyone
with access to the API
* If we allow non-github users, some of those fields would be
invalid; some oauth providers may not have avatars or bios
* Made CORS origins should configurable
* Made `--reconfigure-indices` and `--reset-indices` exit after
completion instead of starting the server
54 lines
1.5 KiB
Rust
54 lines
1.5 KiB
Rust
use log::info;
|
|
use sqlx::migrate::{Migrate, MigrateDatabase, Migrator};
|
|
use sqlx::postgres::{PgPool, PgPoolOptions};
|
|
use sqlx::{Connection, PgConnection, Postgres};
|
|
use std::path::Path;
|
|
|
|
const MIGRATION_FOLDER: &str = "migrations";
|
|
|
|
pub async fn connect() -> Result<PgPool, sqlx::Error> {
|
|
info!("Initializing database connection");
|
|
|
|
let database_url = dotenv::var("DATABASE_URL").expect("`DATABASE_URL` not in .env");
|
|
let pool = PgPoolOptions::new()
|
|
.max_connections(20)
|
|
.connect(&database_url)
|
|
.await?;
|
|
|
|
Ok(pool)
|
|
}
|
|
pub async fn check_for_migrations() -> Result<(), sqlx::Error> {
|
|
let uri = &*dotenv::var("DATABASE_URL").expect("`DATABASE_URL` not in .env");
|
|
if !Postgres::database_exists(uri).await? {
|
|
info!("Creating database...");
|
|
Postgres::create_database(uri).await?;
|
|
}
|
|
info!("Applying migrations...");
|
|
run_migrations(uri).await?;
|
|
|
|
Ok(())
|
|
}
|
|
|
|
pub async fn run_migrations(uri: &str) -> Result<(), sqlx::Error> {
|
|
let migrator = Migrator::new(Path::new(MIGRATION_FOLDER)).await?;
|
|
let mut conn: PgConnection = PgConnection::connect(uri).await?;
|
|
|
|
conn.ensure_migrations_table().await?;
|
|
|
|
let (version, dirty) = conn.version().await?.unwrap_or((0, false));
|
|
|
|
if dirty {
|
|
panic!("The database is dirty ! Please check your database status.");
|
|
}
|
|
|
|
for migration in migrator.iter() {
|
|
if migration.version > version {
|
|
let _elapsed = conn.apply(migration).await?;
|
|
} else {
|
|
conn.validate(migration).await?;
|
|
}
|
|
}
|
|
|
|
Ok(())
|
|
}
|