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"
|
"os"
|
)
|
|
var (
|
// config file name
|
configName = "aps-crm"
|
// 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 // 签发者
|
}
|
|
JWT2 struct {
|
SigningKey string // jwt签名
|
ExpiresTime string // 过期时间
|
BufferTime string // 缓冲时间
|
Issuer string // 签发者
|
}
|
|
System struct {
|
Env string // 环境值 develop test public
|
Port int // 端口
|
GrpcPort string //grpc端口
|
DbType string // 数据库类型
|
UseMultipoint bool // 多点登录拦截
|
UseRedis bool // 使用redis
|
LimitCountIP int
|
LimitTimeIP int
|
RouterPrefix string // 路由前缀
|
SudoPassword string // sudo密码
|
}
|
|
GrpcServiceAddr struct {
|
Aps string // aps服务地址
|
Admin string // admin服务地址
|
WMS string //wms服务地址
|
SRM string //srm服务地址
|
}
|
|
config struct {
|
// 系统配置
|
System System
|
|
// 日志
|
Log logx.Conf
|
|
// mysql配置
|
Mysql mysqlx.Conf
|
|
// redis配置
|
Redis redisx.Conf
|
|
// 验证码
|
Captcha Captcha
|
|
// JWT配置
|
JWT JWT
|
|
GrpcServiceAddr GrpcServiceAddr
|
}
|
)
|
|
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)
|
}
|
|
DBHost := os.Getenv("DB_HOST")
|
DBName := os.Getenv("DB_NAME")
|
DBPort := os.Getenv("DB_PORT")
|
DBUser := os.Getenv("DB_USER")
|
DBPasswd := os.Getenv("DB_PASSWD")
|
if len(DBHost) > 0 &&
|
len(DBName) > 0 &&
|
len(DBPort) > 0 &&
|
len(DBUser) > 0 &&
|
len(DBPasswd) > 0 {
|
Conf.Mysql.Dsn = DBUser + ":" + DBPasswd + "@tcp(" + DBHost + ":" + DBPort + ")/" + DBName + "?charset=utf8&parseTime=True&loc=Local"
|
}
|
|
AdminGrpc := os.Getenv("ADMIN_GRPC")
|
if len(AdminGrpc) > 0 {
|
Conf.GrpcServiceAddr.Admin = AdminGrpc
|
}
|
GrpcPort := os.Getenv("WMS_GRPC")
|
if len(GrpcPort) > 0 {
|
Conf.GrpcServiceAddr.WMS = GrpcPort
|
}
|
srmPort := os.Getenv("SRM_GRPC")
|
if len(srmPort) > 0 {
|
Conf.GrpcServiceAddr.SRM = srmPort
|
}
|
ApsGrpc := os.Getenv("GRPC_PORT")
|
Host := os.Getenv("HOST")
|
|
if len(Host) > 0 && len(ApsGrpc) > 0 {
|
Conf.GrpcServiceAddr.Aps = Host + ":" + ApsGrpc
|
//Conf.GrpcServiceAddr.Aps = DBUser + DBUser +
|
}
|
|
ShowConfig()
|
CheckConfig()
|
}
|
|
// 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.Printf(" GrpcServiceAddr: %+v", Conf.GrpcServiceAddr)
|
log.Println("......................................................")
|
}
|
|
func CheckConfig() {
|
if Conf.GrpcServiceAddr.Aps == "" {
|
log.Fatalf("%v 尚未配置", "GrpcServiceAddr.Aps")
|
}
|
}
|
