127 lines
3.7 KiB
Go
127 lines
3.7 KiB
Go
package migrate
|
|
|
|
import (
|
|
"strings"
|
|
"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"`
|
|
Statements []string `gorm:"-" json:"Statements" yaml:"Statements,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"
|
|
}
|
|
|
|
// Finished tests if the migration has been finished yet.
|
|
func (m *Migration) Finished() bool {
|
|
if m.FinishedAt == nil {
|
|
return false
|
|
}
|
|
|
|
return !m.FinishedAt.IsZero()
|
|
}
|
|
|
|
// RunDuration returns the run duration of started migrations.
|
|
func (m *Migration) RunDuration() time.Duration {
|
|
if m.Error != "" || m.StartedAt.IsZero() {
|
|
return time.Duration(0)
|
|
}
|
|
|
|
if m.Finished() {
|
|
return m.FinishedAt.UTC().Sub(m.StartedAt.UTC())
|
|
}
|
|
|
|
return time.Now().UTC().Sub(m.StartedAt.UTC())
|
|
}
|
|
|
|
// Repeat tests if the migration should be repeated.
|
|
func (m *Migration) Repeat(runFailed bool) bool {
|
|
if runFailed && m.Error != "" {
|
|
return true
|
|
}
|
|
|
|
// Don't repeat if finished.
|
|
if m.Finished() {
|
|
return false
|
|
}
|
|
|
|
// Repeat not started yet.
|
|
if m.StartedAt.IsZero() {
|
|
return true
|
|
}
|
|
|
|
// Repeat if "running" for more than 60 minutes.
|
|
return m.RunDuration().Minutes() >= 60
|
|
}
|
|
|
|
// Fail marks the migration as failed by adding an error message and removing the FinishedAt timestamp.
|
|
func (m *Migration) Fail(err error, db *gorm.DB) {
|
|
if err == nil {
|
|
return
|
|
}
|
|
|
|
m.FinishedAt = nil
|
|
|
|
if err.Error() == "" {
|
|
m.Error = "unknown error"
|
|
} else {
|
|
m.Error = err.Error()
|
|
}
|
|
|
|
db.Model(m).Updates(Values{"FinishedAt": m.FinishedAt, "Error": m.Error})
|
|
}
|
|
|
|
// Finish updates the FinishedAt timestamp and removes the error message when the migration was successful.
|
|
func (m *Migration) Finish(db *gorm.DB) error {
|
|
finished := time.Now().UTC().Truncate(time.Second)
|
|
m.FinishedAt = &finished
|
|
m.Error = ""
|
|
return db.Model(m).Updates(Values{"FinishedAt": m.FinishedAt, "Error": m.Error}).Error
|
|
}
|
|
|
|
// Execute runs the migration.
|
|
func (m *Migration) Execute(db *gorm.DB) error {
|
|
for _, s := range m.Statements { // ADD
|
|
if err := db.Exec(s).Error; err != nil {
|
|
// Normalize query and error for comparison.
|
|
q := strings.ToLower(s)
|
|
e := strings.ToLower(err.Error())
|
|
|
|
// Log the errors triggered by ALTER and DROP statements
|
|
// and otherwise ignore them, since some databases do not
|
|
// support "IF EXISTS".
|
|
if strings.HasPrefix(q, "alter table ") &&
|
|
strings.Contains(s, " add ") &&
|
|
strings.Contains(e, "duplicate") {
|
|
log.Tracef("migrate: %s (ignored, column already exists)", err)
|
|
} else if strings.HasPrefix(q, "drop index ") &&
|
|
strings.Contains(e, "drop") {
|
|
log.Tracef("migrate: %s (ignored, probably no longer existed)", err)
|
|
} else if strings.HasPrefix(q, "drop table ") &&
|
|
strings.Contains(e, "drop") {
|
|
log.Tracef("migrate: %s (ignored, probably no longer existed)", err)
|
|
} else if (strings.Contains(q, " ignore ") || strings.Contains(q, "replace")) &&
|
|
(strings.Contains(e, "no such table") || strings.Contains(e, "doesn't exist")) {
|
|
log.Tracef("migrate: %s (ignored, old table no longer exists)", err)
|
|
} else if strings.Contains(q, "rename column") && strings.Contains(e, "no such column") {
|
|
log.Tracef("migrate: %s (ignored, column no longer exists)", err)
|
|
} else {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|