package config
|
|
import (
|
"basic.com/valib/logger.git"
|
"github.com/fsnotify/fsnotify"
|
"github.com/spf13/viper"
|
"log"
|
)
|
|
type server struct {
|
AnalyServerId string `mapstructure:"analyServerId"`
|
NetworkAdapter string `mapstructure:"networkAdapter"`
|
}
|
|
var Server = &server{}
|
|
type esinfo struct {
|
EsIndex esindexlist `mapstructure:"index"`
|
}
|
|
type esindexlist struct {
|
AiOcean index `mapstructure:"aiOcean"`
|
}
|
|
type index struct {
|
IndexName string `mapstructure:"index"`
|
IndexType string `mapstructure:"type"`
|
}
|
|
type dbpersoncompare struct {
|
MysqlAddr string `mapstructure:"mysqlAddr"`
|
Username string `mapstructure:"username"`
|
Password string `mapstructure:"password"`
|
Database string `mapstructure:"database"`
|
PersonTable string `mapstructure:"personTable"`
|
ServePort int `mapstructure:"servePort"`
|
}
|
|
type espersoncompare struct {
|
ServePort int `mapstructure:"servePort"`
|
ESIP string `mapstructure:"esip"`
|
ESPort string `mapstructure:"esPort"`
|
}
|
|
var DbPersonCompInfo = &dbpersoncompare{}
|
|
var EsCompServerInfo = &espersoncompare{}
|
|
var EsInfo = &esinfo{}
|
|
type LogConfig struct {
|
Path string `mapstructure:"path"` //日志存储路径
|
Level int `mapstructure:"level"` //日志等级
|
MaxSize int `mapstructure:"maxSize"` //日志文件大小上限
|
MaxBackups int `mapstructure:"maxBackups"` //日志压缩包个数
|
MaxAge int `mapstructure:"maxAge"` //保留压缩包天数
|
}
|
|
var LogConf = &LogConfig{}
|
|
func Init(env string) {
|
var err error
|
v := viper.New()
|
v.SetConfigType("yaml")
|
v.SetConfigName(env)
|
v.AddConfigPath("./")
|
v.AddConfigPath("./config/")
|
err = v.ReadInConfig()
|
if err != nil {
|
log.Fatal("error on parsing configuration file")
|
}
|
read2Conf(v)
|
v.WatchConfig()
|
v.OnConfigChange(func(in fsnotify.Event) {
|
read2Conf(v)
|
})
|
}
|
|
func read2Conf(v *viper.Viper) {
|
v.UnmarshalKey("es", EsInfo)
|
v.UnmarshalKey("server", Server)
|
v.UnmarshalKey("dbpersoncompare", DbPersonCompInfo)
|
v.UnmarshalKey("espersoncompare", EsCompServerInfo)
|
v.UnmarshalKey("log", LogConf)
|
logger.SetLevel(LogConf.Level)
|
}
|
