修改配置文件处理参数

chenshijun

2019-09-16 94bc742bf1a76cb587d77c9058a1d53b485b18fe

 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 !")