package migration

import (
	"gorm.io/gorm"
)

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

		if err != nil {
			return err
		}

		err = db.Exec(
			"ALTER TABLE `component_tiers` MODIFY COLUMN `component_type` ENUM('pd', 'tikv', 'tidb', 'tiflash', 'tiproxy') NOT NULL;",
		).Error

		if err != nil {
			return err
		}

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