docstore/pkg/db/migration.go
2019-08-20 23:57:43 +02:00

57 lines
1,021 B
Go

// Copyright (C) 2019 Marius Schellenberger
package db
import (
"git.giftfish.de/ston1th/docstore/pkg/log"
"git.giftfish.de/ston1th/docstore/pkg/store"
)
const (
versionKey = "version/version"
currentVersion = 2
)
type migrator func(*DB) error
var migrators = []migrator{
func(db *DB) error {
return nil
},
func(db *DB) error {
// dummy placeholder
return nil
},
func(db *DB) error {
return db.addDefaultTags()
},
}
func (db *DB) runMigrations() (err error) {
var version int
err = db.store.Get(versionKey, &version)
if err == store.ErrKeyNotFound {
err = db.store.Set(versionKey, currentVersion)
if err != nil {
return
}
err = db.store.Get(versionKey, &version)
if err != nil {
return
}
}
log.Printf("db: database version %d", version)
for version < currentVersion {
v := version + 1
log.Printf("db: running migration %d", v)
err = migrators[v](db)
if err != nil {
return
}
err = db.store.Set(versionKey, v)
if err != nil {
return
}
}
return
}