From 94bc742bf1a76cb587d77c9058a1d53b485b18fe Mon Sep 17 00:00:00 2001
From: chenshijun <csj_sky@126.com>
Date: 星期一, 16 九月 2019 14:11:13 +0800
Subject: [PATCH] 修改配置文件处理参数
---
main.go | 22 +++++++++++++++-------
1 files changed, 15 insertions(+), 7 deletions(-)
diff --git a/main.go b/main.go
index 7f2c6ec..263c91a 100644
--- a/main.go
+++ b/main.go
@@ -18,29 +18,37 @@
const (
configFilePath = "/opt/vasystem/config/"
- configFileName = "config"
- configFileType = "json"
+ configFileType = "yaml"
LOGBASEPATH = "/data/disk1/valog/taskpubsub.log"
)
+var envirment = flag.String("e", "pro", "")
+
func init(){
viper.SetConfigType(configFileType)
- viper.SetConfigName(configFileName)
+ viper.SetConfigName(*envirment)
viper.AddConfigPath(configFilePath)
viper.AddConfigPath("./")
err := viper.ReadInConfig()
if err != nil {
- panic(err)
+ logger.Error("viper.ReadInConfig() Error")
+ //panic(err)
}
var logFile = LOGBASEPATH
if viper.GetString("LogBasePath") != "" {
logFile = viper.GetString("LogBasePath") + "/taskpubsub.log"
}
- var logSaveDays = 15
-
+
// 鏃ュ織鍒濆鍖�
- logger.Config(logFile, logger.InfoLevel)
+ if viper.IsSet("LogLevel") &&
+ viper.GetInt("LogLevel") >= logger.PanicLevel &&
+ viper.GetInt("LogLevel") <= logger.DebugLevel {
+ logger.Config(logFile, viper.GetInt("LogLevel"))
+ }else{
+ logger.Config(logFile, logger.DebugLevel)
+ }
+ var logSaveDays = 15
logger.SetSaveDays(logSaveDays)
logger.Info("loginit success !")
--
Gitblit v1.8.0