package migrate

import (
	"gorm.io/gorm"
)

var _ = AddCoreMigrateRecord(
	"202409201010",
	func(db *gorm.DB) error {
		err := db.Exec("ALTER TABLE `tier_units` MODIFY COLUMN `component_type` ENUM('pd', 'tikv', 'tidb', 'tiflash', 'ticdc', 'tiproxy') NOT NULL;").Error

		if err != nil {
			return err
		}

		return db.Exec("ALTER TABLE `tiers` MODIFY COLUMN `component_type` ENUM('pd', 'tikv', 'tidb', 'tiflash', 'ticdc', 'tiproxy') NOT NULL;").Error
	},
)