package util import ( "os" "strings" "github.com/spf13/viper" ) // FillParams 填充MapParams func FillParams(key, value string) { MapParames[key] = value } // FindStringInArray find func FindStringInArray(str string, array []string) (string, bool) { found := false ret := "" for _, v := range array { if strings.EqualFold(str, v) { found = true ret = v break } } return ret, found } // IsFileExist file exist func IsFileExist(path string) bool { _, err := os.Stat(path) if err != nil { if os.IsNotExist(err) { return false } } return true } // InitConfig 读取配置文件 func InitConfig() { const ( configFilePath = "/opt/vasystem/config/" configFileName = "pro" configFileType = "yaml" LOGBASEPATH = "/data/disk1/valog/" ) viper.SetConfigType(configFileType) viper.SetConfigName(configFileName) viper.AddConfigPath(configFilePath) viper.AddConfigPath("./") LogFile = "./log/analysis-" if err := viper.ReadInConfig(); err == nil { logPath := viper.GetString("LogBasePath") if len(logPath) > 0 && IsFileExist(logPath) { if logPath[len(logPath)-1] == '/' { LogFile = logPath + "analysis-" } else { LogFile = logPath + "/analysis-" } } else if IsFileExist(LOGBASEPATH) { LogFile = LOGBASEPATH + "analysis-" } } else if IsFileExist(LOGBASEPATH) { LogFile = LOGBASEPATH + "analysis-" } }