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