You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
paopao-ce/internal/migration/migration_embed.go

95 lines
2.8 KiB

// Copyright 2022 ROC. All rights reserved.
// Use of this source code is governed by a MIT style
// license that can be found in the LICENSE file.
//go:build migration
// +build migration
package migration
import (
"database/sql"
"github.com/alimy/cfg"
"github.com/golang-migrate/migrate/v4"
"github.com/golang-migrate/migrate/v4/database"
"github.com/golang-migrate/migrate/v4/database/mysql"
"github.com/golang-migrate/migrate/v4/database/postgres"
"github.com/golang-migrate/migrate/v4/database/sqlite3"
"github.com/golang-migrate/migrate/v4/source"
"github.com/golang-migrate/migrate/v4/source/iofs"
"github.com/rocboss/paopao-ce/internal/conf"
"github.com/rocboss/paopao-ce/scripts/migration"
"github.com/sirupsen/logrus"
)
func Run() {
if !cfg.If("Migration") {
logrus.Infoln("skip migrate because not add Migration feature in config.yaml")
return
}
var (
db *sql.DB
dbName string
dbDriver database.Driver
srcDriver source.Driver
err, err2 error
)
if cfg.If("MySQL") {
dbName = conf.MysqlSetting.DBName
db, err = sql.Open("mysql", conf.MysqlSetting.Dsn()+"&multiStatements=true")
} else if cfg.If("PostgreSQL") || cfg.If("Postgres") {
dbName = (*conf.PostgresSetting)["DBName"]
db, err = sql.Open("pgx", conf.PostgresSetting.Dsn())
} else if cfg.If("Sqlite3") {
_, db, err = conf.OpenSqlite3()
} else {
dbName = conf.MysqlSetting.DBName
db, err = sql.Open("mysql", conf.MysqlSetting.Dsn())
}
if err != nil {
logrus.Errorf("initial db for migration failed: %s", err)
return
}
migrationsTable := conf.DatabaseSetting.TablePrefix + "schema_migrations"
if cfg.If("MySQL") {
srcDriver, err = iofs.New(migration.Files, "mysql")
dbDriver, err2 = mysql.WithInstance(db, &mysql.Config{MigrationsTable: migrationsTable})
} else if cfg.If("PostgreSQL") || cfg.If("Postgres") {
srcDriver, err = iofs.New(migration.Files, "postgres")
dbDriver, err2 = postgres.WithInstance(db, &postgres.Config{MigrationsTable: migrationsTable})
} else if cfg.If("Sqlite3") {
srcDriver, err = iofs.New(migration.Files, "sqlite3")
dbDriver, err2 = sqlite3.WithInstance(db, &sqlite3.Config{MigrationsTable: migrationsTable})
} else {
srcDriver, err = iofs.New(migration.Files, "mysql")
dbDriver, err2 = mysql.WithInstance(db, &mysql.Config{MigrationsTable: migrationsTable})
}
if err2 != nil {
logrus.Errorf("new database driver failed: %s", err)
return
} else {
defer dbDriver.Close()
}
if err != nil {
logrus.Errorf("new source driver failed: %s", err)
return
}
m, err := migrate.NewWithInstance("iofs", srcDriver, dbName, dbDriver)
if err != nil {
logrus.Errorf("new migrate instance failed: %s", err)
return
}
if err = m.Up(); err != nil && err != migrate.ErrNoChange {
logrus.Errorf("migrate up failed: %s", err)
return
}
logrus.Infoln("migrate up success")
}