zhangzengfei
2024-06-05 52830a58e2907f83e8f891728b8ba3080ca59fb1
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"`
@@ -28,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"`
}
@@ -43,8 +47,21 @@
   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"`
   Enable bool `mapstructure:"enable"`
}
type image struct {
   CutFaceImage bool    `mapstructure:"cut-face-image"`
   OriginWidth  float64 `mapstructure:"origin-width"`
   OriginHeight float64 `mapstructure:"origin-height"`
}
// 梯控设备
type nvcs struct {
   Model string `mapstructure:"model"` // 型号
   Port  string `mapstructure:"port"`  // 端口
   OSD   string `mapstructure:"osd"`
}
type rateLimitConfig struct {
@@ -57,6 +74,8 @@
var ForwardConf = &forward{}
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.
@@ -65,6 +84,7 @@
   v := viper.New()
   v.SetConfigType("yaml")
   v.SetConfigName("gat1400")
   v.AddConfigPath("./")
   v.AddConfigPath("./config/")
   v.AddConfigPath("../config/")
   err = v.ReadInConfig()
@@ -85,6 +105,8 @@
   v.UnmarshalKey("forward", ForwardConf)
   v.UnmarshalKey("rate-limit", RateLimitConf)
   v.UnmarshalKey("client", ClientConf)
   v.UnmarshalKey("nvcs", NVCSConf)
   v.UnmarshalKey("image", ImageConf)
   if LogConf.Level == "" {
      LogConf.Level = "info"