修改配置文件处理参数

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