mirror of https://github.com/rocboss/paopao-ce
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.
95 lines
2.8 KiB
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/tryst/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")
|
|
}
|