40 lines
892 B
Go
40 lines
892 B
Go
package models
|
||
|
||
import (
|
||
"log"
|
||
"os"
|
||
|
||
"github.com/spf13/viper"
|
||
)
|
||
|
||
var Viper *viper.Viper
|
||
|
||
func init() {
|
||
//如果命令行参数中有test，则使用测试环境的配置
|
||
var config_file string = "./data/config.yaml"
|
||
if len(os.Args) > 1 && os.Args[1] == "test" {
|
||
log.Println("使用测试环境配置")
|
||
config_file = "./data/config_test.yaml"
|
||
}
|
||
viper.SetConfigFile(config_file)
|
||
if err := viper.ReadInConfig(); err != nil {
|
||
log.Println("读取配置文件失败", err)
|
||
生成配置文件()
|
||
}
|
||
Viper = viper.GetViper()
|
||
}
|
||
|
||
func 生成配置文件() {
|
||
viper.Set("mysql.host", "")
|
||
viper.Set("mysql.port", 3306)
|
||
viper.Set("mysql.user", "")
|
||
viper.Set("mysql.password", "")
|
||
viper.Set("mysql.dbname", "")
|
||
viper.Set("mysql.charset", "utf8mb4")
|
||
viper.Set("mysql.maxOpenConns", 100)
|
||
|
||
viper.Set("oss.endpoint", "")
|
||
viper.Set("oss.accessID", "")
|
||
viper.Set("oss.accessKey", "")
|
||
}
|