85 lines
1.9 KiB
Go
85 lines
1.9 KiB
Go
package config
|
||
|
||
import (
|
||
"fmt"
|
||
"io/ioutil"
|
||
"log"
|
||
"os"
|
||
"strings"
|
||
|
||
"github.com/spf13/viper"
|
||
)
|
||
|
||
// 数据库配置项
|
||
var cfgDatabase *viper.Viper
|
||
|
||
// 应用配置项
|
||
var cfgApplication *viper.Viper
|
||
|
||
// Token配置项
|
||
var cfgJwt *viper.Viper
|
||
|
||
// Log配置项
|
||
var cfgLogger *viper.Viper
|
||
|
||
// Ssl配置项 非必须
|
||
var cfgSsl *viper.Viper
|
||
|
||
// 代码生成配置项 非必须
|
||
var cfgGen *viper.Viper
|
||
|
||
//载入配置文件
|
||
func Setup(path string) {
|
||
viper.SetConfigFile(path)
|
||
content, err := ioutil.ReadFile(path)
|
||
if err != nil {
|
||
log.Fatal(fmt.Sprintf("Read config file fail: %s", err.Error()))
|
||
}
|
||
|
||
//Replace environment variables
|
||
err = viper.ReadConfig(strings.NewReader(os.ExpandEnv(string(content))))
|
||
if err != nil {
|
||
log.Fatal(fmt.Sprintf("Parse config file fail: %s", err.Error()))
|
||
}
|
||
|
||
cfgDatabase = viper.Sub("settings.database")
|
||
if cfgDatabase == nil {
|
||
panic("No found settings.database in the configuration")
|
||
}
|
||
DatabaseConfig = InitDatabase(cfgDatabase)
|
||
|
||
cfgApplication = viper.Sub("settings.application")
|
||
if cfgApplication == nil {
|
||
panic("No found settings.application in the configuration")
|
||
}
|
||
ApplicationConfig = InitApplication(cfgApplication)
|
||
|
||
cfgJwt = viper.Sub("settings.jwt")
|
||
if cfgJwt == nil {
|
||
panic("No found settings.jwt in the configuration")
|
||
}
|
||
JwtConfig = InitJwt(cfgJwt)
|
||
|
||
cfgLogger = viper.Sub("settings.logger")
|
||
if cfgLogger == nil {
|
||
panic("No found settings.logger in the configuration")
|
||
}
|
||
LoggerConfig = InitLog(cfgLogger)
|
||
|
||
cfgSsl = viper.Sub("settings.ssl")
|
||
if cfgSsl == nil {
|
||
// Ssl不是系统强制要求的配置，默认可以不用配置，将设置为关闭状态
|
||
fmt.Println("warning config not found settings.ssl in the configuration")
|
||
SslConfig = new(Ssl)
|
||
SslConfig.Enable = false
|
||
} else {
|
||
SslConfig = InitSsl(cfgSsl)
|
||
}
|
||
|
||
cfgGen = viper.Sub("settings.gen")
|
||
if cfgGen == nil {
|
||
panic("No found settings.gen")
|
||
}
|
||
GenConfig = InitGen(cfgGen)
|
||
}
|