You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
35 lines
767 B
35 lines
767 B
package tool
|
|
|
|
import (
|
|
"github.com/spf13/viper"
|
|
"log"
|
|
)
|
|
|
|
// LoadConfig 加载配置
|
|
func LoadConfig() {
|
|
// 定义默认值
|
|
setConfigDefault()
|
|
|
|
// 配置文件属性设置
|
|
viper.SetConfigName("config")
|
|
viper.AddConfigPath(".")
|
|
|
|
// 读取配置
|
|
if err := viper.ReadInConfig(); err != nil {
|
|
// 如果需要对配置文件不存在错误,做特殊处理,使用:
|
|
//if _, ok := err.(viper.ConfigFileNotFoundError); ok {
|
|
//}
|
|
|
|
log.Println(err)
|
|
}
|
|
}
|
|
|
|
// setConfigDefault 设置默认值
|
|
func setConfigDefault() {
|
|
// app.addr
|
|
viper.SetDefault("app.addr", ":80")
|
|
viper.SetDefault("app.mode", "debug")
|
|
viper.SetDefault("db.type", "mysql")
|
|
viper.SetDefault("db.dsn", "user:password@tcp(127.0.0.1:3306)/test?charset=utf8mb4&parseTime=True&loc=Local")
|
|
}
|