package migration

import (
	"braces.dev/errtrace"
	"gorm.io/gorm"

	"github.com/tidbcloud/tidb-management-service/nextgen-global-service/infra/repoimpl/schema"
)

func init() {
	AddDBMigrateRecord("202605110001", migrate202605110001)
}

func migrate202605110001(db *gorm.DB) error {
	if err := db.AutoMigrate(&schema.ReservedCapacityTemplate{}); err != nil {
		return errtrace.Wrap(err)
	}
	if !db.Migrator().HasColumn(&schema.Cluster{}, "reserved_capacity_template_id") {
		if err := db.Migrator().AddColumn(&schema.Cluster{}, "reserved_capacity_template_id"); err != nil {
			return errtrace.Wrap(err)
		}
	}
	return nil
}