package conf
|
|
import (
|
"aps_crm/pkg/convertx"
|
"aps_crm/pkg/logx"
|
"aps_crm/pkg/mysqlx"
|
"aps_crm/pkg/redisx"
|
"flag"
|
"github.com/spf13/viper"
|
"log"
|
)
|
|
var (
|
// config file name
|
configName = "aps-admin"
|
// config file paths
|
configPaths = []string{
|
"./",
|
"../",
|
"./conf",
|
"../conf",
|
}
|
)
|
|
type (
|
Captcha struct {
|
KeyLong int // 验证码长度
|
ImgWidth int // 验证码宽度
|
ImgHeight int // 验证码高度
|
OpenCaptcha int // 防爆破验证码开启此数，0代表每次登录都需要验证码，其他数字代表错误密码次数，如3代表错误三次后出现验证码
|
OpenCaptchaTimeOut int // 防爆破验证码超时时间，单位：s(秒)
|
}
|
|
JWT struct {
|
SigningKey string // jwt签名
|
ExpiresTime string // 过期时间
|
BufferTime string // 缓冲时间
|
Issuer string // 签发者
|
}
|
|
System struct {
|
Env string // 环境值 develop test public
|
Port int // 端口
|
DbType string // 数据库类型
|
UseMultipoint bool // 多点登录拦截
|
UseRedis bool // 使用redis
|
LimitCountIP int
|
LimitTimeIP int
|
RouterPrefix string // 路由前缀
|
SudoPassword string // sudo密码
|
}
|
|
config struct {
|
// 系统配置
|
System System
|
|
// 日志
|
Log logx.Conf
|
|
// mysql配置
|
Mysql mysqlx.Conf
|
|
// redis配置
|
Redis redisx.Conf
|
|
// 验证码
|
Captcha Captcha
|
|
// JWT配置
|
JWT JWT
|
}
|
)
|
|
var (
|
Conf config
|
Viper *viper.Viper
|
)
|
|
func init() {
|
var configFile string
|
flag.StringVar(&configFile, "config", "", "config file in json")
|
flag.Parse()
|
|
if len(configFile) > 0 {
|
configName = configFile
|
}
|
|
Viper = viper.New()
|
Viper.SetConfigName(configName)
|
for _, path := range configPaths {
|
Viper.AddConfigPath(path)
|
}
|
|
if err := Viper.ReadInConfig(); err != nil {
|
log.Fatalf("ReadInConfig err:%v", err)
|
}
|
|
if err := Viper.Unmarshal(&Conf); err != nil {
|
log.Fatalf("Unmarshal err:%v", err)
|
}
|
ShowConfig()
|
}
|
|
// WriteConfig 回写配置
|
func (c *config) WriteConfig() error {
|
cs := convertx.StructToMap(Conf)
|
for k, v := range cs {
|
Viper.Set(k, v)
|
}
|
return Viper.WriteConfig()
|
}
|
|
func ShowConfig() {
|
log.Println("......................................................")
|
log.Printf(" System: %+v", Conf.System)
|
log.Printf(" Log: %+v", Conf.Log)
|
log.Printf(" Mysql: %+v", Conf.Mysql)
|
log.Printf(" Captcha: %+v", Conf.Captcha)
|
log.Printf(" JWT: %+v", Conf.JWT)
|
log.Println("......................................................")
|
}
|
