58 lines
1.6 KiB
Go
58 lines
1.6 KiB
Go
|
package migrate
|
||
|
|
|
||
|
|
import (
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/jinzhu/gorm"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Migration represents a database schema migration.
|
||
|
|
type Migration struct {
|
||
|
|
ID string `gorm:"size:16;primary_key;auto_increment:false;" json:"ID" yaml:"ID"`
|
||
|
|
Dialect string `gorm:"size:16;" json:"Dialect" yaml:"Dialect,omitempty"`
|
||
|
|
Error string `gorm:"size:255;" json:"Error" yaml:"Error,omitempty"`
|
||
|
|
Source string `gorm:"size:16;" json:"Source" yaml:"Source,omitempty"`
|
||
|
|
Query string `gorm:"-" json:"Query" yaml:"Query,omitempty"`
|
||
|
|
StartedAt time.Time `json:"StartedAt" yaml:"StartedAt,omitempty"`
|
||
|
|
FinishedAt *time.Time `json:"FinishedAt" yaml:"FinishedAt,omitempty"`
|
||
|
|
}
|
||
|
|
|
||
|
|
// TableName returns the entity database table name.
|
||
|
|
func (Migration) TableName() string {
|
||
|
|
return "migrations"
|
||
|
|
}
|
||
|
|
|
||
|
|
// Fail marks the migration as failed by adding an error message.
|
||
|
|
func (m *Migration) Fail(err error, db *gorm.DB) {
|
||
|
|
if err == nil {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
m.Error = err.Error()
|
||
|
|
db.Model(m).Updates(Values{"Error": m.Error})
|
||
|
|
}
|
||
|
|
|
||
|
|
// Finish updates the FinishedAt timestamp when the migration was successful.
|
||
|
|
func (m *Migration) Finish(db *gorm.DB) {
|
||
|
|
db.Model(m).Updates(Values{"FinishedAt": time.Now().UTC()})
|
||
|
|
}
|
||
|
|
|
||
|
|
// Execute runs the migration.
|
||
|
|
func (m *Migration) Execute(db *gorm.DB) {
|
||
|
|
start := time.Now()
|
||
|
|
|
||
|
|
m.StartedAt = start.UTC().Round(time.Second)
|
||
|
|
|
||
|
|
if err := db.Create(m).Error; err != nil {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
if err := db.Exec(m.Query).Error; err != nil {
|
||
|
|
m.Fail(err, db)
|
||
|
|
log.Errorf("migration %s failed: %s [%s]", m.ID, err, time.Since(start))
|
||
|
|
} else {
|
||
|
|
m.Finish(db)
|
||
|
|
log.Infof("migration %s successful [%s]", m.ID, time.Since(start))
|
||
|
|
}
|
||
|
|
}
|