zhangzengfei
2024-08-25 5f31d07ff92d87dd1016e9279df84d8f3641feb5
config/config.go
@@ -36,7 +36,7 @@
   Enable             bool   `mapstructure:"enable"`
}
type logConfig struct {
type log struct {
   LogWay string `mapstructure:"log-way"` // 日志输出
   Path   string `mapstructure:"path"`    // 日志存储路径
   Level  string `mapstructure:"level"`   // 日志等级
@@ -67,18 +67,24 @@
   WaitRunTime int    `mapstructure:"wait-run-time"`
}
type rateLimitConfig struct {
type rateLimit struct {
   FillInterval int64 `mapstructure:"fill-interval" json:"fillInterval"`
   Capacity     int64 `mapstructure:"capacity" json:"capacity"`
}
type sysTime struct {
   NTPServer    string `mapstructure:"ntp-server" json:"ntp-server"`
   SyncInterval int    `mapstructure:"sync-interval" json:"sync-interval"`
}
var ServeConf = &serve{}
var LogConf = &logConfig{}
var LogConf = &log{}
var ForwardConf = &forward{}
var RateLimitConf = &rateLimitConfig{}
var RateLimitConf = &rateLimit{}
var ClientConf = &client{}
var NVCSConf = &nvcs{}
var ImageConf = &image{}
var SysTimeConf = &sysTime{}
// Init is an exported method that takes the environment starts the viper
// (external lib) and returns the configuration struct.
@@ -139,5 +145,13 @@
      NVCSConf.WaitRunTime = 60
   }
   if SysTimeConf.SyncInterval == 0 {
      SysTimeConf.SyncInterval = 10
   }
   if SysTimeConf.NTPServer == "" {
      SysTimeConf.NTPServer = "ntp.aliyun.com"
   }
   logger.SetLogLevel(LogConf.Level)
}