package config
|
|
import (
|
"fmt"
|
|
"basic.com/valib/logger.git"
|
"github.com/fsnotify/fsnotify"
|
"github.com/spf13/viper"
|
)
|
|
type mainConfig struct {
|
ServePort int `mapstructure:"servePort"`
|
WorkerNum int `mapstructure:"workerNum"`
|
MysqlAddr string `mapstructure:"mysqlAddr"`
|
Username string `mapstructure:"username"`
|
Password string `mapstructure:"password"`
|
Database string `mapstructure:"database"`
|
PersonTable string `mapstructure:"personTable"`
|
}
|
|
var MainConf = &mainConfig{}
|
|
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() error {
|
var err error
|
v := viper.New()
|
v.SetConfigType("yaml")
|
v.SetConfigName("compare")
|
v.AddConfigPath("./")
|
v.AddConfigPath("./config/")
|
v.AddConfigPath("../config/")
|
err = v.ReadInConfig()
|
if err != nil {
|
fmt.Printf("error on parsing configuration file, %s, config file compare.yaml\n", err.Error())
|
return err
|
}
|
|
read2Conf(v)
|
v.WatchConfig()
|
v.OnConfigChange(func(in fsnotify.Event) {
|
read2Conf(v)
|
})
|
|
return nil
|
}
|
|
func read2Conf(v *viper.Viper) {
|
v.UnmarshalKey("main", MainConf)
|
v.UnmarshalKey("log", LogConf)
|
|
logger.SetLevel(LogConf.Level)
|
|
if MainConf.WorkerNum == 0 {
|
MainConf.WorkerNum = 30
|
}
|
}
|
