package migrate

import (
	"github.com/pingcap/log"
	"go.uber.org/zap"
	"gorm.io/gorm"
)

var _ = AddDBMigrateRecord(
	"202408121630",
	func(db *gorm.DB) error {
		err := db.Exec("ALTER TABLE exports add column `gcs_uri` varchar(1024) NULL DEFAULT NULL;").Error
		if err != nil {
			log.Error("add export column `gcs_uri` failed", zap.Error(err))
		}
		err = db.Exec("ALTER TABLE exports add column `az_blob_uri` varchar(1024) NULL DEFAULT NULL;").Error
		if err != nil {
			log.Error("add export column `az_blob_uri` failed", zap.Error(err))
		}
		err = db.Exec(
			"ALTER TABLE exports modify column `target_type` enum('LOCAL','S3','GCS','Azure') NOT NULL DEFAULT 'LOCAL';",
		).Error
		if err != nil {
			log.Error("alter export column `target_type` failed", zap.Error(err))
		}
		err = db.Exec("ALTER TABLE exports add column `parquet_format` json NULL;").Error
		if err != nil {
			log.Error("add export column `parquet_format` failed", zap.Error(err))
		}
		return nil
	},
)