package migrations import ( "github.com/emvi/logbuch" "github.com/muety/wakapi/config" "github.com/muety/wakapi/models" "gorm.io/gorm" "regexp" "strings" ) // 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) }