package migration

import (
	"gorm.io/gorm"

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

// Migration 202602030001: Add vcpus_limit to resource_pools and vcpus_limit_per_resource_pool to rcu_quotas
var _ = AddDBMigrateRecord(
	"202602030001",
	func(db *gorm.DB) error {
		return db.Transaction(func(tx *gorm.DB) error {
			if err := tx.AutoMigrate(&schema.ResourcePool{}); err != nil {
				return err
			}
			if err := tx.AutoMigrate(&schema.RcuQuota{}); err != nil {
				return err
			}
			return nil
		})
	},
)