48 lines
1023 B
Go
48 lines
1023 B
Go
package models
|
||
|
||
import (
|
||
"fmt"
|
||
"log"
|
||
"path/filepath"
|
||
"runtime"
|
||
|
||
"github.com/spf13/viper"
|
||
)
|
||
|
||
var (
|
||
_, b, _, _ = runtime.Caller(0)
|
||
Root = filepath.Join(filepath.Dir(b), "..")
|
||
Viper *viper.Viper
|
||
)
|
||
|
||
func init() {
|
||
//如果命令行参数中有test，则使用测试环境的配置
|
||
config_file := filepath.Join(Root, "data", "config.yaml")
|
||
fmt.Println(config_file)
|
||
|
||
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", "")
|
||
|
||
viper.Set("video.endpoint", "")
|
||
viper.Set("video.accessKeyID", "")
|
||
viper.Set("video.accessKey", "")
|
||
}
|