fix: Set explicit transaction mode on Go register

pressly · Nov 9, 2023 · 3fade11 · 3fade11
1 parent 04e12b8
commit 3fade11
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/register.go b/register.go
@@ -22,8 +22,8 @@ func AddNamedMigrationContext(filename string, up, down GoMigrationContext) {
 	if err := register(
 		filename,
 		true,
-		&GoFunc{RunTx: up},
-		&GoFunc{RunTx: down},
+		&GoFunc{RunTx: up, Mode: TransactionEnabled},
+		&GoFunc{RunTx: down, Mode: TransactionEnabled},
 	); err != nil {
 		panic(err)
 	}
@@ -44,8 +44,8 @@ func AddNamedMigrationNoTxContext(filename string, up, down GoMigrationNoTxConte
 	if err := register(
 		filename,
 		false,
-		&GoFunc{RunDB: up},
-		&GoFunc{RunDB: down},
+		&GoFunc{RunDB: up, Mode: TransactionDisabled},
+		&GoFunc{RunDB: down, Mode: TransactionDisabled},
 	); err != nil {
 		panic(err)
 	}