mirror of https://github.com/muety/wakapi.git
76 lines
1.6 KiB
Go
76 lines
1.6 KiB
Go
package migrations
|
|
|
|
import (
|
|
"github.com/emvi/logbuch"
|
|
"github.com/muety/wakapi/config"
|
|
"gorm.io/gorm"
|
|
"sort"
|
|
"strings"
|
|
)
|
|
|
|
type migrationFunc struct {
|
|
f func(db *gorm.DB, cfg *config.Config) error
|
|
name string
|
|
}
|
|
|
|
type migrationFuncs []migrationFunc
|
|
|
|
var (
|
|
preMigrations migrationFuncs
|
|
postMigrations migrationFuncs
|
|
)
|
|
|
|
func registerPreMigration(f migrationFunc) {
|
|
preMigrations = append(preMigrations, f)
|
|
}
|
|
|
|
func registerPostMigration(f migrationFunc) {
|
|
postMigrations = append(postMigrations, f)
|
|
}
|
|
|
|
func Run(db *gorm.DB, cfg *config.Config) {
|
|
RunPreMigrations(db, cfg)
|
|
RunSchemaMigrations(db, cfg)
|
|
RunPostMigrations(db, cfg)
|
|
}
|
|
|
|
func RunSchemaMigrations(db *gorm.DB, cfg *config.Config) {
|
|
if err := cfg.GetMigrationFunc(cfg.Db.Dialect)(db); err != nil {
|
|
logbuch.Fatal(err.Error())
|
|
}
|
|
}
|
|
|
|
func RunPreMigrations(db *gorm.DB, cfg *config.Config) {
|
|
sort.Sort(preMigrations)
|
|
|
|
for _, m := range preMigrations {
|
|
logbuch.Info("potentially running migration '%s'", m.name)
|
|
if err := m.f(db, cfg); err != nil {
|
|
logbuch.Fatal("migration '%s' failed - %v", m.name, err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func RunPostMigrations(db *gorm.DB, cfg *config.Config) {
|
|
sort.Sort(postMigrations)
|
|
|
|
for _, m := range postMigrations {
|
|
logbuch.Info("potentially running migration '%s'", m.name)
|
|
if err := m.f(db, cfg); err != nil {
|
|
logbuch.Fatal("migration '%s' failed - %v", m.name, err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func (m migrationFuncs) Len() int {
|
|
return len(m)
|
|
}
|
|
|
|
func (m migrationFuncs) Less(i, j int) bool {
|
|
return strings.Compare(m[i].name, m[j].name) < 0
|
|
}
|
|
|
|
func (m migrationFuncs) Swap(i, j int) {
|
|
m[i], m[j] = m[j], m[i]
|
|
}
|