53 lines
959 B
Go
53 lines
959 B
Go
package db
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"time"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type localMigration struct {
|
|
Name string
|
|
Query string
|
|
}
|
|
|
|
var migrations = []localMigration{
|
|
{
|
|
Name: "2021_06_24_04_42_SetUserDisabledFalse",
|
|
Query: "update users set is_disabled=0",
|
|
},
|
|
{
|
|
Name: "2021_02_07_00_09_LowerCaseEmails",
|
|
Query: "update users set email=lower(email)",
|
|
|
|
},
|
|
{
|
|
Name: "2022_03_08_13_16_AddVIN",
|
|
Query: "ALTER TABLE vehicles ADD COLUMN vin text",
|
|
},
|
|
}
|
|
|
|
func RunMigrations() {
|
|
for _, mig := range migrations {
|
|
ExecuteAndSaveMigration(mig.Name, mig.Query)
|
|
}
|
|
}
|
|
func ExecuteAndSaveMigration(name string, query string) error {
|
|
var migration Migration
|
|
result := DB.Where("name=?", name).First(&migration)
|
|
if errors.Is(result.Error, gorm.ErrRecordNotFound) {
|
|
fmt.Println(query)
|
|
result = DB.Debug().Exec(query)
|
|
if result.Error == nil {
|
|
DB.Save(&Migration{
|
|
Date: time.Now(),
|
|
Name: name,
|
|
})
|
|
}
|
|
return result.Error
|
|
}
|
|
return nil
|
|
}
|