cmdMigrate.gotmpl

// Code generated by go-swagger/go-orusapi; DO NOT EDIT.

package cmd

import (
	"orus.io/orus-io/go-orusapi/database"
	"github.com/golang-migrate/migrate/v4/source"
	bindata "github.com/golang-migrate/migrate/v4/source/go_bindata"

	"{{ joinFilePath .TargetImportPath "migration" }}"
)

func initMigrationSource() source.Driver {
	s := bindata.Resource(migration.AssetNames(), migration.Asset)
	source, err := bindata.WithInstance(s)
	if err != nil {
		panic(err)
	}
	return source
}

var migrationSource = initMigrationSource()

// NewMigrateCmd instanciates a MigrateCmd
func NewMigrateCmd(dbOptions *database.Options, loggingOptions *orusapi.LoggingOptions) *MigrateCmd {
	return &MigrateCmd{
		db: dbOptions,
		log: loggingOptions,
	}
}

// MigrateCmd is the "migrate" command
type MigrateCmd struct{
	log *orusapi.LoggingOptions
	db *database.Options
}

// Execute does the actual migration
func (cmd *MigrateCmd) Execute([]string) error {
	cmd.log.SetMinLoggingLevel(zerolog.InfoLevel)
	log := cmd.log.Logger()
	m, err := database.NewMigrate(cmd.db.DSN, migrationSource)
	if err != nil {
		return fmt.Errorf("failed to init migration engine: %s", err)
	}
	defer m.Close()

	if err := m.Up(); err != nil {
		if err == migrate.ErrNoChange {
			log.Info().Msg("The database is already up-to-date")
		} else {
			return fmt.Errorf("failed to run migration: %s", err)
		}
	} else {
		log.Info().Msg("database successfully upgraded")
	}

	return nil
}

var Migrate = NewMigrateCmd(DatabaseOptions, LoggingOptions)

func init() {
	var migrateCmd = NewMigrateCmd(DatabaseOptions, LoggingOptions)

	_, err := parser.AddCommand("migrate", "Init/migrate the database schema", "", migrateCmd)
	if err != nil {
		Logger.Fatal().Msg(err.Error())
	}
}