liuxiaolong
2019-09-20 5d78b36e851a5862302f2e40c99a4477020c79bc
extend/config/config.go
@@ -1,9 +1,8 @@
package config
import (
   "log"
   "github.com/spf13/viper"
   "log"
)
type server struct {
@@ -11,6 +10,7 @@
   JwtSecret string `mapstructure: "jwtSecret"`
   JwtExpire string `mapstructure: "jwtExpire"`
   Url       string `mapstructure: "url"`
   AnalyServerId string `mapstructure: "analyServerId"`
   ImageUrl  string `mapstructure: "imageUrl"`
   PublicDomain string `mapstructure: "publicDomain"`
   NetworkAdapter string `mapstructure: "networkAdapter"`
@@ -26,15 +26,6 @@
var WeedFs = &weedfs{}
type redis struct {
   Host     string `mapstructure:"host"`
   Port     int    `mapstructure:"port"`
   Password string `mapstructure:"password`
   DBNum    int    `mapstructure:"db"`
}
var RedisConf = &redis{}
type database struct {
   Drive    string `mapstructure:"drive"`
   Name     string `mapstructure:"name"`
@@ -45,6 +36,7 @@
type esinfo struct {
   Masterip string      `mapstructure:"masterip"`
   Httpport string      `mapstructure:"httpport"`
   Shards string       `mapstructure:"shards"`
   EsIndex  esindexlist `mapstructure:"index"`
}
@@ -54,22 +46,42 @@
   Dbtablepersons index `mapstructure:"dbtablepersons"`
   Personaction   index `mapstructure:"personaction"`
}
type index struct {
   IndexName string `mapstructure:"index"`
   IndexType string `mapstructure:"type"`
}
type sopath struct {
   Ip string `mapstructure:"ip"`
   Port string `mapstructure:"port"`
}
var SoPath = &sopath{}
var EsInfo = &esinfo{}
var DBconf = &database{}
type CompareServer struct {
   Url []string `mapstructure:"url"`
type facedetect struct {
   Ip string `mapstructure:"Ip"`
   Port int `mapstructure:"port"`
}
var CompServerInfo = &CompareServer{}
type dbpersoncompare struct {
   Ip string `mapstructure:"ip"`
   Port int `mapstructure:"port"`
}
type espersoncompare struct {
   Port int `mapstructure:"port"`
   Ips []string `mapstructure:"ips"`
}
var DbPersonCompInfo = &dbpersoncompare{}
var EsCompServerInfo = &espersoncompare{}
var FaceDetectSet = &facedetect{}
// Init is an exported method that takes the environment starts the viper
// (external lib) and returns the configuration struct.
@@ -77,16 +89,17 @@
   var err error
   viper.SetConfigType("yaml")
   viper.SetConfigName(env)
   viper.AddConfigPath("../config/")
   viper.AddConfigPath("config/")
   viper.AddConfigPath("/opt/vasystem/config/")
   err = viper.ReadInConfig()
   if err != nil {
      log.Fatal("error on parsing configuration file")
   }
   viper.UnmarshalKey("es", EsInfo)
   viper.UnmarshalKey("server", Server)
   viper.UnmarshalKey("redis", RedisConf)
   viper.UnmarshalKey("database", DBconf)
   viper.UnmarshalKey("weedfs", WeedFs)
   viper.UnmarshalKey("compare", CompServerInfo)
   viper.UnmarshalKey("sopath",SoPath)
   viper.UnmarshalKey("facedetect", FaceDetectSet)
   viper.UnmarshalKey("dbpersoncompare",DbPersonCompInfo)
   viper.UnmarshalKey("espersoncompare", EsCompServerInfo)
}