connect/config/config.go

package config

import (
	"log"

	"github.com/spf13/viper"
	"x.sikey.com.cn/serverx/confx"
)

var Websocket websocket

func loadWebsocketConfig() error {
	return viper.UnmarshalKey("websocket", &Websocket)
}

// MustLoadConfig 加载配置
func MustLoadConfig(file string) {
	confx.SetEnvConfig("NODE_ID", "ENVIRONMENT")
	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")
}

func GetEnvironment() string {
	return viper.GetString("ENVIRONMENT")
}

// // 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
// }