zhangzengfei
2024-09-29 3737ab3dd0cc753be986638316c96cb3114601e4
config/config.go
@@ -15,10 +15,12 @@
}
type database struct {
   Driver string `mapstructure: "driver"`
   Host   string `mapstructure: "host"`
   Port   string `mapstructure: "port"`
   Name   string `mapstructure: "name"`
   Driver   string `mapstructure: "driver"`
   Host     string `mapstructure: "host"`
   Port     string `mapstructure: "port"`
   Name     string `mapstructure: "name"`
   Username string `mapstructure: "username"`
   Password string `mapstructure: "password"`
}
type elastic struct {
@@ -30,6 +32,7 @@
   TopHitsSize  int    `mapstructure: "topHitsSize"`
   CameraSize   int    `mapstructure: "cameraSize"`
   TimeInterval int    `mapstructure: "timeInterval"`
   BatchSize    int    `mapstructure: "batchSize"`
}
type app struct {
@@ -38,10 +41,21 @@
   LogLevel string `mapstructure: "logLevel"`
}
type api struct {
   Host                 string `mapstructure: "host"`
   Port                 string `mapstructure: "port"`
   CapAddrDaysThreshold int    `mapstructure:"capAddrDaysThreshold"`
   TimeThreshold        int    `mapstructure:"timeThreshold"`
   CsTimes              int    `mapstructure:"csTimes"`
   CsHours              int    `mapstructure:"csHours"`
   AInterval            int    `mapstructure:"aInterval"`
}
var LogConf = &LogConfig{}
var DataBase = &database{}
var Elastic = &elastic{}
var App = &app{}
var Api = &api{}
var LogBasePath string
var LogLevel int
@@ -49,7 +63,9 @@
   var err error
   viper.SetConfigType("yaml")
   viper.SetConfigName(env)
   viper.AddConfigPath("config")
   viper.AddConfigPath("./")
   viper.AddConfigPath("./config")
   viper.AddConfigPath("../config")
   err = viper.ReadInConfig()
   if err != nil {
      log.Fatal("error on parsing configuration file", err)
@@ -57,6 +73,7 @@
   viper.UnmarshalKey("elastic", Elastic)
   viper.UnmarshalKey("database", DataBase)
   viper.UnmarshalKey("app", App)
   viper.UnmarshalKey("api", Api)
   viper.UnmarshalKey("log", LogConf)
   logger.SetLevel(LogConf.Level)
   if viper.GetString("LogBasePath") != "" {