package migration

import (
	"braces.dev/errtrace"
	"gorm.io/gorm"

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

func init() {
	AddDBMigrateRecord("202605210001", migrate202605210001)
}

func migrate202605210001(db *gorm.DB) error {
	if !db.Migrator().HasColumn(&schema.ResourcePool{}, "PlacementGroupEnabled") {
		if err := db.Migrator().AddColumn(&schema.ResourcePool{}, "PlacementGroupEnabled"); err != nil {
			return errtrace.Wrap(err)
		}
	}
	return nil
}