package config import ( "log" "github.com/spf13/viper" "x.sikey.com.cn/serverx/confx" ) const ( envNodeId = "NODE_ID" ) var Websocket websocket func loadWebsocketConfig() error { return viper.UnmarshalKey("websocket", &Websocket) } // MustLoadConfig 加载配置 func MustLoadConfig(file string) { confx.SetEnvConfig(envNodeId) if err := confx.LoadConfig(file, loadWebsocketConfig); err != nil { log.Fatalln(err) } log.Printf("Load configs [ toml ] from path [%s] \n", file) } func GetServerPort() int { return viper.GetInt("port") } // // MustLoadLogger 初始化日志 // func MustLoadLogger() zlog.Writer { // var ( // name = Config.Name // env = Config.Environment // path = viper.GetString("log.path") // ) // writer, err := zlog.NewZeroWriter(name, path, env) // if err != nil { // panic(err) // } // return writer // }