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

refactor long if to switch case, more error messages

Browse Source
This commit is contained in:
ZhengYang 2014-08-21 10:45:52 +08:00
parent 9bf23ea88a
commit 4fe9c4ddcd
1 changed files with 18 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
g_appcode.go
View File

@ -198,13 +198,14 @@ func (tag *OrmTag) String() string {
func generateAppcode(driver, connStr, level, tables, currpath string) { func generateAppcode(driver, connStr, level, tables, currpath string) {
var mode byte var mode byte
if level == "1" { switch level {
case "1":
mode = O_MODEL mode = O_MODEL
} else if level == "2" { case "2":
mode = O_MODEL | O_CONTROLLER mode = O_MODEL | O_CONTROLLER
} else if level == "3" { case "3":
mode = O_MODEL | O_CONTROLLER | O_ROUTER mode = O_MODEL | O_CONTROLLER | O_ROUTER
} else { default:
ColorLog("[ERRO] Invalid 'level' option: %s\n", level) ColorLog("[ERRO] Invalid 'level' option: %s\n", level)
ColorLog("[HINT] Level must be either 1, 2 or 3\n") ColorLog("[HINT] Level must be either 1, 2 or 3\n")
os.Exit(2) os.Exit(2)
@ -216,6 +217,19 @@ func generateAppcode(driver, connStr, level, tables, currpath string) {
selectedTables[v] = true selectedTables[v] = true
} }
} }
switch driver {
case "mysql":
fmt.Println("MySQL")
case "postgres":
fmt.Println("PostgreSQL")
case "sqlite":
ColorLog("[ERRO] Generating app code from SQLite database is not supported yet.\n")
os.Exit(2)
default:
ColorLog("[ERRO] Unknown database driver: %s\n", driver)
ColorLog("[HINT] Driver must be one of mysql, postgres or sqlite\n")
os.Exit(2)
}
gen(driver, connStr, mode, selectedTables, currpath) gen(driver, connStr, mode, selectedTables, currpath)
} }