145 lines
4.9 KiB
Go
145 lines
4.9 KiB
Go
package config
|
||
|
||
import (
|
||
"fmt"
|
||
"os"
|
||
|
||
"github.com/fsnotify/fsnotify"
|
||
"github.com/spf13/viper"
|
||
"go.uber.org/zap/zapcore"
|
||
)
|
||
|
||
// 系统配置，对应yml
|
||
// viper内置了mapstructure, yml文件用"-"区分单词, 转为驼峰方便
|
||
|
||
// 全局配置变量
|
||
var Conf = new(config)
|
||
|
||
type config struct {
|
||
System *SystemConfig `mapstructure:"system" json:"system"`
|
||
Logs *LogsConfig `mapstructure:"logs" json:"logs"`
|
||
Mysql *MysqlConfig `mapstructure:"mysql" json:"mysql"`
|
||
Casbin *CasbinConfig `mapstructure:"casbin" json:"casbin"`
|
||
Jwt *JwtConfig `mapstructure:"jwt" json:"jwt"`
|
||
RateLimit *RateLimitConfig `mapstructure:"rate-limit" json:"rateLimit"`
|
||
Ldap *LdapConfig `mapstructure:"ldap" json:"ldap"`
|
||
Email *EmailConfig `mapstructure:"email" json:"email"`
|
||
}
|
||
|
||
// 设置读取配置信息
|
||
func InitConfig() {
|
||
workDir, err := os.Getwd()
|
||
if err != nil {
|
||
panic(fmt.Errorf("读取应用目录失败:%s", err))
|
||
}
|
||
viper.SetConfigName("config")
|
||
viper.SetConfigType("yml")
|
||
viper.AddConfigPath(workDir + "/")
|
||
// 读取配置信息
|
||
err = viper.ReadInConfig()
|
||
|
||
// 热更新配置
|
||
viper.WatchConfig()
|
||
viper.OnConfigChange(func(e fsnotify.Event) {
|
||
// 将读取的配置信息保存至全局变量Conf
|
||
if err := viper.Unmarshal(Conf); err != nil {
|
||
panic(fmt.Errorf("初始化配置文件失败:%s \n", err))
|
||
}
|
||
// 读取rsa key
|
||
Conf.System.RSAPublicBytes = RSAReadKeyFromFile(Conf.System.RSAPublicKey)
|
||
Conf.System.RSAPrivateBytes = RSAReadKeyFromFile(Conf.System.RSAPrivateKey)
|
||
})
|
||
|
||
if err != nil {
|
||
panic(fmt.Errorf("读取配置文件失败:%s \n", err))
|
||
}
|
||
// 将读取的配置信息保存至全局变量Conf
|
||
if err := viper.Unmarshal(Conf); err != nil {
|
||
panic(fmt.Errorf("初始化配置文件失败:%s \n", err))
|
||
}
|
||
// 读取rsa key
|
||
Conf.System.RSAPublicBytes = RSAReadKeyFromFile(Conf.System.RSAPublicKey)
|
||
Conf.System.RSAPrivateBytes = RSAReadKeyFromFile(Conf.System.RSAPrivateKey)
|
||
|
||
}
|
||
|
||
// 从文件中读取RSA key
|
||
func RSAReadKeyFromFile(filename string) []byte {
|
||
f, err := os.Open(filename)
|
||
var b []byte
|
||
|
||
if err != nil {
|
||
return b
|
||
}
|
||
defer f.Close()
|
||
fileInfo, _ := f.Stat()
|
||
b = make([]byte, fileInfo.Size())
|
||
f.Read(b)
|
||
return b
|
||
}
|
||
|
||
type SystemConfig struct {
|
||
Mode string `mapstructure:"mode" json:"mode"`
|
||
UrlPathPrefix string `mapstructure:"url-path-prefix" json:"urlPathPrefix"`
|
||
Port int `mapstructure:"port" json:"port"`
|
||
InitData bool `mapstructure:"init-data" json:"initData"`
|
||
RSAPublicKey string `mapstructure:"rsa-public-key" json:"rsaPublicKey"`
|
||
RSAPrivateKey string `mapstructure:"rsa-private-key" json:"rsaPrivateKey"`
|
||
RSAPublicBytes []byte `mapstructure:"-" json:"-"`
|
||
RSAPrivateBytes []byte `mapstructure:"-" json:"-"`
|
||
}
|
||
|
||
type LogsConfig struct {
|
||
Level zapcore.Level `mapstructure:"level" json:"level"`
|
||
Path string `mapstructure:"path" json:"path"`
|
||
MaxSize int `mapstructure:"max-size" json:"maxSize"`
|
||
MaxBackups int `mapstructure:"max-backups" json:"maxBackups"`
|
||
MaxAge int `mapstructure:"max-age" json:"maxAge"`
|
||
Compress bool `mapstructure:"compress" json:"compress"`
|
||
}
|
||
|
||
type MysqlConfig struct {
|
||
Username string `mapstructure:"username" json:"username"`
|
||
Password string `mapstructure:"password" json:"password"`
|
||
Database string `mapstructure:"database" json:"database"`
|
||
Host string `mapstructure:"host" json:"host"`
|
||
Port int `mapstructure:"port" json:"port"`
|
||
Query string `mapstructure:"query" json:"query"`
|
||
LogMode bool `mapstructure:"log-mode" json:"logMode"`
|
||
TablePrefix string `mapstructure:"table-prefix" json:"tablePrefix"`
|
||
Charset string `mapstructure:"charset" json:"charset"`
|
||
Collation string `mapstructure:"collation" json:"collation"`
|
||
}
|
||
|
||
type CasbinConfig struct {
|
||
ModelPath string `mapstructure:"model-path" json:"modelPath"`
|
||
}
|
||
|
||
type JwtConfig struct {
|
||
Realm string `mapstructure:"realm" json:"realm"`
|
||
Key string `mapstructure:"key" json:"key"`
|
||
Timeout int `mapstructure:"timeout" json:"timeout"`
|
||
MaxRefresh int `mapstructure:"max-refresh" json:"maxRefresh"`
|
||
}
|
||
|
||
type RateLimitConfig struct {
|
||
FillInterval int64 `mapstructure:"fill-interval" json:"fillInterval"`
|
||
Capacity int64 `mapstructure:"capacity" json:"capacity"`
|
||
}
|
||
|
||
type LdapConfig struct {
|
||
LdapUrl string `mapstructure:"ldap-url" json:"ldapUrl"`
|
||
LdapBaseDN string `mapstructure:"ldap-base-dn" json:"ldapBaseDN"`
|
||
LdapAdminDN string `mapstructure:"ldap-admin-dn" json:"ldapAdminDN"`
|
||
LdapAdminPass string `mapstructure:"ldap-admin-pass" json:"ldapAdminPass"`
|
||
LdapUserDN string `mapstructure:"ldap-user-dn" json:"ldapUserDN"`
|
||
LdapGroupDN string `mapstructure:"ldap-group-dn" json:"ldapGroupDN"`
|
||
}
|
||
type EmailConfig struct {
|
||
Host string `mapstructure:"host" json:"host"`
|
||
Port string `mapstructure:"port" json:"port"`
|
||
User string `mapstructure:"user" json:"user"`
|
||
Pass string `mapstructure:"pass" json:"pass"`
|
||
From string `mapstructure:"from" json:"from"`
|
||
}
|