caisin/sea-orm
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #755 from SeaQL/apply-old-pending-migration

Browse Source 
Allow old pending migration to be applied
This commit is contained in:
Chris Tsang 2022-07-01 02:40:37 +08:00 committed by GitHub
commit 50f9f5ad2f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 28 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
sea-orm-migration/src/migrator.rs
View File

@ -1,3 +1,4 @@
use std::collections::HashSet;
use std::fmt::Display; use std::fmt::Display;
use std::time::SystemTime; use std::time::SystemTime;
use tracing::info; use tracing::info;
@ -66,19 +67,36 @@ pub trait MigratorTrait: Send {
Self::install(db).await?; Self::install(db).await?;
let mut migration_files = Self::get_migration_files(); let mut migration_files = Self::get_migration_files();
let migration_models = Self::get_migration_models(db).await?; let migration_models = Self::get_migration_models(db).await?;
for (i, migration_model) in migration_models.into_iter().enumerate() {
if let Some(migration_file) = migration_files.get_mut(i) { let migration_in_db: HashSet<String> = migration_models
if migration_file.migration.name() == migration_model.version.as_str() { .into_iter()
.map(|model| model.version)
.collect();
let migration_in_fs: HashSet<String> = migration_files
.iter()
.map(|file| file.migration.name().to_string())
.collect();
let pending_migrations = &migration_in_fs - &migration_in_db;
for migration_file in migration_files.iter_mut() {
if !pending_migrations.contains(migration_file.migration.name()) {
migration_file.status = MigrationStatus::Applied; migration_file.status = MigrationStatus::Applied;
}
}
let missing_migrations_in_fs = &migration_in_db - &migration_in_fs;
let errors: Vec<String> = missing_migrations_in_fs
.iter()
.map(|missing_migration| {
format!("Migration file of version '{}' is missing, this migration has been applied but its file is missing", missing_migration)
}).collect();
if !errors.is_empty() {
Err(DbErr::Custom(errors.join("\n")))
} else { } else {
return Err(DbErr::Custom(format!("Migration mismatch: applied migration != migration file, '{0}' != '{1}'\nMigration '{0}' has been applied but its corresponding migration file is missing.", migration_file.migration.name(), migration_model.version)));
}
} else {
return Err(DbErr::Custom(format!("Migration file of version '{}' is missing, this migration has been applied but its file is missing", migration_model.version)));
}
}
Ok(migration_files) Ok(migration_files)
} }
}
/// Get list of pending migrations /// Get list of pending migrations
async fn get_pending_migrations(db: &DbConn) -> Result<Vec<Migration>, DbErr> { async fn get_pending_migrations(db: &DbConn) -> Result<Vec<Migration>, DbErr> {