package migration

import (
	"gorm.io/gorm"

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

var _ = AddDBMigrateRecord(
	"202404230900",
	func(db *gorm.DB) error {
		if err := db.AutoMigrate(&schema.ComponentTierUnit{}); err != nil {
			return err
		}

		return db.Exec(
			"alter table storage_templates drop column min_storage_gi, drop column max_storage_gi, drop column default_storage_gi, drop column default_iops, drop column default_through_put",
		).Error
	},
)