94 lines
2.0 KiB
Go
94 lines
2.0 KiB
Go
package config
|
|
|
|
import (
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
const (
|
|
LogLevelKey = "LOG_LEVEL" // 日志级别
|
|
LogPathKey = "LOG_PATH" // 日志路径
|
|
LogFormatKey = "LOG_FORMAT" // 日志格式
|
|
PortKey = "PORT" // 端口
|
|
AppEnvKey = "APP_ENV" // 环境
|
|
|
|
DBUserNameKey = "DB_USERNAME" // 数据库用户名
|
|
DBPasswordKey = "DB_PASSWORD" // 数据库密码
|
|
DBHostKey = "DB_HOST" // 数据库地址
|
|
DBPortKey = "DB_PORT" // 数据库端口
|
|
DBNameKey = "DB_NAME" // 数据库名
|
|
)
|
|
|
|
func init() {
|
|
// 设置默认值
|
|
|
|
// 日志
|
|
viper.SetDefault(LogLevelKey, "debug")
|
|
viper.SetDefault(LogPathKey, "./data/logs/ecpm.log")
|
|
viper.SetDefault(LogFormatKey, "json")
|
|
|
|
// 端口
|
|
viper.SetDefault(PortKey, "8080")
|
|
|
|
// 环境
|
|
viper.SetDefault(AppEnvKey, "")
|
|
|
|
// 数据库
|
|
viper.SetDefault(DBUserNameKey, "root")
|
|
viper.SetDefault(DBPasswordKey, "youtu!0113")
|
|
viper.SetDefault(DBHostKey, "localhost")
|
|
viper.SetDefault(DBPortKey, "3306")
|
|
viper.SetDefault(DBNameKey, "ecpm")
|
|
|
|
viper.AutomaticEnv()
|
|
}
|
|
|
|
// GetLogLevel 获取日志级别
|
|
func GetLogLevel() string {
|
|
return viper.GetString(LogLevelKey)
|
|
}
|
|
|
|
// GetLogPath 获取日志路径
|
|
func GetLogPath() string {
|
|
return viper.GetString(LogPathKey)
|
|
}
|
|
|
|
// GetLogFormat 获取日志格式
|
|
func GetLogFormat() string {
|
|
return viper.GetString(LogFormatKey)
|
|
}
|
|
|
|
// GetPort 获取端口
|
|
func GetPort() string {
|
|
return viper.GetString(PortKey)
|
|
}
|
|
|
|
// GetAppEnv 获取环境
|
|
func GetAppEnv() string {
|
|
return viper.GetString(AppEnvKey)
|
|
}
|
|
|
|
// GetDBUserName 获取数据库用户名
|
|
func GetDBUserName() string {
|
|
return viper.GetString(DBUserNameKey)
|
|
}
|
|
|
|
// GetDBPassword 获取数据库密码
|
|
func GetDBPassword() string {
|
|
return viper.GetString(DBPasswordKey)
|
|
}
|
|
|
|
// GetDBHost 获取数据库地址
|
|
func GetDBHost() string {
|
|
return viper.GetString(DBHostKey)
|
|
}
|
|
|
|
// GetDBPort 获取数据库端口
|
|
func GetDBPort() int {
|
|
return viper.GetInt(DBPortKey)
|
|
}
|
|
|
|
// GetDBName 获取数据库名
|
|
func GetDBName() string {
|
|
return viper.GetString(DBNameKey)
|
|
}
|