better migrator

This commit is contained in:
ston1th 2019-08-24 18:32:27 +02:00
commit 1cf4c7cd77

View file

@ -12,17 +12,17 @@ const (
currentVersion = 2 currentVersion = 2
) )
type migrator func(*DB) error type migrator func(*DB, int) error
var migrators = []migrator{ var migrators = []migrator{
func(db *DB) error { func(_ *DB, _ int) error {
return nil return nil
}, },
func(db *DB) error { func(_ *DB, _ int) error {
// dummy placeholder // dummy placeholder
return nil return nil
}, },
func(db *DB) (err error) { func(db *DB, _ int) (err error) {
err = db.addDefaultTags() err = db.addDefaultTags()
if err != nil { if err != nil {
return return
@ -48,7 +48,7 @@ func (db *DB) runMigrations() (err error) {
for version < currentVersion { for version < currentVersion {
v := version + 1 v := version + 1
log.Printf("db: running migration version %d", v) log.Printf("db: running migration version %d", v)
err = migrators[v](db) err = migrators[v](db, v)
if err != nil { if err != nil {
return return
} }