2024-07-12 00:16:17 +00:00
|
|
|
package main
|
|
|
|
|
2024-07-12 00:28:57 +00:00
|
|
|
import (
|
|
|
|
"log"
|
|
|
|
"os"
|
|
|
|
)
|
|
|
|
|
2024-07-12 00:16:17 +00:00
|
|
|
type Migration struct {
|
|
|
|
Path string
|
|
|
|
Name string
|
|
|
|
Requires string
|
|
|
|
Run bool
|
|
|
|
}
|
|
|
|
|
|
|
|
func NewMigration(path string, name string, requires string) Migration {
|
|
|
|
return Migration{
|
|
|
|
Path: path,
|
|
|
|
Name: name,
|
|
|
|
Requires: requires,
|
|
|
|
Run: false,
|
|
|
|
}
|
|
|
|
}
|
2024-07-12 00:28:57 +00:00
|
|
|
|
|
|
|
// Applies a migration to the given database connection.
|
|
|
|
//
|
|
|
|
// If an error is returned while trying to read the migration file
|
|
|
|
// or execute the SQL it contains, the error is returned.
|
|
|
|
func (m *Migration) ApplyMigration(db DB) error {
|
|
|
|
migrationPath := m.Path
|
|
|
|
migrationBytes, err := os.ReadFile(migrationPath)
|
|
|
|
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
migrationSql := string(migrationBytes)
|
|
|
|
|
|
|
|
log.Printf("SQL: %s", migrationSql)
|
|
|
|
|
|
|
|
err = db.Execute(migrationSql)
|
|
|
|
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
return nil
|
|
|
|
|
|
|
|
}
|