zhangzengfei
2024-06-13 15dc6df83a839f5913997e4e2a252a0714dc2978
config/config.go
@@ -31,6 +31,7 @@
   ChannelNo          string `mapstructure:"channel-number"`     // 通道号, 同id
   HeartbeatInterval  int    `mapstructure:"heartbeat-interval"` // 心跳周期
   HeartbeatFailCount int    `mapstructure:"heartbeat-count"`    // 心跳超时次数
   AddFloorToFaceId   bool   `mapstructure:"add-floor-faceId"`
   Enable             bool   `mapstructure:"enable"`
}
@@ -46,14 +47,19 @@
   ReportServer   string `mapstructure:"report-server"`
   ReportInterval int    `mapstructure:"report-interval"`
   RetryInterval  int    `mapstructure:"retry-interval"`
   CutFaceImage   bool   `mapstructure:"cut-face-image"`
   Enable         bool   `mapstructure:"enable"`
}
type image struct {
   CutFaceImage bool `mapstructure:"cut-face-image"`
   Enlarge      int  `mapstructure:"enlarge"`
}
// 梯控设备
type nvcs struct {
   Model string // 型号
   Port  string // 端口
   Model string `mapstructure:"model"` // 型号
   Port  string `mapstructure:"port"`  // 端口
   OSD   string `mapstructure:"osd"`
}
type rateLimitConfig struct {
@@ -67,6 +73,7 @@
var RateLimitConf = &rateLimitConfig{}
var ClientConf = &client{}
var NVCSConf = &nvcs{}
var ImageConf = &image{}
// Init is an exported method that takes the environment starts the viper
// (external lib) and returns the configuration struct.
@@ -75,6 +82,7 @@
   v := viper.New()
   v.SetConfigType("yaml")
   v.SetConfigName("gat1400")
   v.AddConfigPath("./")
   v.AddConfigPath("./config/")
   v.AddConfigPath("../config/")
   err = v.ReadInConfig()
@@ -96,6 +104,7 @@
   v.UnmarshalKey("rate-limit", RateLimitConf)
   v.UnmarshalKey("client", ClientConf)
   v.UnmarshalKey("nvcs", NVCSConf)
   v.UnmarshalKey("image", ImageConf)
   if LogConf.Level == "" {
      LogConf.Level = "info"
@@ -117,5 +126,9 @@
      ClientConf.Proto = "http"
   }
   if ImageConf.Enlarge == 0 {
      ImageConf.Enlarge = 100
   }
   logger.SetLogLevel(LogConf.Level)
}