lbsadmin/bee
1
0
Fork 0
mirror of https://github.com/beego/bee.git synced 2024-11-21 18:40:54 +00:00
Code Issues Releases Activity

support beego pro migration exec sql

Browse Source
This commit is contained in:
yitea 2020-07-11 17:44:39 +08:00
parent cf3c3bdc8b
commit 2c329fd606
2 changed files with 52 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
cmd/commands/beegopro/beegopro.go
View File

@ -32,6 +32,8 @@ var CmdBeegoPro = &commands.Command{
func init() { func init() {
CmdBeegoPro.Flag.Var(&beegopro.SQL, "sql", "sql file path") CmdBeegoPro.Flag.Var(&beegopro.SQL, "sql", "sql file path")
CmdBeegoPro.Flag.Var(&beegopro.SQLMode, "sqlmode", "sql mode")
CmdBeegoPro.Flag.Var(&beegopro.SQLModePath, "sqlpath", "sql mode path")
commands.AvailableCommands = append(commands.AvailableCommands, CmdBeegoPro) commands.AvailableCommands = append(commands.AvailableCommands, CmdBeegoPro)
} }

56
internal/app/module/beegopro/migration.go
View File

@ -6,9 +6,17 @@ import (
"github.com/beego/bee/utils" "github.com/beego/bee/utils"
"io/ioutil" "io/ioutil"
"path/filepath" "path/filepath"
"strings"
) )
var SQL utils.DocValue var SQL utils.DocValue
var SQLMode utils.DocValue
var SQLModePath utils.DocValue
var (
SQLModeUp = "up"
SQLModeDown = "down"
)
func (c *Container) Migration(args []string) { func (c *Container) Migration(args []string) {
c.initUserOption() c.initUserOption()
@ -17,19 +25,55 @@ func (c *Container) Migration(args []string) {
beeLogger.Log.Fatalf("Could not connect to '%s' database using '%s': %s", c.UserOption.Driver, c.UserOption.Dsn, err) beeLogger.Log.Fatalf("Could not connect to '%s' database using '%s': %s", c.UserOption.Driver, c.UserOption.Dsn, err)
return return
} }
defer db.Close() defer db.Close()
switch SQLMode.String() {
case SQLModeUp:
doByMode(db, "up.sql")
case SQLModeDown:
doByMode(db, "down.sql")
default:
doBySqlFile(db)
}
}
absFile, _ := filepath.Abs(SQL.String()) func doBySqlFile(db *sql.DB) {
content, err := ioutil.ReadFile(SQL.String()) fileName := SQL.String()
if !utils.IsExist(fileName) {
beeLogger.Log.Fatalf("sql mode path not exist, path %s", SQL.String())
}
doDb(db, fileName)
}
func doByMode(db *sql.DB, suffix string) {
pathName := SQLModePath.String()
if !utils.IsExist(pathName) {
beeLogger.Log.Fatalf("sql mode path not exist, path %s", SQLModePath.String())
}
rd, err := ioutil.ReadDir(pathName)
if err != nil {
beeLogger.Log.Fatalf("read dir err, path %s, err %s", pathName, err)
}
for _, fi := range rd {
if !fi.IsDir() {
if !strings.HasSuffix(fi.Name(), suffix) {
continue
}
doDb(db, filepath.Join(pathName, fi.Name()))
}
}
}
func doDb(db *sql.DB, filePath string) {
absFile, _ := filepath.Abs(filePath)
content, err := ioutil.ReadFile(filePath)
if err != nil { if err != nil {
beeLogger.Log.Errorf("read file err %s, abs file %s", err, absFile) beeLogger.Log.Errorf("read file err %s, abs file %s", err, absFile)
} }
result, err := db.Exec(string(content)) _, err = db.Exec(string(content))
if err != nil { if err != nil {
beeLogger.Log.Errorf("db exec err %s", err) beeLogger.Log.Errorf("db exec err %s", err)
} }
beeLogger.Log.Infof("db exec info %v", result) beeLogger.Log.Infof("db exec info %s", filePath)
} }