zhangzengfei
2024-10-19 1ca4879df9cc3c22181b2a200aa4b793116d3d35
config/config.go
@@ -7,19 +7,29 @@
)
type common struct {
   EsUrl     string `mapstructure: "esUrl"`
   ServerUrl string `mapstructure: "serverUrl"`
   OrgName   string `mapstructure: "orgName"`
   EsUrl     string `mapstructure:"esUrl"`
   ServerUrl string `mapstructure:"serverUrl"`
   OrgName   string `mapstructure:"orgName"`
   Interval  int    `mapstructure:"pushInterval"`
}
type logConfig struct {
   Path       string `mapstructure:"path"`       //日志存储路径
   Level      int    `mapstructure:"level"`      //日志等级
   MaxSize    int    `mapstructure:"maxSize"`    //日志文件大小上限
   MaxBackups int    `mapstructure:"maxBackups"` //日志压缩包个数
   MaxAge     int    `mapstructure:"maxAge"`     //保留压缩包天数
}
var Options = &common{}
var LogConf = &logConfig{}
func Init() {
   var err error
   viper.SetConfigType("yaml")
   viper.SetConfigName("esSync")
   viper.AddConfigPath("../config")
   viper.SetConfigName("esSyncClient")
   viper.AddConfigPath("config")
   viper.AddConfigPath("../config")
   viper.AddConfigPath("")
   err = viper.ReadInConfig()
   if err != nil {
@@ -27,6 +37,8 @@
   }
   viper.UnmarshalKey("common", Options)
   viper.UnmarshalKey("log", LogConf)
   viper.WatchConfig()
   viper.OnConfigChange(func(in fsnotify.Event) {
      viper.UnmarshalKey("common", Options)