nightingale/models/migrate/migrate.go

package migrate

import (
	"github.com/toolkits/pkg/logger"
	"gorm.io/gorm"
)

func Migrate(db *gorm.DB) {
	MigrateRecordingTable(db)
	MigrateEsIndexPatternTable(db)
}

type RecordingRule struct {
	QueryConfigs string `gorm:"type:text;not null;column:query_configs"` // query_configs
}

func MigrateRecordingTable(db *gorm.DB) error {
	err := db.AutoMigrate(&RecordingRule{})
	if err != nil {
		logger.Errorf("failed to migrate recording rule table: %v", err)
		return err
	}

	err = db.AutoMigrate(&AlertRule{})
	if err != nil {
		logger.Errorf("failed to migrate recording rule table: %v", err)
		return err
	}

	err = db.AutoMigrate(&AlertSubscribe{})
	if err != nil {
		logger.Errorf("failed to migrate recording rule table: %v", err)
		return err
	}

	err = db.AutoMigrate(&AlertMute{})
	if err != nil {
		logger.Errorf("failed to migrate recording rule table: %v", err)
		return err
	}
	return nil
}

type AlertRule struct {
	ExtraConfig string `gorm:"type:text;not null;column:extra_config"` // extra config
}

type AlertSubscribe struct {
	ExtraConfig string `gorm:"type:text;not null;column:extra_config"` // extra config
	Severities  string `gorm:"column:severities;type:varchar(32);not null;default:''"`
}

type AlertMute struct {
	Severities string `gorm:"column:severities;type:varchar(32);not null;default:''"`
}