zhangzengfei
2024-05-22 f946512f64f1e2e9aa887f62f04f8ebedb53d810
config/config.go
@@ -13,11 +13,14 @@
   Mode     string `mapstructure:"mode"`
   Host     string `mapstructure:"host"`
   Port     string `mapstructure:"port"`
   Realm    string `mapstructure:"realm"`
   Username string `mapstructure:"username"`
   Password string `mapstructure:"password"`
   Role     string `mapstructure:"role"` // agent 设备端, proxy 1400中转 不处理deivce, server 全功能
}
type client struct {
   ServerId           string `mapstructure:"server-id"`
   DeviceID           string `mapstructure:"device-id"`
   Username           string `mapstructure:"username"`
   Password           string `mapstructure:"password"`
@@ -47,6 +50,13 @@
   Enable         bool   `mapstructure:"enable"`
}
// 梯控设备
type nvcs struct {
   Model string `mapstructure:"model"` // 型号
   Port  string `mapstructure:"port"`  // 端口
   OSD   string `mapstructure:"osd"`
}
type rateLimitConfig struct {
   FillInterval int64 `mapstructure:"fill-interval" json:"fillInterval"`
   Capacity     int64 `mapstructure:"capacity" json:"capacity"`
@@ -57,6 +67,7 @@
var ForwardConf = &forward{}
var RateLimitConf = &rateLimitConfig{}
var ClientConf = &client{}
var NVCSConf = &nvcs{}
// Init is an exported method that takes the environment starts the viper
// (external lib) and returns the configuration struct.
@@ -85,6 +96,7 @@
   v.UnmarshalKey("forward", ForwardConf)
   v.UnmarshalKey("rate-limit", RateLimitConf)
   v.UnmarshalKey("client", ClientConf)
   v.UnmarshalKey("nvcs", NVCSConf)
   if LogConf.Level == "" {
      LogConf.Level = "info"