From a45d5a83c5a283444a407d648432e9dd08f89478 Mon Sep 17 00:00:00 2001
From: chenshijun <csj_sky@126.com>
Date: 星期三, 11 九月 2019 20:34:16 +0800
Subject: [PATCH] 增加配置文件处理参数
---
main.go | 24 +++++++++++++++++++++---
1 files changed, 21 insertions(+), 3 deletions(-)
diff --git a/main.go b/main.go
index 2888fd0..7f2c6ec 100644
--- a/main.go
+++ b/main.go
@@ -10,15 +10,33 @@
"taskpubsub/tasktag"
"taskpubsub/util"
"basic.com/valib/logger.git"
-
+ "github.com/spf13/viper"
)
var initchan = make(chan bool)
-
var useShm bool
+const (
+ configFilePath = "/opt/vasystem/config/"
+ configFileName = "config"
+ configFileType = "json"
+ LOGBASEPATH = "/data/disk1/valog/taskpubsub.log"
+)
+
func init(){
- var logFile = "./taskpubsub.log"
+ viper.SetConfigType(configFileType)
+ viper.SetConfigName(configFileName)
+ viper.AddConfigPath(configFilePath)
+ viper.AddConfigPath("./")
+ err := viper.ReadInConfig()
+ if err != nil {
+ panic(err)
+ }
+
+ var logFile = LOGBASEPATH
+ if viper.GetString("LogBasePath") != "" {
+ logFile = viper.GetString("LogBasePath") + "/taskpubsub.log"
+ }
var logSaveDays = 15
// 鏃ュ織鍒濆鍖�
--
Gitblit v1.8.0