mirror of https://github.com/muety/wakapi.git
73 lines
1.8 KiB
Go
73 lines
1.8 KiB
Go
package migrations
|
|
|
|
import (
|
|
"regexp"
|
|
"strings"
|
|
|
|
"github.com/emvi/logbuch"
|
|
"github.com/muety/wakapi/config"
|
|
"github.com/muety/wakapi/models"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// due to an error in the model definition, idx_time_user used to only cover 'user_id', but not time column
|
|
// if that's the case in the current state of the database, drop the index and let it be recreated by auto migration afterwards
|
|
func init() {
|
|
const name = "20221028-fix_heartbeats_time_user_idx"
|
|
f := migrationFunc{
|
|
name: name,
|
|
f: func(db *gorm.DB, cfg *config.Config) error {
|
|
migrator := db.Migrator()
|
|
|
|
if !migrator.HasTable(&models.Heartbeat{}) {
|
|
return nil
|
|
}
|
|
|
|
var drop bool
|
|
if cfg.Db.IsSQLite() {
|
|
// sqlite migrator doesn't support GetIndexes() currently
|
|
var ddl string
|
|
if err := db.
|
|
Table("sqlite_schema").
|
|
Select("sql").
|
|
Where("type = 'index'").
|
|
Where("tbl_name = 'heartbeats'").
|
|
Where("name = 'idx_time_user'").
|
|
Scan(&ddl).Error; err != nil {
|
|
return err
|
|
}
|
|
|
|
matches := regexp.MustCompile("(?i)\\((.+)\\)$").FindStringSubmatch(ddl)
|
|
if len(matches) > 0 && (!strings.Contains(matches[0], "`user_id`") || !strings.Contains(matches[0], "`time`")) {
|
|
drop = true
|
|
}
|
|
} else {
|
|
indexes, err := migrator.GetIndexes(&models.Heartbeat{})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
for _, idx := range indexes {
|
|
if idx.Table() == "heartbeats" && idx.Name() == "idx_time_user" && len(idx.Columns()) == 1 {
|
|
drop = true
|
|
break
|
|
}
|
|
}
|
|
}
|
|
|
|
if !drop {
|
|
return nil
|
|
}
|
|
|
|
if err := migrator.DropIndex(&models.Heartbeat{}, "idx_time_user"); err != nil {
|
|
return err
|
|
}
|
|
logbuch.Info("index 'idx_time_user' needs to be recreated, this may take a while")
|
|
|
|
return nil
|
|
},
|
|
}
|
|
|
|
registerPreMigration(f)
|
|
}
|