lbsadmin/bee
1
0
Fork 0
mirror of https://github.com/beego/bee.git synced 2024-11-25 20:10:55 +00:00
Code Issues Releases Activity

change directory

Browse Source
This commit is contained in:
ZhengYang 2014-08-12 16:04:50 +08:00
parent 4c735647c3
commit 9a728714aa
1 changed files with 8 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
migrate.go
View File

@ -48,10 +48,6 @@ bee migrate refresh
`,
}
const (
TMP_DIR = "temp"
)
var mDriver docValue
var mConn docValue
@ -218,9 +214,9 @@ func runMigrationBinary(filename string) {
}
}
func cleanUpMigrationFiles(tmpPath string) {
if err := os.RemoveAll(tmpPath); err != nil {
ColorLog("[ERRO] Could not remove temporary migration directory: %s\n", err)
func removeMigrationBinary(path string) {
if err := os.Remove(path); err != nil {
ColorLog("[ERRO] Could not remove migration binary: %s\n", err)
os.Exit(2)
}
}
@ -242,7 +238,7 @@ func migrateRefresh(driver, connStr string) {
}
func migrate(goal, driver, connStr string) {
filename := path.Join(TMP_DIR, "migrate")
filepath := path.Join("database", "migrations", "migrate")
// connect to database
db, err := sql.Open(driver, connStr)
if err != nil {
@ -252,11 +248,10 @@ func migrate(goal, driver, connStr string) {
defer db.Close()
checkForSchemaUpdateTable(db)
latestName, latestTime := getLatestMigration(db)
createTempMigrationDir(TMP_DIR)
writeMigrationSourceFile(filename, driver, connStr, latestTime, latestName, goal)
buildMigrationBinary(filename)
runMigrationBinary(filename)
cleanUpMigrationFiles(TMP_DIR)
writeMigrationSourceFile(filepath, driver, connStr, latestTime, latestName, goal)
buildMigrationBinary(filepath)
runMigrationBinary(filepath)
removeMigrationBinary(filepath)
}
const (