From ba96922f40f82d12e9a5f4cfa4d887e78f460125 Mon Sep 17 00:00:00 2001
From: zhangmeng <775834166@qq.com>
Date: 星期四, 26 十二月 2019 15:23:17 +0800
Subject: [PATCH] update
---
main.go | 38 ++++++++++++++++++++++++++++++++------
1 files changed, 32 insertions(+), 6 deletions(-)
diff --git a/main.go b/main.go
index 2888fd0..263c91a 100644
--- a/main.go
+++ b/main.go
@@ -10,19 +10,45 @@
"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/"
+ configFileType = "yaml"
+ LOGBASEPATH = "/data/disk1/valog/taskpubsub.log"
+)
+
+var envirment = flag.String("e", "pro", "")
+
func init(){
- var logFile = "./taskpubsub.log"
- var logSaveDays = 15
-
+ viper.SetConfigType(configFileType)
+ viper.SetConfigName(*envirment)
+ viper.AddConfigPath(configFilePath)
+ viper.AddConfigPath("./")
+ err := viper.ReadInConfig()
+ if err != nil {
+ logger.Error("viper.ReadInConfig() Error")
+ //panic(err)
+ }
+
+ var logFile = LOGBASEPATH
+ if viper.GetString("LogBasePath") != "" {
+ logFile = viper.GetString("LogBasePath") + "/taskpubsub.log"
+ }
+
// 鏃ュ織鍒濆鍖�
- 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