package migration

import (
	"gorm.io/gorm"

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

var _ = AddDBMigrateRecord(
	"202405201646",
	func(db *gorm.DB) error {
		if err := db.AutoMigrate(&schema.TierTierUnit{}); err != nil {
			return err
		}
		if err := db.Exec(
			"delete from tier_unit_conditions where component_tier_unit_id in (select id from component_tier_units where component_type != 'pd')",
		).Error; err != nil {
			return err
		}
		return nil
	},
)