From d671ea8ec8658da21990d5418c981d74f6015045 Mon Sep 17 00:00:00 2001
From: liuxiaolong <736321739@qq.com>
Date: 星期三, 11 十二月 2019 11:50:43 +0800
Subject: [PATCH] add conf path

---
 server.go               |    3 ++-
 extend/config/config.go |    4 ++--
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/extend/config/config.go b/extend/config/config.go
index 3912a95..54b3bf9 100644
--- a/extend/config/config.go
+++ b/extend/config/config.go
@@ -19,11 +19,11 @@
 
 var ServerMap = make(map[string]string,0)
 var NgxMap = make(map[string]string,0)
-func Init(env string) {
+func Init(env string, configPath string) {
 	var err error
 	viper.SetConfigType("yaml")
 	viper.SetConfigName(env)
-	viper.AddConfigPath("./config/")
+	viper.AddConfigPath(configPath)
 	err = viper.ReadInConfig()
 	if err != nil {
 		log.Fatal("error on parsing configuration file",err)
diff --git a/server.go b/server.go
index 404fb40..0a1a4c8 100644
--- a/server.go
+++ b/server.go
@@ -28,12 +28,13 @@
 var mqPort = flag.Int("mqPort", 5672, "default mq port")
 var mqUser = flag.String("mqUser", "basic", "default rabbitmq user")
 var mqPass = flag.String("mqPass", "basic", "default rabbitmq pass")
+var confPath = flag.String("confPath", "/home/basic/Apps/go/AndroidAlarmServer/config/","default config path")
 
 func main() {
 	flag.Parse()
 
 	log.SetLogLevel(*Level)
-	config.Init(*env)
+	config.Init(*env, *confPath)
 	fmt.Println(*port)
 	//src := *addr + ":" + strconv.Itoa(*port)
 	//listener, err := net.Listen("tcp", src)

--
Gitblit v1.8.0