feat: support custom config path via "config"

This commit is contained in:
Yang Luo 2024-01-06 14:09:48 +08:00
parent 741cff99df
commit 2d6fae32be

View File

@ -38,10 +38,12 @@ import (
var ( var (
ormer *Ormer = nil ormer *Ormer = nil
createDatabase = true createDatabase = true
configPath = "conf/app.conf"
) )
func InitFlag() { func InitFlag() {
createDatabase = getCreateDatabaseFlag() createDatabase = getCreateDatabaseFlag()
configPath = getConfigFlag()
} }
func getCreateDatabaseFlag() bool { func getCreateDatabaseFlag() bool {
@ -50,6 +52,12 @@ func getCreateDatabaseFlag() bool {
return *res return *res
} }
func getConfigFlag() string {
res := flag.String("config", "conf/app.conf", "set it to \"/your/path/app.conf\" if your config file is not in: \"/conf/app.conf\"")
flag.Parse()
return *res
}
func InitConfig() { func InitConfig() {
err := beego.LoadAppConfig("ini", "../conf/app.conf") err := beego.LoadAppConfig("ini", "../conf/app.conf")
if err != nil { if err != nil {
@ -64,7 +72,7 @@ func InitConfig() {
func InitAdapter() { func InitAdapter() {
if conf.GetConfigString("driverName") == "" { if conf.GetConfigString("driverName") == "" {
if !util.FileExist("conf/app.conf") { if !util.FileExist(configPath) {
dir, err := os.Getwd() dir, err := os.Getwd()
if err != nil { if err != nil {
panic(err) panic(err)