Allow multi-statement migration scripts
Some checks failed
Build / build (push) Has been cancelled

This commit is contained in:
Sebastiaan de Schaetzen 2025-03-12 08:23:16 +01:00
parent 9d5c0bcbb1
commit 82c7f57078
2 changed files with 20 additions and 9 deletions

View File

@ -50,7 +50,7 @@ func (d *Db) MigrateDb(filesystem ReadDirFileFS, directory string) error {
return fmt.Errorf("error opening migration script %s: %v", migrationScript, err)
}
err = performSingleMigration(err, d, migrationScript, targetVersion)
err = performSingleMigration(d, migrationScript, targetVersion)
if err != nil {
return err
}
@ -61,14 +61,20 @@ func (d *Db) MigrateDb(filesystem ReadDirFileFS, directory string) error {
return nil
}
func performSingleMigration(err error, d *Db, migrationScript []byte, targetVersion int) error {
func performSingleMigration(d *Db, migrationScript []byte, targetVersion int) error {
script := string(migrationScript)
// Split script based on semicolon
statements := strings.Split(script, ";")
tx, err := d.Begin()
if err != nil {
return fmt.Errorf("error beginning transaction: %v", err)
}
defer tx.MustRollback()
err = tx.Query(string(migrationScript)).Exec()
for _, statement := range statements {
statement = strings.TrimSpace(statement)
err = tx.Query(statement).Exec()
if err != nil {
return fmt.Errorf("error performing migration: %v", err)
}
@ -78,6 +84,8 @@ func performSingleMigration(err error, d *Db, migrationScript []byte, targetVers
return fmt.Errorf("error updating version: %v", err)
}
}
err = tx.Commit()
if err != nil {
return fmt.Errorf("error commiting transaction: %v", err)

View File

@ -0,0 +1,3 @@
create table testTable(value text);
insert into testTable(value) values ('testValue');