Add env var option to prevent wrapping migrations in a transaction

sea-orm-migration/src/migrator.rs

@@ -1,5 +1,6 @@
 use futures::Future;
 use std::collections::HashSet;
+use std::env;
 use std::fmt::Display;
 use std::pin::Pin;
 use std::time::SystemTime;
@@ -258,10 +259,16 @@ where
 
  match db.get_database_backend() {
  DbBackend::Postgres => {
- let transaction = db.begin().await?;
- let manager = SchemaManager::new(&transaction);
- f(&manager).await?;
- transaction.commit().await
+ let non_atomic = env::var("SEA_ORM_NON_ATOMIC_PG_MIGRATIONS").unwrap_or(String::from("false"));
+ match non_atomic.as_str() {
+ "true" => f(&SchemaManager::new(db)).await,
+ _ =>{
+ let transaction = db.begin().await?;
+ let manager = SchemaManager::new(&transaction);
+ f(&manager).await?;
+ transaction.commit().await
+ }
+ }
  }
  DbBackend::MySql | DbBackend::Sqlite => {
  let manager = SchemaManager::new(db);

sea-orm-migration/tests/common/migration/m20230109_000002_create_index_concurrently.rs

@@ -0,0 +1,29 @@
+use sea_orm_migration::prelude::*;
+
+#[derive(DeriveMigrationName)]
+pub struct Migration;
+
+#[async_trait::async_trait]
+impl MigrationTrait for Migration {
+ async fn up(&self, manager: &SchemaManager) -> Result<(), DbErr> {
+ let db = manager.get_connection();
+
+ db.execute_unprepared(
+ "CREATE INDEX CONCURRENTLY fruit_name_index ON fruit(name)"
+ )
+ .await?;
+
+ Ok(())
+ }
+
+ async fn down(&self, manager: &SchemaManager) -> Result<(), DbErr> {
+ let db = manager.get_connection();
+
+ db.execute_unprepared(
+ "DROP INDEX fruit_name_index"
+ )
+ .await?;
+
+ Ok(())
+ }
+}

sea-orm-migration/tests/common/migration/mod.rs

@@ -4,3 +4,4 @@ pub mod m20220118_000003_seed_cake_table;
 pub mod m20220118_000004_create_tea_enum;
 pub mod m20220923_000001_seed_cake_table;
 pub mod m20230109_000001_seed_cake_table;
+pub mod m20230109_000002_create_index_concurrently;

sea-orm-migration/tests/common/migrator/default.rs

@@ -13,6 +13,7 @@ impl MigratorTrait for Migrator {
  Box::new(m20220118_000004_create_tea_enum::Migration),
  Box::new(m20220923_000001_seed_cake_table::Migration),
  Box::new(m20230109_000001_seed_cake_table::Migration),
+ Box::new(m20230109_000002_create_index_concurrently::Migration),
  ]
  }
 }

sea-orm-migration/tests/common/migrator/override_migration_table_name.rs

@@ -13,6 +13,7 @@ impl MigratorTrait for Migrator {
  Box::new(m20220118_000004_create_tea_enum::Migration),
  Box::new(m20220923_000001_seed_cake_table::Migration),
  Box::new(m20230109_000001_seed_cake_table::Migration),
+ Box::new(m20230109_000002_create_index_concurrently::Migration),
  ]
  }
 

sea-orm-migration/tests/main.rs

@@ -127,7 +127,7 @@ where
 
  println!("\nMigrator::get_pending_migrations");
  let migrations = Migrator::get_pending_migrations(db).await?;
- assert_eq!(migrations.len(), 5);
+ assert_eq!(migrations.len(), 6);
 
  let migration = migrations.get(0).unwrap();
  assert_eq!(migration.name(), "m20220118_000002_create_fruit_table");
@@ -158,7 +158,7 @@ where
  // Should throw an error
  println!("\nMigrator::up");
  assert_eq!(
- Migrator::up(db, None).await,
+ Migrator::up(db, Some(6)).await,
  Err(DbErr::Migration(
  "Abort migration and rollback changes".into()
  ))
@@ -176,11 +176,13 @@ where
  }
 
  println!("\nMigrator::up");
+ std::env::set_var("SEA_ORM_NON_ATOMIC_PG_MIGRATIONS", "true");
  Migrator::up(db, None).await?;
+ std::env::remove_var("SEA_ORM_NON_ATOMIC_PG_MIGRATIONS");
 
  println!("\nMigrator::get_applied_migrations");
  let migrations = Migrator::get_applied_migrations(db).await?;
- assert_eq!(migrations.len(), 6);
+ assert_eq!(migrations.len(), 7);
 
  assert!(!manager.has_index("cake", "non_existent_index").await?);
  assert!(manager.has_index("cake", "cake_name_index").await?);
@@ -237,13 +239,15 @@ where
  assert!(!manager.has_table("fruit").await?);
 
  println!("\nMigrator::fresh");
+ std::env::set_var("SEA_ORM_NON_ATOMIC_PG_MIGRATIONS", "true");
  Migrator::fresh(db).await?;
 
  assert!(manager.has_table("cake").await?);
  assert!(manager.has_table("fruit").await?);
 
  println!("\nMigrator::refresh");
  Migrator::refresh(db).await?;
+ std::env::remove_var("SEA_ORM_NON_ATOMIC_PG_MIGRATIONS");
 
  assert!(manager.has_table("cake").await?);
  assert!(manager.has_table("fruit").await?);