mirror of https://github.com/beego/bee.git
refactor function signatures
This commit is contained in:
ZhengYang
parent
c9b12183f8
commit
399e3ffefc
89
migrate.go
89
migrate.go
|
|
@ -19,7 +19,9 @@ import (
|
|||
"os"
|
||||
"os/exec"
|
||||
"path"
|
||||
"strconv"
|
||||
"strings"
|
||||
"time"
|
||||
)
|
||||
|
||||
var cmdMigrate = &Command{
|
||||
|
|
@ -58,6 +60,8 @@ func init() {
|
|||
}
|
||||
|
||||
func runMigration(cmd *Command, args []string) {
|
||||
crupath, _ := os.Getwd()
|
||||
|
||||
gopath := os.Getenv("GOPATH")
|
||||
Debugf("gopath:%s", gopath)
|
||||
if gopath == "" {
|
||||
|
|
@ -81,19 +85,19 @@ func runMigration(cmd *Command, args []string) {
|
|||
if len(args) == 0 {
|
||||
// run all outstanding migrations
|
||||
ColorLog("[INFO] Running all outstanding migrations\n")
|
||||
migrateUpdate(driverStr, connStr)
|
||||
migrateUpdate(crupath, driverStr, connStr)
|
||||
} else {
|
||||
mcmd := args[0]
|
||||
switch mcmd {
|
||||
case "rollback":
|
||||
ColorLog("[INFO] Rolling back the last migration operation\n")
|
||||
migrateRollback(driverStr, connStr)
|
||||
migrateRollback(crupath, driverStr, connStr)
|
||||
case "reset":
|
||||
ColorLog("[INFO] Reseting all migrations\n")
|
||||
migrateReset(driverStr, connStr)
|
||||
migrateReset(crupath, driverStr, connStr)
|
||||
case "refresh":
|
||||
ColorLog("[INFO] Refreshing all migrations\n")
|
||||
migrateReset(driverStr, connStr)
|
||||
migrateReset(crupath, driverStr, connStr)
|
||||
default:
|
||||
ColorLog("[ERRO] Command is missing\n")
|
||||
os.Exit(2)
|
||||
|
|
@ -151,41 +155,47 @@ func checkForSchemaUpdateTable(db *sql.DB) {
|
|||
}
|
||||
}
|
||||
|
||||
func getLatestMigration(db *sql.DB) (file string, createdAt string) {
|
||||
func getLatestMigration(db *sql.DB) (file string, createdAt int64) {
|
||||
sql := "SELECT name, created_at FROM migrations where status = 'update' ORDER BY id_migration DESC LIMIT 1"
|
||||
if rows, err := db.Query(sql); err != nil {
|
||||
ColorLog("[ERRO] Could not retrieve migrations: %s\n", err)
|
||||
os.Exit(2)
|
||||
} else {
|
||||
var fileBytes, createdAtBytes []byte
|
||||
var createdAtStr string
|
||||
if rows.Next() {
|
||||
if err := rows.Scan(&fileBytes, &createdAtBytes); err != nil {
|
||||
if err := rows.Scan(&file, &createdAtStr); err != nil {
|
||||
ColorLog("[ERRO] Could not read migrations in database: %s\n", err)
|
||||
os.Exit(2)
|
||||
}
|
||||
file, createdAt = string(fileBytes), string(createdAtBytes)
|
||||
if t, err := time.Parse("2006-01-02 15:04:05", createdAtStr); err != nil {
|
||||
ColorLog("[ERRO] Could not parse time: %s\n", err)
|
||||
os.Exit(2)
|
||||
} else {
|
||||
createdAt = t.Unix()
|
||||
}
|
||||
} else {
|
||||
file, createdAt = "", "0"
|
||||
file, createdAt = "", 0
|
||||
}
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func createTempMigrationDir(path string) {
|
||||
if err := os.MkdirAll(path, 0777); err != nil {
|
||||
ColorLog("[ERRO] Could not create path: %s\n", err)
|
||||
func createTempMigrationDir(dir string) {
|
||||
if err := os.MkdirAll(dir, 0777); err != nil {
|
||||
ColorLog("[ERRO] Could not create directory: %s\n", err)
|
||||
os.Exit(2)
|
||||
}
|
||||
}
|
||||
|
||||
func writeMigrationSourceFile(filename string, driver string, connStr string, latestTime string, latestName string, task string) {
|
||||
if f, err := os.OpenFile(filename+".go", os.O_CREATE|os.O_EXCL|os.O_RDWR, 0666); err != nil {
|
||||
func writeMigrationSourceFile(dir, source, driver, connStr string, latestTime int64, latestName string, task string) {
|
||||
os.Chdir(dir)
|
||||
if f, err := os.OpenFile(source, os.O_CREATE|os.O_EXCL|os.O_RDWR, 0666); err != nil {
|
||||
ColorLog("[ERRO] Could not create file: %s\n", err)
|
||||
os.Exit(2)
|
||||
} else {
|
||||
content := strings.Replace(MIGRATION_MAIN_TPL, "{{DBDriver}}", driver, -1)
|
||||
content = strings.Replace(content, "{{ConnStr}}", connStr, -1)
|
||||
content = strings.Replace(content, "{{LatestTime}}", latestTime, -1)
|
||||
content = strings.Replace(content, "{{LatestTime}}", strconv.FormatInt(latestTime, 10), -1)
|
||||
content = strings.Replace(content, "{{LatestName}}", latestName, -1)
|
||||
content = strings.Replace(content, "{{Task}}", task, -1)
|
||||
if _, err := f.WriteString(content); err != nil {
|
||||
|
|
@ -196,9 +206,9 @@ func writeMigrationSourceFile(filename string, driver string, connStr string, la
|
|||
}
|
||||
}
|
||||
|
||||
func buildMigrationBinary(filename string) {
|
||||
os.Chdir(path.Join("database", "migrations"))
|
||||
cmd := exec.Command("go", "build", "-o", filename)
|
||||
func buildMigrationBinary(dir, binary string) {
|
||||
os.Chdir(dir)
|
||||
cmd := exec.Command("go", "build", "-o", binary)
|
||||
if out, err := cmd.CombinedOutput(); err != nil {
|
||||
ColorLog("[ERRO] Could not build migration binary: %s\n", err)
|
||||
formatShellErrOutput(string(out))
|
||||
|
|
@ -206,8 +216,9 @@ func buildMigrationBinary(filename string) {
|
|||
}
|
||||
}
|
||||
|
||||
func runMigrationBinary(filename string) {
|
||||
cmd := exec.Command("./" + filename)
|
||||
func runMigrationBinary(dir, binary string) {
|
||||
os.Chdir(dir)
|
||||
cmd := exec.Command("./" + binary)
|
||||
if out, err := cmd.CombinedOutput(); err != nil {
|
||||
formatShellOutput(string(out))
|
||||
ColorLog("[ERRO] Could not run migration binary: %s\n", err)
|
||||
|
|
@ -217,31 +228,34 @@ func runMigrationBinary(filename string) {
|
|||
}
|
||||
}
|
||||
|
||||
func removeMigrationBinary(path string) {
|
||||
if err := os.Remove(path); err != nil {
|
||||
ColorLog("[ERRO] Could not remove migration binary: %s\n", err)
|
||||
func removeTempFile(dir, file string) {
|
||||
os.Chdir(dir)
|
||||
if err := os.Remove(file); err != nil {
|
||||
ColorLog("[ERRO] Could not remove temporary migration files: %s\n", err)
|
||||
os.Exit(2)
|
||||
}
|
||||
}
|
||||
|
||||
func migrateUpdate(driver, connStr string) {
|
||||
migrate("upgrade", driver, connStr)
|
||||
func migrateUpdate(crupath, driver, connStr string) {
|
||||
migrate("upgrade", crupath, driver, connStr)
|
||||
}
|
||||
|
||||
func migrateRollback(driver, connStr string) {
|
||||
migrate("rollback", driver, connStr)
|
||||
func migrateRollback(crupath, driver, connStr string) {
|
||||
migrate("rollback", crupath, driver, connStr)
|
||||
}
|
||||
|
||||
func migrateReset(driver, connStr string) {
|
||||
migrate("reset", driver, connStr)
|
||||
func migrateReset(crupath, driver, connStr string) {
|
||||
migrate("reset", crupath, driver, connStr)
|
||||
}
|
||||
|
||||
func migrateRefresh(driver, connStr string) {
|
||||
migrate("refresh", driver, connStr)
|
||||
func migrateRefresh(crupath, driver, connStr string) {
|
||||
migrate("refresh", crupath, driver, connStr)
|
||||
}
|
||||
|
||||
func migrate(goal, driver, connStr string) {
|
||||
filepath := path.Join("database", "migrations", "migrate")
|
||||
func migrate(goal, crupath, driver, connStr string) {
|
||||
dir := path.Join(crupath, "database", "migrations")
|
||||
binary := "m"
|
||||
source := binary + ".go"
|
||||
// connect to database
|
||||
db, err := sql.Open(driver, connStr)
|
||||
if err != nil {
|
||||
|
|
@ -251,10 +265,11 @@ func migrate(goal, driver, connStr string) {
|
|||
defer db.Close()
|
||||
checkForSchemaUpdateTable(db)
|
||||
latestName, latestTime := getLatestMigration(db)
|
||||
writeMigrationSourceFile(filepath, driver, connStr, latestTime, latestName, goal)
|
||||
buildMigrationBinary(filepath)
|
||||
runMigrationBinary(filepath)
|
||||
removeMigrationBinary(filepath)
|
||||
writeMigrationSourceFile(dir, source, driver, connStr, latestTime, latestName, goal)
|
||||
buildMigrationBinary(dir, binary)
|
||||
runMigrationBinary(dir, binary)
|
||||
removeTempFile(dir, source)
|
||||
removeTempFile(dir, binary)
|
||||
}
|
||||
|
||||
func formatShellErrOutput(o string) {
|
||||
|
|
|
|||
Loading…
Reference in New Issue